Update app.py

addressing a error

app.py

@@ -16,26 +16,22 @@ from trellis.pipelines import TrellisImageTo3DPipeline
 from trellis.representations import Gaussian, MeshExtractResult
 from trellis.utils import render_utils, postprocessing_utils
 
-
 # Constants
 MAX_SEED = np.iinfo(np.int32).max
 TMP_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'tmp')
 os.makedirs(TMP_DIR, exist_ok=True)
 
-
 # Session Management Functions
 def start_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     print(f'Creating user directory: {user_dir}')
     os.makedirs(user_dir, exist_ok=True)
 
-
 def end_session(req: gr.Request):
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     print(f'Removing user directory: {user_dir}')
     shutil.rmtree(user_dir)
 
-
 # Image Preprocessing Function
 def preprocess_image(image: Image.Image) -> Image.Image:
     """
@@ -50,7 +46,6 @@ def preprocess_image(image: Image.Image) -> Image.Image:
     processed_image = pipeline.preprocess_image(image)
     return processed_image
 
-
 # State Packing and Unpacking Functions
 def pack_state(gs: Gaussian, mesh: MeshExtractResult, trial_id: str) -> dict:
     return {
@@ -69,7 +64,6 @@ def pack_state(gs: Gaussian, mesh: MeshExtractResult, trial_id: str) -> dict:
         'trial_id': trial_id,
     }
 
-
 def unpack_state(state: dict) -> Tuple[Gaussian, edict, str]:
     gs = Gaussian(
         aabb=state['gaussian']['aabb'],
@@ -92,7 +86,6 @@ def unpack_state(state: dict) -> Tuple[Gaussian, edict, str]:
     
     return gs, mesh, state['trial_id']
 
-
 # Seed Management Function
 def get_seed(randomize_seed: bool, seed: int) -> int:
     """
@@ -107,7 +100,6 @@ def get_seed(randomize_seed: bool, seed: int) -> int:
     """
     return np.random.randint(0, MAX_SEED) if randomize_seed else seed
 
-
 # Core 3D Generation Function
 @spaces.GPU
 def image_to_3d(
@@ -159,7 +151,6 @@ def image_to_3d(
     torch.cuda.empty_cache()
     return state, video_path
 
-
 # Existing GLB Extraction Function
 @spaces.GPU
 def extract_glb(
@@ -188,7 +179,6 @@ def extract_glb(
     torch.cuda.empty_cache()
     return glb_path, glb_path
 
-
 # New High-Quality GLB Extraction Function
 @spaces.GPU
 def extract_glb_high_quality(
@@ -215,7 +205,6 @@ def extract_glb_high_quality(
     torch.cuda.empty_cache()
     return glb_path, glb_path
 
-
 # Gradio Interface Definition
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
@@ -325,14 +314,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             # Existing Download GLB Button
             download_glb = gr.DownloadButton(
                 label="Download GLB",
-                file_name="model.glb",
-                interactive=False
+                interactive=False  # Initially disabled
             )
             # New Download High Quality GLB Button
             download_high_quality_glb = gr.DownloadButton(
                 label="Download High Quality GLB",
-                file_name="model_high_quality.glb",
-                interactive=False
+                interactive=False  # Initially disabled
             )
 
     # State Variables
@@ -382,7 +369,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         ],
         outputs=[output_buf, video_output],
     ).then(
-        lambda: gr.Button.update(interactive=True),
+        lambda: (gr.Button.update(interactive=True), gr.Button.update(interactive=True)),
         outputs=[extract_glb_btn, extract_glb_high_quality_btn],
     )
 
@@ -392,7 +379,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.update(interactive=True),
+        lambda: gr.DownloadButton.update(interactive=True),
         outputs=[download_glb],
     )
 
@@ -402,11 +389,11 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         inputs=[output_buf],
         outputs=[model_output, glb_high_quality_path_state],
     ).then(
-        lambda glb_path: {"value": glb_path} if glb_path else None,
+        lambda glb_path: glb_path,  # Pass the file path directly
         inputs=[glb_high_quality_path_state],
         outputs=[download_high_quality_glb],
     ).then(
-        lambda: gr.Button.update(interactive=True),
+        lambda: gr.DownloadButton.update(interactive=True),
         outputs=[download_high_quality_glb],
     )
 
@@ -422,7 +409,6 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         outputs=[download_glb, download_high_quality_glb],
     )
 
-
 # Launch the Gradio app
 if __name__ == "__main__":
     # Initialize the pipeline