fixed new model weights loading

perception_roi_server.py

@@ -34,6 +34,7 @@ from fastapi.responses import FileResponse, StreamingResponse, JSONResponse
 from ultralytics import YOLO
 
 DEFAULT_WEIGHTS = os.environ.get("YOLO_WEIGHTS", "yolov8s.pt")
+WEIGHTS_DIR = os.environ.get("WEIGHTS_DIR", "")
 DEFAULT_CONF = float(os.environ.get("YOLO_CONF", "0.25"))
 DEFAULT_DEVICE = os.environ.get("YOLO_DEVICE", "auto")
 FAST_DETECT_SCALE = float(os.environ.get("FAST_DETECT_SCALE", "0.65"))
@@ -58,8 +59,26 @@ def root():
 _model_lock = threading.Lock()
 _models: Dict[str, YOLO] = {}
 
+def _resolve_weights_path(weights: str) -> str:
+    if not weights:
+        return DEFAULT_WEIGHTS
+    w = str(weights)
+    if os.path.isabs(w) or os.path.exists(w):
+        return w
+    # Try configured weights dir first
+    if WEIGHTS_DIR:
+        cand = os.path.join(WEIGHTS_DIR, w)
+        if os.path.exists(cand):
+            return cand
+    # Try common local directories
+    for base in (os.getcwd(), os.path.dirname(__file__), DATA_DIR, "/app", "/workspace", "/data"):
+        cand = os.path.join(base, w)
+        if os.path.exists(cand):
+            return cand
+    return w
+
 def get_model(weights: str) -> YOLO:
-    key = str(weights or DEFAULT_WEIGHTS)
+    key = _resolve_weights_path(weights or DEFAULT_WEIGHTS)
     with _model_lock:
         if key not in _models:
             _models[key] = YOLO(key)