Update app.py

app.py

@@ -7,6 +7,23 @@ import shutil
 import subprocess
 import sys
 
+# Install local wheels at runtime
+def install_local_wheels():
+    """Install the local wheel files that couldn't be installed during Docker build."""
+    wheels_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'wheels')
+    if os.path.exists(wheels_dir):
+        wheel_files = [f for f in os.listdir(wheels_dir) if f.endswith('.whl')]
+        for wheel_file in wheel_files:
+            wheel_path = os.path.join(wheels_dir, wheel_file)
+            try:
+                subprocess.check_call([sys.executable, '-m', 'pip', 'install', wheel_path])
+                print(f"Successfully installed {wheel_file}")
+            except subprocess.CalledProcessError as e:
+                print(f"Failed to install {wheel_file}: {e}")
+
+# Install wheels before importing trellis
+install_local_wheels()
+
 os.environ['SPCONV_ALGO'] = 'native'
 from typing import *
 import torch
@@ -257,11 +274,11 @@ def split_image(image: Image.Image) -> List[Image.Image]:
 
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
-    ## Image to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
+    ## ASM - Advanced Spatial Modeling for 3D Generation
     * Upload an image and click "Generate" to create a 3D asset. If the image has alpha channel, it be used as the mask. Otherwise, we use `rembg` to remove the background.
     * If you find the generated 3D asset satisfactory, click "Extract GLB" to extract the GLB file and download it.
     
-    ✨New: 1) Experimental multi-image support. 2) Gaussian file extraction.
+    ✨Features: 1) Multi-image support. 2) Gaussian file extraction. 3) Advanced 3D generation.
     """)
     
     with gr.Row():
@@ -342,11 +359,11 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
     demo.unload(end_session)
     
     single_image_input_tab.select(
-        lambda: tuple([False, gr.Row.update(visible=True), gr.Row.update(visible=False)]),
+        lambda: tuple([False, gr.update(visible=True), gr.update(visible=False)]),
         outputs=[is_multiimage, single_image_example, multiimage_example]
     )
     multiimage_input_tab.select(
-        lambda: tuple([True, gr.Row.update(visible=False), gr.Row.update(visible=True)]),
+        lambda: tuple([True, gr.update(visible=False), gr.update(visible=True)]),
         outputs=[is_multiimage, single_image_example, multiimage_example]
     )
     
@@ -370,12 +387,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         inputs=[image_prompt, multiimage_prompt, is_multiimage, seed, ss_guidance_strength, ss_sampling_steps, slat_guidance_strength, slat_sampling_steps, multiimage_algo],
         outputs=[output_buf, video_output],
     ).then(
-        lambda: tuple([gr.Button(interactive=True), gr.Button(interactive=True)]),
+        lambda: tuple([gr.update(interactive=True), gr.update(interactive=True)]),
         outputs=[extract_glb_btn, extract_gs_btn],
     )
 
     video_output.clear(
-        lambda: tuple([gr.Button(interactive=False), gr.Button(interactive=False)]),
+        lambda: tuple([gr.update(interactive=False), gr.update(interactive=False)]),
         outputs=[extract_glb_btn, extract_gs_btn],
     )
 
@@ -384,7 +401,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         inputs=[output_buf, mesh_simplify, texture_size],
         outputs=[model_output, download_glb],
     ).then(
-        lambda: gr.Button(interactive=True),
+        lambda: gr.update(interactive=True),
         outputs=[download_glb],
     )
     
@@ -393,12 +410,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         inputs=[output_buf],
         outputs=[model_output, download_gs],
     ).then(
-        lambda: gr.Button(interactive=True),
+        lambda: gr.update(interactive=True),
         outputs=[download_gs],
     )
 
     model_output.clear(
-        lambda: gr.Button(interactive=False),
+        lambda: gr.update(interactive=False),
         outputs=[download_glb],
     )