V1.1

app.py

@@ -17,6 +17,21 @@ import sys
 import time
 import numpy as np
 import gradio as gr
+# OpenCV import - wrapped in try-except to make it optional
+import cv2
+
+from torchvision import transforms
+from PIL import Image
+# Detectron2 imports - wrapped in try-except to make them optional
+
+from detectron2.engine import DefaultPredictor
+from detectron2.config import get_cfg
+from detectron2.utils.visualizer import Visualizer, ColorMode
+from detectron2 import model_zoo
+
+from configs.get_config import load_config
+from models import *
+
 
 # Status flags for optional dependencies
 CV2_AVAILABLE = False
@@ -28,41 +43,7 @@ MODELS_IMPORTED = False
 if not os.getcwd() in sys.path:
     sys.path.append(os.getcwd())
 
-# OpenCV import - wrapped in try-except to make it optional
-try:
-    import cv2
-    CV2_AVAILABLE = True
-except ImportError:
-    print("Warning: OpenCV (cv2) is not installed. Using demo mode.")
-
-# PyTorch imports - wrapped in try-except to make them optional
-try:
-    import torch
-    from torchvision import transforms
-    from PIL import Image
-    TORCH_AVAILABLE = True
-except ImportError:
-    print("Warning: PyTorch or related libraries are not installed. Using demo mode.")
-
-# Detectron2 imports - wrapped in try-except to make them optional
-if TORCH_AVAILABLE and CV2_AVAILABLE:
-    try:
-        from detectron2.engine import DefaultPredictor
-        from detectron2.config import get_cfg
-        from detectron2.utils.visualizer import Visualizer, ColorMode
-        from detectron2 import model_zoo
-        DETECTRON2_AVAILABLE = True
-    except ImportError:
-        print("Warning: Detectron2 is not installed. Damage detection will not be available.")
 
-# Check for custom path for models if all required dependencies are available
-if TORCH_AVAILABLE and CV2_AVAILABLE:
-    try:
-        from configs.get_config import load_config
-        from models import *
-        MODELS_IMPORTED = True
-    except ImportError:
-        print("Warning: Custom models couldn't be imported. Only damage detection will work if available.")
 
 def check_model_files(damage_model_path, deepfake_model_path, deepfake_cfg_path):
     """Check if required model files exist and return status"""