Update lunacode.py

lunacode.py

@@ -15,16 +15,25 @@ model = AutoModelForCausalLM.from_pretrained(
 ).to("cuda" if torch.cuda.is_available() else "cpu")
 
 def detect_mode(query):
+    if not isinstance(query, str):
+        return "general"
+    query = query.lower()
     code_keywords = ["code", "program", "python", "javascript", "function", "script", "build", "html", "css"]
     creative_keywords = ["story", "write a story", "poem", "creative", "imagine", "novel", "dialogue"]
 
-    if any(kw in query.lower() for kw in code_keywords):
+    if any(kw in query for kw in code_keywords):
         return "code"
-    elif any(kw in query.lower() for kw in creative_keywords):
+    elif any(kw in query for kw in creative_keywords):
         return "creative"
     else:
         return "general"
 
+def text_to_speech_luna(text):
+    tts = gTTS(text=text, lang="en")
+    filename = f"/tmp/luna_reply_{uuid.uuid4().hex}.mp3"
+    tts.save(filename)
+    return filename        
+
 def get_generation_params(query):
     mode = detect_mode(query)
     if mode == "code":