Update app.py

app.py

@@ -26,34 +26,64 @@ print("Installing PyTorch3D from source...")
 os.system("pip uninstall -y pytorch3d")
 
 # Install dependencies required for building PyTorch3D
+print("Installing build dependencies...")
 os.system("apt-get update && apt-get install -y git build-essential libglib2.0-0 libsm6 libxrender-dev libxext6 ninja-build")
 os.system("pip install 'imageio>=2.5.0' 'matplotlib>=3.1.2' 'numpy>=1.17.3' 'psutil>=5.6.5' 'scipy>=1.3.2' 'tqdm>=4.42.1' 'trimesh>=3.0.0'")
 os.system("pip install fvcore iopath")
 
 # Clone the PyTorch3D repository
+print("Cloning PyTorch3D repository...")
 os.system("rm -rf pytorch3d")  # Remove any existing directory
-os.system("git clone https://github.com/facebookresearch/pytorch3d.git")
+clone_result = os.system("git clone https://github.com/facebookresearch/pytorch3d.git")
+if clone_result != 0:
+    print("Failed to clone PyTorch3D repository. Trying with git protocol...")
+    clone_result = os.system("git clone git://github.com/facebookresearch/pytorch3d.git")
+    if clone_result != 0:
+        print("Failed to clone PyTorch3D repository with both HTTPS and git protocols.")
 
 # Use a specific release tag that is known to be stable
-os.system("cd pytorch3d && git checkout v0.7.4")
+checkout_result = os.system("cd pytorch3d && git checkout v0.7.4")
+if checkout_result != 0:
+    print("Failed to checkout v0.7.4 tag. Trying with main branch...")
+    checkout_result = os.system("cd pytorch3d && git checkout main")
+    if checkout_result != 0:
+        print("Failed to checkout main branch.")
 
 # Install PyTorch3D from source with CPU support
-os.system("cd pytorch3d && pip install -e .")
+print("Building PyTorch3D from source...")
+build_result = os.system("cd pytorch3d && pip install -v -e .")
+if build_result != 0:
+    print("Failed to build PyTorch3D from source with default settings.")
+    
+    # Try with specific build flags
+    print("Trying with specific build flags...")
+    os.environ["FORCE_CUDA"] = "0"  # Explicitly disable CUDA for build
+    build_result = os.system("cd pytorch3d && pip install -v -e .")
+    
+    if build_result != 0:
+        print("Failed to build PyTorch3D from source with specific build flags.")
+        
+        # Try with setup.py directly
+        print("Trying with setup.py directly...")
+        build_result = os.system("cd pytorch3d && python setup.py install")
+        
+        if build_result != 0:
+            print("All attempts to build from source failed.")
 
 # Verify the installation
-import_result = os.popen('python -c "import pytorch3d; from pytorch3d import renderer; print(\'PyTorch3D and renderer successfully imported\')" 2>&1').read()
+import_result = os.popen('python -c "import pytorch3d; print(\'pytorch3d imported successfully\'); try: from pytorch3d import renderer; print(\'renderer module imported successfully\'); except ImportError as e: print(f\'Error importing renderer: {e}\');" 2>&1').read()
 print(import_result)
 
-# If the installation fails, try a different approach with a specific wheel
-if "No module named" in import_result or "Error" in import_result:
-    print("Source installation failed, trying with a specific wheel...")
+# If the installation fails, try a different approach with wheels from PyPI
+if "Error importing renderer" in import_result or "No module named" in import_result:
+    print("Source installation failed to provide renderer module, trying with PyPI...")
     os.system("pip uninstall -y pytorch3d")
     
-    # Try with a specific wheel that's known to work
-    os.system("pip install https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cpu_pyt201/pytorch3d-0.7.4-cp310-cp310-linux_x86_64.whl")
+    # Try with PyPI version first (which might be CPU-only but should have renderer)
+    os.system("pip install pytorch3d")
     
     # Verify again
-    import_result = os.popen('python -c "import pytorch3d; from pytorch3d import renderer; print(\'PyTorch3D and renderer successfully imported\')" 2>&1').read()
+    import_result = os.popen('python -c "import pytorch3d; print(\'pytorch3d imported successfully\'); try: from pytorch3d import renderer; print(\'renderer module imported successfully\'); except ImportError as e: print(f\'Error importing renderer: {e}\');" 2>&1').read()
     print(import_result)
 
 # Patch the shap_e renderer to handle PyTorch3D renderer import error if needed