Update app.py

app.py

@@ -204,6 +204,35 @@ def simulate_damage_detection(rgb_image: np.ndarray, seed_from: np.ndarray = Non
         boxes.append({"bbox":[x1,y1,x2,y2], "score":conf, "label":"damage"})
     return boxes
 
+def auto_install_dependencies():
+    """Attempt to install dependencies if needed"""
+    try:
+        import importlib.util
+
+        # Check for PyTorch
+        if importlib.util.find_spec("torch") is None:
+            print("Installing PyTorch...")
+            os.system("pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu")
+
+        # Check for Detectron2
+        if importlib.util.find_spec("detectron2") is None:
+            print("Installing Detectron2...")
+            os.system("pip install git+https://github.com/facebookresearch/detectron2.git")
+
+        # Check for Gradio
+        if importlib.util.find_spec("gradio") is None:
+            print("Installing Gradio...")
+            os.system("pip install gradio")
+
+        print("Dependencies installation complete!")
+        return True
+    except Exception as e:
+        print(f"Error installing dependencies: {e}")
+        return False
+
+
+
+
 def run_damage_detection(pil_image: Image.Image, score_thresh: float = 0.5):
     """
     Returns:
@@ -687,28 +716,22 @@ if __name__ == "__main__":
     print(f"📍 Device: {DEVICE}")
     print(f"📦 Classifier Path: {huggingface_model_path or DEFAULT_AI_DETECTION_MODEL_PATH}")
     print(f"🛠️ Damage Model Path: {DEFAULT_DAMAGE_MODEL_PATH} ({'exists' if os.path.exists(DEFAULT_DAMAGE_MODEL_PATH) else 'missing'})")
+  
+    # Check if dependencies are installed
+    auto_install_dependencies()
 
-    # Preload RADIO model (Stage 2)
-    models_loaded = preload_models()
+    # Preload C model at startup
+    preload_models()
 
     # Load classifier
     model_path = huggingface_model_path or DEFAULT_AI_DETECTION_MODEL_PATH
     classifier_loaded = load_ai_detection_classifier(model_path) is not None
 
     print("=" * 60)
-    print(f"✅ RADIO Model: {'Loaded' if models_loaded else 'Not Available (Demo Mode)'}")
-    print(f"✅ Classifier: {'Loaded' if classifier_loaded else 'Not Available (Demo Mode)'}")
-    print(f"✅ Detectron2: {'Available' if DETECTRON2_AVAILABLE else 'Not Available'}")
     print("=" * 60)
 
-    if not models_loaded or not classifier_loaded:
-        print("⚠️ Running in DEMO MODE for Stage 2 (RADIO/classifier missing)")
-        print("   To enable full functionality, ensure:")
-        print("   1) MODEL_REPO environment variable is set")
-        print("   2) V1.pkl classifier is available (or set PRIVATE_REPO + HF token)")
 
-    if not DETECTRON2_AVAILABLE or not os.path.exists(DEFAULT_DAMAGE_MODEL_PATH):
-        print("ℹ️ Stage 1 will use simulator unless Detectron2 and model_final.pth are present.")
+
 
     app = create_gradio_interface()
     app.launch(