Update app.py

app.py

@@ -46,6 +46,8 @@ def install_dependencies():
 
     # 2. User Defined Package List
     packages = [
+        # MUST INSTALL FIRST: PyTorch (required for building flash_attn, pytorch3d, etc.)
+        "torch", "torchvision",
         # Core ML & Hydra
         "hydra-core", "hydra-submitit-launcher", "omegaconf", "numpy", "einops",
         "einops-exts", "timm", "diffusers", "transformers", "accelerate", "safetensors",
@@ -55,7 +57,7 @@ def install_dependencies():
         # Visualization & UI
         "seaborn", "gradio", "tensorboard", "wandb", "polyscope",
         # 3D & Graphics
-        "kaolin", "flash_attn", "open3d", "pyrender", "point-cloud-utils",
+        "open3d", "pyrender", "point-cloud-utils",
         "pymeshfix", "xatlas", "panda3d-gltf", "fvcore", "roma", "smplx",
         "OpenEXR", "imath",
         # Video & Audio
@@ -84,13 +86,20 @@ def install_dependencies():
         # Jupyter & Misc
         "jupyter", "dataclasses", "crcmod", "conda-pack", "pip-system-certs",
         "python-pycg", "pymongo", "sagemaker", "mosaicml-streaming", "bpy",
-        # Git installs
+        # Git installs (pytorch3d requires torch, so it's near the end)
         "git+https://github.com/nerfstudio-project/gsplat.git",
-        "git+https://github.com/facebookresearch/pytorch3d",
+        "git+https://github.com/facebookresearch/pytorch3d.git",
         "git+https://github.com/microsoft/MoGe.git",
         "utils3d",
     ]
 
+    # Packages that require special handling - install with --no-build-isolation
+    # so they can find torch during build
+    special_packages = [
+        "flash_attn",
+        "kaolin",
+    ]
+
     for pkg in packages:
         print(f"----------------------------------------")
         print(f"Installing: {pkg}")
@@ -101,6 +110,17 @@ def install_dependencies():
         except subprocess.CalledProcessError as e:
             print(f"❌ Failed to install {pkg}. Continuing to next package...")
 
+    # Install packages that need --no-build-isolation (require torch at build time)
+    for pkg in special_packages:
+        print(f"----------------------------------------")
+        print(f"Installing (no-build-isolation): {pkg}")
+        print(f"----------------------------------------")
+        try:
+            cmd = [sys.executable, "-m", "pip", "install", "--no-build-isolation", pkg]
+            subprocess.run(cmd, env=env, check=True)
+        except subprocess.CalledProcessError as e:
+            print(f"❌ Failed to install {pkg}. Continuing to next package...")
+
     # 3. Clone & Install Main Repo (SAM 3D Objects)
     if not os.path.exists(REPO_DIR):
         print(f"Cloning repository to {REPO_DIR}...")