Update app.py

still working on the buttons

app.py

@@ -110,9 +110,9 @@ def image_to_3d(
     slat_guidance_strength: float,
     slat_sampling_steps: int,
     req: gr.Request,
-) -> Tuple[dict, str]:
+) -> Tuple[dict, str, str]:
     """
-    Convert an image to a 3D model.
+    Convert an image to a 3D model and generate a GLB file.
 
     Args:
         image (Image.Image): The input image.
@@ -124,7 +124,7 @@ def image_to_3d(
         req (gr.Request): Gradio request object.
 
     Returns:
-        Tuple[dict, str]: The state dictionary and the path to the generated video.
+        Tuple[dict, str, str]: The state dictionary, path to the generated video, and path to the standard GLB file.
     """
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     outputs = pipeline.run(
@@ -148,8 +148,20 @@ def image_to_3d(
     video_path = os.path.join(user_dir, f"{trial_id}.mp4")
     imageio.mimsave(video_path, video, fps=15)
     state = pack_state(outputs['gaussian'][0], outputs['mesh'][0], trial_id)
+    
+    # Generate standard GLB file with default simplification and texture size
+    glb = postprocessing_utils.to_glb(
+        outputs['gaussian'][0],
+        outputs['mesh'][0],
+        simplify=0.95,          # Default simplification
+        texture_size=1024,      # Default texture size
+        verbose=False
+    )
+    glb_path = os.path.join(user_dir, f"{trial_id}.glb")
+    glb.export(glb_path)
+    
     torch.cuda.empty_cache()
-    return state, video_path
+    return state, video_path, glb_path
 
 # Existing GLB Extraction Function
 @spaces.GPU
@@ -173,7 +185,13 @@ def extract_glb(
     """
     user_dir = os.path.join(TMP_DIR, str(req.session_hash))
     gs, mesh, trial_id = unpack_state(state)
-    glb = postprocessing_utils.to_glb(gs, mesh, simplify=mesh_simplify, texture_size=texture_size, verbose=False)
+    glb = postprocessing_utils.to_glb(
+        gs,
+        mesh,
+        simplify=mesh_simplify,
+        texture_size=texture_size,
+        verbose=False
+    )
     glb_path = os.path.join(user_dir, f"{trial_id}.glb")
     glb.export(glb_path)
     # Read the GLB file as bytes
@@ -202,7 +220,13 @@ def extract_glb_high_quality(
     gs, mesh, trial_id = unpack_state(state)
     # Set simplify to 0.0 to disable polygon reduction
     # Set texture_size to 2048 for maximum texture quality
-    glb = postprocessing_utils.to_glb(gs, mesh, simplify=0.0, texture_size=2048, verbose=False)
+    glb = postprocessing_utils.to_glb(
+        gs,
+        mesh,
+        simplify=0.0,
+        texture_size=2048,
+        verbose=False
+    )
     glb_path = os.path.join(user_dir, f"{trial_id}_high_quality.glb")
     glb.export(glb_path)
     # Read the GLB file as bytes
@@ -215,11 +239,12 @@ def extract_glb_high_quality(
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
     ## Image to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
-    * Upload an image and click "Generate" to create a 3D asset. If the image has an alpha channel, it will be used as the mask. Otherwise, the background will be removed automatically.
-    * If you find the generated 3D asset satisfactory, click "Extract GLB" to extract the GLB file and download it.
-    * **New:** Click "Download High Quality GLB" to download the GLB file without any polygon reduction and with maximum texture quality.
+    * **Generate:** Upload an image and click "Generate" to create a 3D asset. If the image has an alpha channel, it will be used as the mask. Otherwise, the background will be removed automatically.
+    * **Extract GLB:** If the generated 3D asset is satisfactory, click "Extract GLB" to extract the GLB file based on your chosen settings and download it.
+    * **Download High Quality GLB:** Click this button to download a high-quality GLB file without any polygon reduction and with maximum texture quality.
+    * **Status:** View messages and feedback about your actions below.
     """)
-
+    
     with gr.Row():
         with gr.Column():
             # Image Input
@@ -297,12 +322,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                     step=512
                 )
             
-            # Existing Extract GLB Button
-            extract_glb_btn = gr.Button("Extract GLB", interactive=False)
+            # Extract GLB Button
+            extract_glb_btn = gr.Button("Extract GLB")
             
-            # New Extract High Quality GLB Button
-            extract_glb_high_quality_btn = gr.Button("Download High Quality GLB", interactive=False)
-
+            # Download High Quality GLB Button
+            download_glb_high_quality_btn = gr.Button("Download High Quality GLB")
+        
         with gr.Column():
             # Video Output
             video_output = gr.Video(
@@ -317,15 +342,20 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                 exposure=20.0,
                 height=300
             )
-            # Existing Download GLB Button
+            # Download GLB Buttons
             download_glb = gr.DownloadButton(
-                label="Download GLB",
-                interactive=False  # Initially disabled
+                label="Download GLB"
             )
-            # New Download High Quality GLB Button
             download_high_quality_glb = gr.DownloadButton(
-                label="Download High Quality GLB",
-                interactive=False  # Initially disabled
+                label="Download High Quality GLB"
+            )
+            
+            # Status Message
+            status_message = gr.Textbox(
+                label="Status",
+                value="Awaiting your action...",
+                interactive=False,
+                lines=2
             )
 
     # State Variables
@@ -373,55 +403,57 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             slat_guidance_strength,
             slat_sampling_steps
         ],
-        outputs=[output_buf, video_output],
+        outputs=[output_buf, video_output, glb_path := gr.State()]
     ).then(
-        # Enable both Extract GLB buttons after generation
-        lambda: (gr.Button.update(interactive=True), gr.Button.update(interactive=True)),
-        outputs=[extract_glb_btn, extract_glb_high_quality_btn],
+        lambda state, video, glb: "Generation successful! You can now extract and download GLB files.",
+        inputs=[output_buf, video_output, glb_path],
+        outputs=[status_message]
     )
 
-    # Existing Extract GLB Button Click Handler
+    # Extract GLB Button Click Handler
     extract_glb_btn.click(
         extract_glb,
         inputs=[output_buf, mesh_simplify, texture_size],
-        outputs=[model_output, glb_bytes_state],
+        outputs=[output_buf, glb_bytes_state],
     ).then(
-        # Map the GLB bytes to the DownloadButton
-        lambda glb_bytes: (glb_bytes, "model.glb"),
-        inputs=[glb_bytes_state],
+        # Map the GLB bytes to the DownloadButton with filename
+        lambda state, glb_bytes: (glb_bytes, "model.glb"),
+        inputs=[output_buf, glb_bytes_state],
         outputs=[download_glb],
     ).then(
-        # Enable the Download GLB button
-        lambda: gr.DownloadButton.update(interactive=True),
-        outputs=[download_glb],
+        # Update status message
+        lambda: "GLB extraction successful! Click 'Download GLB' to save your model.",
+        inputs=None,
+        outputs=[status_message]
     )
 
-    # New Extract High Quality GLB Button Click Handler
-    extract_glb_high_quality_btn.click(
+    # Download High Quality GLB Button Click Handler
+    download_glb_high_quality_btn.click(
         extract_glb_high_quality,
         inputs=[output_buf],
-        outputs=[model_output, glb_high_quality_bytes_state],
+        outputs=[output_buf, glb_high_quality_bytes_state],
     ).then(
-        # Map the high-quality GLB bytes to the DownloadButton
-        lambda glb_bytes: (glb_bytes, "model_high_quality.glb"),
-        inputs=[glb_high_quality_bytes_state],
+        # Map the high-quality GLB bytes to the DownloadButton with filename
+        lambda state, glb_bytes: (glb_bytes, "model_high_quality.glb"),
+        inputs=[output_buf, glb_high_quality_bytes_state],
         outputs=[download_high_quality_glb],
     ).then(
-        # Enable the Download High Quality GLB button
-        lambda: gr.DownloadButton.update(interactive=True),
-        outputs=[download_high_quality_glb],
+        # Update status message
+        lambda: "High-quality GLB extraction successful! Click 'Download High Quality GLB' to save your model.",
+        inputs=None,
+        outputs=[status_message]
     )
 
     # Handle Clearing of Video Output
     video_output.clear(
-        lambda: (gr.Button.update(interactive=False), gr.Button.update(interactive=False)),
-        outputs=[extract_glb_btn, extract_glb_high_quality_btn],
+        lambda: "Video output cleared. Please generate a new 3D asset.",
+        outputs=[status_message],
     )
 
     # Handle Clearing of Model Output
     model_output.clear(
-        lambda: (gr.File.update(value=None), gr.File.update(value=None)),
-        outputs=[download_glb, download_high_quality_glb],
+        lambda: "Model output cleared. Please extract and download GLB files again.",
+        outputs=[status_message],
     )
 
 # Launch the Gradio app