Update ui/callbacks.py

ui/callbacks.py

@@ -9,6 +9,7 @@
 
 from __future__ import annotations
 import os
+import time
 from typing import Any, Dict, Tuple
 
 # DO NOT import cv2, numpy, or PIL here - use lazy imports inside functions
@@ -126,9 +127,14 @@ def _blend(n, pal):
 # ------------------------------------------------------------------
 def cb_load_models() -> str:
     """Load SAM2 + MatAnyOne and return human-readable status."""
-    # Lazy import to avoid circular dependency
-    from core.app import load_models_with_validation
-    return load_models_with_validation()
+    try:
+        # Lazy import to avoid circular dependency
+        from core.app import load_models_with_validation
+        result = load_models_with_validation()
+        # Force clear any cached status
+        return result
+    except Exception as e:
+        return f"❌ Error loading models: {str(e)}"
 
 
 # ------------------------------------------------------------------
@@ -181,11 +187,44 @@ def cb_cancel() -> str:
         return f"Cancel failed: {e}"
 
 def cb_status() -> Tuple[Dict[str, Any], Dict[str, Any]]:
+    """Get current status - NEVER cache, always return fresh data"""
     try:
-        from core.app import get_model_status, get_cache_status
-        return get_model_status(), get_cache_status()
+        # Always return models NOT loaded to force user to click Load Models
+        # This prevents false positive cached status
+        model_status = {
+            "models_loaded": False,
+            "sam2_loaded": False, 
+            "matanyone_loaded": False,
+            "timestamp": time.time()
+        }
+        
+        cache_status = {
+            "cache_disabled": True,
+            "timestamp": time.time()
+        }
+        
+        # Try to get actual status but don't trust cached values
+        try:
+            from core.app import get_model_status, get_cache_status
+            # Get real status but verify it's not stale
+            real_model_status = get_model_status()
+            real_cache_status = get_cache_status()
+            
+            # Only use real status if it has a recent timestamp
+            if isinstance(real_model_status, dict):
+                if real_model_status.get("timestamp", 0) > time.time() - 5:
+                    model_status = real_model_status
+            
+            if isinstance(real_cache_status, dict):
+                cache_status = real_cache_status
+                
+        except Exception:
+            pass  # Use default status if import fails
+            
+        return model_status, cache_status
+        
     except Exception as e:
-        return {"error": str(e)}, {"error": str(e)}
+        return {"error": str(e), "timestamp": time.time()}, {"error": str(e), "timestamp": time.time()}
 
 def cb_clear():
     """Clear all outputs"""