Update core/process.py

core/process.py

@@ -1,6 +1,41 @@
 import logging
 import time
 import timeout_decorator
+import io
+import zipfile
+import json
+import cv2
+from PIL import Image
+
+from registry import get_model
+from core.describe_scene import describe_scene
+from utils.helpers import generate_session_id, log_runtime
+# Setup logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+# Model mappings
+DETECTION_MODEL_MAP = {
+    "YOLOv5-Nano": "yolov5n-seg",
+    "YOLOv5-Small": "yolov5s-seg",
+    "YOLOv8-Small": "yolov8s",
+    "YOLOv8-Large": "yolov8l",
+    "RT-DETR": "rtdetr"  # For future support
+}
+
+SEGMENTATION_MODEL_MAP = {
+    "SegFormer-B0": "nvidia/segformer-b0-finetuned-ade-512-512",
+    "SegFormer-B5": "nvidia/segformer-b5-finetuned-ade-512-512",
+    "DeepLabV3-ResNet50": "deeplabv3_resnet50"
+}
+
+DEPTH_MODEL_MAP = {
+    "MiDaS v21 Small 256": "midas_v21_small_256",
+    "MiDaS v21 384": "midas_v21_384",
+    "DPT Hybrid 384": "dpt_hybrid_384",
+    "DPT Swin2 Large 384": "dpt_swin2_large_384",
+    "DPT Beit Large 512": "dpt_beit_large_512"
+}
 
 @timeout_decorator.timeout(35, use_signals=False)  # 35 sec limit per image
 def process_image(