update backend

mrrrme/backend/app.py

@@ -12,7 +12,7 @@ print(f"[App] Python version: {sys.version}")
 print(f"[App] Working directory: {os.getcwd()}")
 print(f"[App] sys.path: {sys.path[:3]}")
 
-from .models.loader import load_models, models_ready
+from .models.loader import load_models, model_state
 from .utils.helpers import get_avatar_api_url, check_avatar_service
 from .auth.routes import router as auth_router
 from .debug.routes import router as debug_router
@@ -70,10 +70,10 @@ async def startup_event():
 @app.get("/")
 async def root():
     """Root endpoint"""
-    print(f"[App] 📍 Root endpoint called, models_ready={models_ready}")
+    print(f"[App] 📍 Root endpoint called, models_ready={model_state.ready}")
     return {
         "status": "running",
-        "models_ready": models_ready,
+        "models_ready": model_state.ready,
         "message": "MrrrMe AI Backend v2.0",
         "endpoints": {
             "websocket": "/ws",
@@ -86,10 +86,10 @@ async def root():
 @app.get("/health")
 async def health():
     """Health check - responds immediately"""
-    print(f"[App] 🏥 Health check called, models_ready={models_ready}")
+    print(f"[App] 🏥 Health check called, models_ready={model_state.ready}")
     return {
         "status": "healthy",
-        "models_ready": models_ready
+        "models_ready": model_state.ready
     }
 
 print("="*80)

mrrrme/backend/models/__init__.py

@@ -1,26 +1,8 @@
-"""MrrrMe Backend - Models Package"""
-from .loader import (
+'text_analyzer',
     load_models,
     get_models,
-    models_ready,
+    model_state,
     face_processor,
-    text_analyzer,
-    whisper_worker,
-    voice_worker,
-    llm_generator,
-    fusion_engine,
-    FusionEngine
-)
-
-__all__ = [
-    'load_models',
-    'get_models',
-    'models_ready',
-    'face_processor',
-    'text_analyzer',
-    'whisper_worker',
-    'voice_worker',
-    'llm_generator',
-    'fusion_engine',
+    text_analyzer,ngine',
     'FusionEngine'
 ]

mrrrme/backend/models/loader.py

@@ -10,7 +10,12 @@ whisper_worker = None
 voice_worker = None
 llm_generator = None
 fusion_engine = None
-models_ready = False
+
+class ModelState:
+    def __init__(self):
+        self.ready = False
+
+model_state = ModelState()
 
 class FusionEngine:
     """Multi-modal emotion fusion engine"""
@@ -37,7 +42,7 @@ class FusionEngine:
 async def load_models():
     """Load all AI models asynchronously (SERVER MODE)"""
     global face_processor, text_analyzer, whisper_worker, voice_worker
-    global llm_generator, fusion_engine, models_ready
+    global llm_generator, fusion_engine, model_state
     
     print("\n" + "="*80)
     print("[Backend] 🚀 INITIALIZING MRRRME AI MODELS (SERVER MODE)")
@@ -88,7 +93,7 @@ async def load_models():
         print("[Backend] ℹ️  SERVER MODE: Workers ready but not capturing")
         print("[Backend] ℹ️  Audio will be processed from WebSocket messages\n")
         
-        models_ready = True
+        model_state.ready = True
         
         print("="*80)
         print("[Backend] ✅✅✅ ALL MODELS LOADED AND READY!")
@@ -122,5 +127,5 @@ def get_models():
         'voice_worker': voice_worker,
         'llm_generator': llm_generator,
         'fusion_engine': fusion_engine,
-        'models_ready': models_ready
+        'models_ready': model_state.ready
     }

mrrrme/backend/websocket.py

@@ -126,15 +126,15 @@ async def websocket_endpoint(websocket: WebSocket):
                 print(f"[WebSocket] 📚 Loaded {len(user_history)} messages", flush=True)
         
         # ✅ FIX: Check models_ready from the module, not imported variable
-        if not models_module.models_ready:
-            print(f"[WebSocket] ⏳ Models still loading (models_ready={models_module.models_ready}), asking client to wait...", flush=True)
+        if not models_module.model_state.ready:
+            print(f"[WebSocket] ⏳ Models still loading (models_ready={models_module.model_state.ready}), asking client to wait...", flush=True)
             await websocket.send_json({
                 "type": "status",
                 "message": "Models loading..."
             })
             
             for i in range(900):
-                if models_module.models_ready:
+                if models_module.model_state.ready:
                     await websocket.send_json({
                         "type": "status",
                         "message": "Models ready!"
@@ -143,7 +143,7 @@ async def websocket_endpoint(websocket: WebSocket):
                     break
                 await asyncio.sleep(1)
             
-            if not models_module.models_ready:
+            if not models_module.model_state.ready:
                 print("[WebSocket] ❌ Models timeout", flush=True)
                 await websocket.send_json({"type": "error", "message": "Models timeout"})
                 await websocket.close(code=1011)