Update app.py

app.py

@@ -37,39 +37,58 @@ def install_dependencies():
 
     # 1. Basic PIP Upgrade
     env = os.environ.copy()
-    # specific links for torch/cuda compatibility
     env["PIP_EXTRA_INDEX_URL"] = "https://pypi.ngc.nvidia.com https://download.pytorch.org/whl/cu121"
     env["PIP_FIND_LINKS"] = "https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu121.html"
     env["CUDA_HOME"] = "/usr/local/cuda"
-    env["MAX_JOBS"] = "4" # Speed up compilation
+    env["MAX_JOBS"] = "4"
 
     subprocess.run([sys.executable, "-m", "pip", "install", "--upgrade", "pip"], env=env, check=True)
 
     # 2. User Defined Package List
-    # ADDED: hydra-core, omegaconf, and other core ML libs required by sam3d 
-    # because we are using --no-deps later.
     packages = [
-        "hydra-core",       # Required for the patch script
-        "omegaconf",        # Required for hydra
-        "numpy",
-        "einops",           # Required by model
-        "timm",             # Required by model
-        "diffusers",        # Required by model
-        "transformers",     # Required by model
-        "accelerate",       # Required by model
-        "safetensors",      # Required by model
-        "pytest",
-        "pipdeptree",
-        "findpydeps",
-        "lovely_tensors",
-        "seaborn==0.13.2",
-        "gradio==5.49.0",
-        "kaolin==0.17.0",
-        "flash_attn==2.8.3",
+        # Core ML & Hydra
+        "hydra-core", "hydra-submitit-launcher", "omegaconf", "numpy", "einops",
+        "einops-exts", "timm", "diffusers", "transformers", "accelerate", "safetensors",
+        # Testing & Dev Tools
+        "pytest", "pipdeptree", "findpydeps", "lovely_tensors", "autoflake",
+        "black", "flake8", "usort",
+        # Visualization & UI
+        "seaborn", "gradio", "tensorboard", "wandb", "polyscope",
+        # 3D & Graphics
+        "kaolin", "flash_attn", "open3d", "pyrender", "point-cloud-utils",
+        "pymeshfix", "xatlas", "panda3d-gltf", "fvcore", "roma", "smplx",
+        "OpenEXR", "imath",
+        # Video & Audio
+        "av", "decord", "librosa",
+        # CUDA & GPU
+        "cuda-python", "nvidia-cuda-nvcc-cu12", "nvidia-pyindex", "spconv-cu121",
+        "xformers", "torchaudio",
+        # ML Optimization
+        "auto_gptq", "bitsandbytes", "peft", "optimum", "optree", "lightning",
+        "sentence-transformers",
+        # Data & Serialization
+        "h5py", "fastavro", "jsonlines", "jsonpickle", "orjson", "simplejson",
+        "webdataset",
+        # Image Processing
+        "opencv-python", "scikit-image", "pycocotools", "ftfy",
+        # Web & Networking
+        "Flask", "Werkzeug", "hdfs", "httplib2", "PySocks", "gdown",
+        # Utilities
+        "astor", "async-timeout", "colorama", "deprecation", "easydict",
+        "exceptiongroup", "fasteners", "loguru", "objsize", "randomname",
+        "rootutils", "Rtree", "tomli",
+        # JSON Schema & URI
+        "fqdn", "isoduration", "jsonpointer", "uri-template", "webcolors",
+        # Graph & Docs
+        "igraph", "pydot", "pdoc3",
+        # Jupyter & Misc
+        "jupyter", "dataclasses", "crcmod", "conda-pack", "pip-system-certs",
+        "python-pycg", "pymongo", "sagemaker", "mosaicml-streaming", "bpy",
         # Git installs
-        "git+https://github.com/nerfstudio-project/gsplat.git@2323de5905d5e90e035f792fe65bad0fedd413e7",
+        "git+https://github.com/nerfstudio-project/gsplat.git",
         "git+https://github.com/facebookresearch/pytorch3d",
-        "utils3d"
+        "git+https://github.com/microsoft/MoGe.git",
+        "utils3d",
     ]
 
     for pkg in packages:
@@ -77,7 +96,6 @@ def install_dependencies():
         print(f"Installing: {pkg}")
         print(f"----------------------------------------")
         try:
-            # We use --no-build-isolation for git repos sometimes to pick up current environment packages
             cmd = [sys.executable, "-m", "pip", "install", pkg]
             subprocess.run(cmd, env=env, check=True)
         except subprocess.CalledProcessError as e:
@@ -89,9 +107,8 @@ def install_dependencies():
         subprocess.run(["git", "clone", REPO_URL, REPO_DIR], check=True)
     
     os.chdir(REPO_DIR)
-    patch_pyproject_toml() # Strip bad dependencies
+    patch_pyproject_toml()
 
-    # Install the repo itself (excluding the faulty dev deps we just manually handled)
     print("Installing sam-3d-objects in editable mode...")
     subprocess.run([sys.executable, "-m", "pip", "install", "--no-deps", "-e", "."], env=env, check=True)
 
@@ -100,7 +117,6 @@ def install_dependencies():
     if os.path.exists(patch_script):
         print("Applying Hydra patch...")
         subprocess.run(["chmod", "+x", patch_script], check=True)
-        # Now this should work because hydra-core is installed
         subprocess.run([patch_script], check=True)
 
 # Run installation
@@ -199,7 +215,6 @@ def lazy_import_sam3d():
         except ImportError as e:
             print(f"Failed to import SAM 3D modules: {e}")
             print("Ensure the installation step completed successfully.")
-            # Helpful debugging for missing deps
             subprocess.run([sys.executable, "-m", "pipdeptree"])
             raise
 
@@ -226,7 +241,6 @@ def run_inference(image: np.ndarray, mask: np.ndarray, config_file: str, seed: O
     """GPU-decorated inference function for ZeroGPU."""
     global _pipeline
     
-    # Ensure pipeline is loaded
     _pipeline = load_pipeline(config_file)
     if hasattr(_pipeline, 'to'):
         _pipeline.to('cuda')