Update app.py

app.py

@@ -15,23 +15,39 @@ from typing import List, Tuple, Optional
 
 # ------------------ 安装SAM2依赖 ------------------
 def install_sam2():
-    """检查并安装SAM2及其依赖"""
+    """安装SAM2及其依赖，并下载检查点"""
     sam2_dir = "third_party/sam2"
+    checkpoints_dir = os.path.join(sam2_dir, "checkpoints")
+    
+    # 创建目录
+    os.makedirs(checkpoints_dir, exist_ok=True)
+    
+    # 克隆仓库（如果不存在）
     if not os.path.exists(sam2_dir):
-        print("Installing SAM2...")
-        os.makedirs("third_party", exist_ok=True)
+        print("Cloning SAM2 repository...")
         subprocess.run(["git", "clone", "https://github.com/facebookresearch/sam2.git", sam2_dir], check=True)
-        
-        # 安装依赖
-        subprocess.run(["pip", "install", "-e", sam2_dir], check=True)
-        
-        # 下载检查点
-        checkpoints_dir = os.path.join(sam2_dir, "checkpoints")
-        os.makedirs(checkpoints_dir, exist_ok=True)
-        subprocess.run([os.path.join(checkpoints_dir, "download_ckpts.sh")], cwd=checkpoints_dir, shell=True, check=True)
-        print("SAM2 installed successfully!")
-    else:
-        print("SAM2 already installed.")
+    
+    # 安装依赖
+    print("Installing SAM2 dependencies...")
+    subprocess.run(["pip", "install", "-e", sam2_dir], check=True)
+    
+    # 直接下载检查点文件（避免使用脚本）
+    checkpoint_urls = {
+        "sam2.1_hiera_large.pt": "https://dl.fbaipublicfiles.com/sam2/sam2.1_hiera_large.pt",
+        "sam2.1_hiera_base.pt": "https://dl.fbaipublicfiles.com/sam2/sam2.1_hiera_base.pt"
+    }
+    
+    print("Downloading SAM2 checkpoints...")
+    for filename, url in checkpoint_urls.items():
+        filepath = os.path.join(checkpoints_dir, filename)
+        if not os.path.exists(filepath):
+            print(f"Downloading {filename}...")
+            response = requests.get(url, stream=True)
+            with open(filepath, "wb") as f:
+                for chunk in response.iter_content(chunk_size=8192):
+                    f.write(chunk)
+    
+    print("SAM2 installation completed successfully!")
 
 # 确保安装SAM2
 install_sam2()