Create core/model_selector.py

core/model_selector.py

@@ -1,14 +1,32 @@
+from langdetect import detect
+
 def choose_model(user_input: str) -> str:
-    """Автоматически выбирает модель Hugging Face по типу запроса"""
+    """
+    Автоматически подбирает оптимальную модель Hugging Face
+    в зависимости от языка, задачи и сложности запроса.
+    """
     text = user_input.lower()
 
-    if any(word in text for word in ["объясни", "физика", "квант", "математика", "наука", "алгоритм", "данные"]):
-        return "microsoft/phi-3.5-mini-instruct"  # логическая, точная
-    elif any(word in text for word in ["придумай", "сочини", "творчески", "история", "поэзия", "вообрази"]):
-        return "mistralai/Mistral-7B-Instruct-v0.2"  # креатив
-    elif any(word in text for word in ["диалог", "совет", "эмоция", "отношения", "помоги понять"]):
-        return "google/gemma-2-2b-it"  # дружелюбная и эмпатичная
-    elif any(word in text for word in ["код", "python", "программа", "ошибка", "скрипт"]):
-        return "codellama/CodeLlama-7b-Instruct-hf"  # тех-запросы
-    else:
-        return "microsoft/phi-3.5-mini-instruct"  # по умолчанию — точная и быстрая
+    # Определяем язык
+    try:
+        lang = detect(user_input)
+    except:
+        lang = "en"
+
+    # === Категории задач ===
+    if any(x in text for x in ["код", "python", "script", "программа", "function", "code"]):
+        return "codellama/CodeLlama-7b-Instruct-hf"
+
+    elif any(x in text for x in ["анализ", "объясни", "научи", "почему", "как работает"]):
+        return "mistralai/Mistral-7B-Instruct-v0.3"
+
+    elif any(x in text for x in ["текст", "переведи", "история", "расскажи", "объяснение"]):
+        return "google/gemma-2b-it"
+
+    elif any(x in text for x in ["чат", "ответь", "расскажи про", "что такое"]):
+        return "microsoft/Phi-3-mini-4k-instruct"
+
+    # === fallback ===
+    if lang != "ru":
+        return "mistralai/Mistral-7B-Instruct-v0.3"
+    return "microsoft/Phi-3-mini-4k-instruct"