Update lunacode.py

lunacode.py

@@ -4,9 +4,6 @@ from bs4 import BeautifulSoup
 from duckduckgo_search import DDGS
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import wikipedia
-from gtts import gTTS
-import os
-import uuid
 
 model_path = "cosmosai471/Luna-v2"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
@@ -15,25 +12,16 @@ 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 for kw in code_keywords):
+    if any(kw in query.lower() for kw in code_keywords):
         return "code"
-    elif any(kw in query for kw in creative_keywords):
+    elif any(kw in query.lower() 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":
@@ -101,13 +89,6 @@ def scrape_first_good_content(urls):
             continue
     return None, None
 
-
-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 smart_luna_answer(user_question, max_tokens=512):
     temperature, top_p = get_generation_params(user_question)