Update app/services/models.py

app/services/models.py

@@ -69,31 +69,19 @@ def load_models():
     
     # ========== STT MODELI ==========
     try:
-        # --- MUHIM O'ZGARTIRISH ---
-        # Noto'g'ri yoki shaxsiy model identifikatori o'rniga
-        # to'g'ri va ommaviy modelni ishlatamiz.
-        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)
+        # --- 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"
         
         MODELS["stt_pipeline"] = pipeline(
             "automatic-speech-recognition",
-            model=stt_model,
-            tokenizer=stt_processor.tokenizer,
-            feature_extractor=stt_processor.feature_extractor,
-            max_new_tokens=128,
+            model=stt_model_id,
             torch_dtype=TORCH_DTYPE,
             device=DEVICE,
         )
+        # ------------------------------------------------
         
         MODEL_STATUS["stt"] = True
         logger.info("   ✅ STT modeli tayyor")
@@ -102,22 +90,14 @@ def load_models():
         logger.error(f"   ❌ STT modelini yuklashda xatolik: {e}", exc_info=True)
         MODEL_STATUS["stt"] = False
 
-    # ========== TTS MODELLARI (3 ta: uzb, eng, rus) ==========
+    # ========== TTS MODELLARI (Bu usulda o'zi to'g'ri edi) ==========
     
     # TTS O'ZBEKCHA
     try:
         logger.info("   🎧 TTS O'ZBEKCHA (facebook/mms-tts-uzb-script_cyrillic) modeli yuklanmoqda...")
-        tts_model_path = "facebook/mms-tts-uzb-script_cyrillic"
-        
-        MODELS["tts_uzb_pipeline"] = pipeline(
-            "text-to-speech",
-            model=tts_model_path,
-            device=DEVICE
-        )
-        
+        MODELS["tts_uzb_pipeline"] = pipeline("text-to-speech", model="facebook/mms-tts-uzb-script_cyrillic", 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
@@ -125,17 +105,9 @@ def load_models():
     # TTS INGLIZCHA
     try:
         logger.info("   🎧 TTS INGLIZCHA (facebook/mms-tts-eng) modeli yuklanmoqda...")
-        tts_eng_path = "facebook/mms-tts-eng"
-        
-        MODELS["tts_eng_pipeline"] = pipeline(
-            "text-to-speech",
-            model=tts_eng_path,
-            device=DEVICE
-        )
-        
+        MODELS["tts_eng_pipeline"] = pipeline("text-to-speech", model="facebook/mms-tts-eng", 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
@@ -143,17 +115,9 @@ def load_models():
     # TTS RUSCHA
     try:
         logger.info("   🎧 TTS RUSCHA (facebook/mms-tts-rus) modeli yuklanmoqda...")
-        tts_rus_path = "facebook/mms-tts-rus"
-        
-        MODELS["tts_rus_pipeline"] = pipeline(
-            "text-to-speech",
-            model=tts_rus_path,
-            device=DEVICE
-        )
-            
+        MODELS["tts_rus_pipeline"] = pipeline("text-to-speech", model="facebook/mms-tts-rus", 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
@@ -161,18 +125,16 @@ def load_models():
     # ========== LLM MODELI (Gemini) ==========
     try:
         logger.info("   🧠 LLM (Gemini) modeli yuklanmoqda...")
-        # Eskirgan "gemini-2.0-flash-exp" o'rniga yangi va barqaror model
         MODELS["llm"] = genai.GenerativeModel("gemini-1.5-flash") 
         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 model yuklanmadi. Loyiha ishlay olmaydi.")
+        raise RuntimeError("❌ CRITICAL: Hech qanday AI modeli yuklanmadi. Ilova ishlay olmaydi.")
     
     logger.info("=" * 60)
     logger.info("✅ Modellar yuklash yakunlandi:")