תיקון שגיאה

backend.py

@@ -8,7 +8,7 @@ import cv2
 from PIL import Image, ImageFilter
 from scipy.ndimage import binary_dilation
 import hydra
-from hydra import initialize, compose
+from hydra import initialize_config_dir, compose
 from omegaconf import OmegaConf
 
 # -----------------------------
@@ -287,10 +287,27 @@ def process_image(
             women_boxes.append(bbox)
 
     # 3) שלב SAM2 (עבור בוקסים של נשים)
+    # אתחול Hydra בתוך הפונקציה
+    global sam2_predictor
+    if sam2_predictor is None:
+        try:
+            config_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), SAM2_CONFIG_PATH))
+            with initialize_config_dir(version_base=None, config_dir=config_dir):
+                cfg = compose(config_name="config") # "config" is the name of the config file, replace if necessary
+                sam2_predictor = SAM2ImagePredictor.from_pretrained(SAM2_MODEL_NAME, config=cfg)
+                sam2_predictor.model.to(device)
+        except Exception as e:
+            print(f"[SAM2] שגיאה בטעינת SAM2: {e}")
+            print(f"  - סוג השגיאה: {type(e).__name__}")
+            print(f"  - הודעת השגיאה: {e}")
+            import traceback
+            print(f"  - Traceback:")
+            traceback.print_exc()
+            sam2_predictor = None
+
     if sam2_predictor is None:
         print("[process_image] SAM2 לא זמין/נטען. מחזירים תמונה ללא טשטוש.")
-        raise ValueError("SAM2 model is not loaded.")  # הוספת זריקת שגיאה
-        return pil_image
+        raise ValueError("SAM2 model is not loaded.")
 
     progress_callback(0.6, f"מתחיל פילוח SAM2 על {len(women_boxes)} נשים...")
     sam2_predictor.set_image(np.array(pil_image))