Update app/services/models.py

app/services/models.py

@@ -8,7 +8,7 @@ YAXSHILANISHLAR:
 3. ✅ TTS - Speed o'chirildi, faqat til parametri
 4. ✅ Error handling - Hamma joyda try-except
 5. ✅ Fallback responses - Xatolik bo'lsa default javob qaytarish
-""" 
+"""
 
 import subprocess
 import numpy as np
@@ -69,35 +69,51 @@ def load_models():
     
     # ========== STT MODELI ==========
     try:
-        # --- SIZNING TAKLIFINGIZ BILAN YAXSHILANDI ---
-        # Model va uning qismlarini (processor, tokenizer) alohida yuklash o'rniga,
-        # to'g'ridan-to'g'ri pipeline orqali yuklaymiz. Bu ancha sodda va ishonchli.
-        logger.info("   📥 STT (islomov/whisper-small-uzbek) modeli pipeline orqali yuklanmoqda...")
-        stt_model_id = "islomov/whisper-small-uzbek"
+        logger.info("   📥 STT (islomov/rubaistt_v2_medium) modeli yuklanmoqda...")
+        stt_model_id = "islomov/rubaistt_v2_medium"
+        
+        stt_model = AutoModelForSpeechSeq2Seq.from_pretrained(
+            stt_model_id, 
+            torch_dtype=TORCH_DTYPE, 
+            low_cpu_mem_usage=True, 
+            use_safetensors=True
+        )
+        stt_model.to(DEVICE)
+        stt_processor = AutoProcessor.from_pretrained(stt_model_id)
         
         MODELS["stt_pipeline"] = pipeline(
             "automatic-speech-recognition",
-            model=stt_model_id,
+            model=stt_model,
+            tokenizer=stt_processor.tokenizer,
+            feature_extractor=stt_processor.feature_extractor,
+            max_new_tokens=128,
             torch_dtype=TORCH_DTYPE,
             device=DEVICE,
         )
-        # ------------------------------------------------
         
         MODEL_STATUS["stt"] = True
         logger.info("   ✅ STT modeli tayyor")
         
     except Exception as e:
-        logger.error(f"   ❌ STT modelini yuklashda xatolik: {e}", exc_info=True)
+        logger.error(f"   ❌ STT modelini yuklashda xatolik: {e}")
         MODEL_STATUS["stt"] = False
 
-    # ========== TTS MODELLARI (Bu usulda o'zi to'g'ri edi) ==========
+    # ========== TTS MODELLARI (3 ta: uzb, eng, rus) ==========
     
     # TTS O'ZBEKCHA
     try:
         logger.info("   🎧 TTS O'ZBEKCHA (facebook/mms-tts-uzb-script_cyrillic) modeli yuklanmoqda...")
-        MODELS["tts_uzb_pipeline"] = pipeline("text-to-speech", model="facebook/mms-tts-uzb-script_cyrillic", device=DEVICE)
+        tts_model_path = "facebook/mms-tts-uzb-script_cyrillic"
+        
+        MODELS["tts_uzb_pipeline"] = pipeline(
+            "text-to-speech",
+            model=tts_model_path,
+            device=DEVICE
+        )
+        
         MODEL_STATUS["tts_uzb"] = True
         logger.info("   ✅ TTS O'ZBEK modeli tayyor")
+        
     except Exception as e:
         logger.error(f"   ❌ TTS O'ZBEK modelini yuklashda xatolik: {e}")
         MODEL_STATUS["tts_uzb"] = False
@@ -105,9 +121,17 @@ def load_models():
     # TTS INGLIZCHA
     try:
         logger.info("   🎧 TTS INGLIZCHA (facebook/mms-tts-eng) modeli yuklanmoqda...")
-        MODELS["tts_eng_pipeline"] = pipeline("text-to-speech", model="facebook/mms-tts-eng", device=DEVICE)
+        tts_eng_path = "facebook/mms-tts-eng"
+        
+        MODELS["tts_eng_pipeline"] = pipeline(
+            "text-to-speech",
+            model=tts_eng_path,
+            device=DEVICE
+        )
+        
         MODEL_STATUS["tts_eng"] = True
         logger.info("   ✅ TTS INGLIZ modeli tayyor")
+        
     except Exception as e:
         logger.error(f"   ❌ TTS INGLIZ modelini yuklashda xatolik: {e}")
         MODEL_STATUS["tts_eng"] = False
@@ -115,9 +139,17 @@ def load_models():
     # TTS RUSCHA
     try:
         logger.info("   🎧 TTS RUSCHA (facebook/mms-tts-rus) modeli yuklanmoqda...")
-        MODELS["tts_rus_pipeline"] = pipeline("text-to-speech", model="facebook/mms-tts-rus", device=DEVICE)
+        tts_rus_path = "facebook/mms-tts-rus"
+        
+        MODELS["tts_rus_pipeline"] = pipeline(
+            "text-to-speech",
+            model=tts_rus_path,
+            device=DEVICE
+        )
+            
         MODEL_STATUS["tts_rus"] = True
         logger.info("   ✅ TTS RUS modeli tayyor")
+        
     except Exception as e:
         logger.error(f"   ❌ TTS RUS modelini yuklashda xatolik: {e}")
         MODEL_STATUS["tts_rus"] = False
@@ -125,16 +157,18 @@ def load_models():
     # ========== LLM MODELI (Gemini) ==========
     try:
         logger.info("   🧠 LLM (Gemini) modeli yuklanmoqda...")
-        MODELS["llm"] = genai.GenerativeModel("gemini-1.5-flash") 
+        # Sizning kodingizda "gemini-2.0-flash-exp" ishlatilgan ekan, shuni qoldiramiz
+        MODELS["llm"] = genai.GenerativeModel("gemini-2.0-flash-exp") 
         MODEL_STATUS["llm"] = True
         logger.info("   ✅ LLM modeli tayyor")
+        
     except Exception as e:
         logger.error(f"   ❌ LLM modelini yuklashda xatolik: {e}")
         MODEL_STATUS["llm"] = False
 
     # ========== NATIJA ==========
     if not any(MODEL_STATUS.values()):
-        raise RuntimeError("❌ CRITICAL: Hech qanday AI modeli yuklanmadi. Ilova ishlay olmaydi.")
+        raise RuntimeError("❌ CRITICAL: Hech qanday model yuklanmadi. Loyiha ishlay olmaydi.")
     
     logger.info("=" * 60)
     logger.info("✅ Modellar yuklash yakunlandi:")