עדכון קצה אחורי

backend.py

@@ -156,21 +156,34 @@ CONF_THRESHOLD = 0.2
 # 4) הכנה ל-SAM2
 # -----------------------------
 try:
-    from hydra import initialize
+    from sam2.build_sam import build_sam2
     from sam2.sam2_image_predictor import SAM2ImagePredictor
 
-    SAM2_CONFIG_PATH = "../../models/sam2.1/"
-    SAM2_MODEL_NAME = "facebook/sam2.1-hiera-tiny"
+    SAM2_CHECKPOINT = "../../checkpoints/sam2.1_hiera_tiny.pt"
+    SAM2_CONFIG = "../../configs/sam2.1/sam2.1_hiera_t.yaml"
+
+    if not os.path.exists(SAM2_CHECKPOINT):
+        print("[SAM2] מודל SAM2 לא נמצא. מנסה להוריד את המודל...")
+        sam2_url = "https://dl.fbaipublicfiles.com/segment_anything_2/092824/sam2.1_hiera_tiny.pt"
+        os.makedirs(os.path.dirname(SAM2_CHECKPOINT), exist_ok=True)
+        response = requests.get(sam2_url)
+        with open(SAM2_CHECKPOINT, 'wb') as f:
+            f.write(response.content)
+        print("[SAM2] מודל SAM2 הורד בהצלחה.")
+
+    if not os.path.exists(SAM2_CONFIG):
+        print("[SAM2] קובץ הקונפיג SAM2 לא נמצא. מנסה להוריד את הקונפיג...")
+        # כאן יש לוודא שקובץ הקונפיג זמין, או לכלול אותו במאגר שלך
+        raise FileNotFoundError(f"לא נמצא קובץ קונפיג SAM2 בנתיב: {SAM2_CONFIG}")
 
-    sam2_predictor = None
     device = "cuda" if torch.cuda.is_available() else "cpu"
 
-    with initialize(config_path=SAM2_CONFIG_PATH):
-        sam2_predictor = SAM2ImagePredictor.from_pretrained(SAM2_MODEL_NAME)
+    sam2_model = build_sam2(SAM2_CONFIG, SAM2_CHECKPOINT, device=device)
+    sam2_predictor = SAM2ImagePredictor(sam2_model)
     sam2_predictor.model.to(device)
 
 except Exception as e:
-    print("[SAM2] לא מצליח לטעון את SAM2. ודא שהנתיב והקונפיג נכונים.")
+    print(f"[SAM2] לא מצליח לטעון את SAM2: {e}")
     sam2_predictor = None
 
 # -----------------------------
@@ -224,6 +237,8 @@ def process_image(
     gemini_api_key: str,
     progress_callback=None
 ) -> Image.Image:
+    if not gemini_api_key:
+        raise ValueError("מפתח API של Gemini אינו מוזן.")
     """
     פונקציה המקבלת תמונת PIL, מפתח API של Gemini, ומחזירה את התמונה לאחר טשטוש נשים,
     תוך שלבי התקדמות מוגדרים:

requirements.txt

@@ -7,4 +7,5 @@ requests
 ultralytics
 scipy
 hydra-core
+torchvision
 git+https://github.com/facebookresearch/sam2.git