Update app.py

app.py

@@ -15,39 +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("Cloning SAM2 repository...")
+        print("Installing SAM2...")
+        os.makedirs("third_party", exist_ok=True)
+        
+        # 克隆仓库
         subprocess.run(["git", "clone", "https://github.com/facebookresearch/sam2.git", sam2_dir], check=True)
-    
-    # 安装依赖
-    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目录安装
+        original_dir = os.getcwd()
+        try:
+            os.chdir(sam2_dir)
+            
+            # 安装依赖
+            subprocess.run(["pip", "install", "-e", "."], check=True)
+            
+            # 切换到checkpoints目录下载模型
+            os.chdir("checkpoints")
+            if not os.path.exists("download_ckpts.sh"):
+                # 如果脚本不存在，可能是权限问题，尝试直接下载
+                subprocess.run(["wget", "https://raw.githubusercontent.com/facebookresearch/sam2/main/checkpoints/download_ckpts.sh"], check=True)
+                subprocess.run(["chmod", "+x", "download_ckpts.sh"], check=True)
+            
+            subprocess.run(["./download_ckpts.sh"], check=True)
+            
+        finally:
+            # 确保切换回原始目录
+            os.chdir(original_dir)
+        
+        print("SAM2 installed successfully!")
+    else:
+        print("SAM2 already installed.")
 
 # 确保安装SAM2
 install_sam2()