Update app.py

still working on making the interface work with the downloads after generation

app.py

@@ -375,6 +375,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         ],
         outputs=[output_buf, video_output],
     ).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],
     )
@@ -385,10 +386,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         inputs=[output_buf, mesh_simplify, texture_size],
         outputs=[model_output, glb_bytes_state],
     ).then(
-        lambda glb_bytes: (glb_bytes, ),
+        # Map the GLB bytes to the DownloadButton
+        lambda glb_bytes: (glb_bytes, "model.glb"),
         inputs=[glb_bytes_state],
         outputs=[download_glb],
     ).then(
+        # Enable the Download GLB button
         lambda: gr.DownloadButton.update(interactive=True),
         outputs=[download_glb],
     )
@@ -399,10 +402,12 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         inputs=[output_buf],
         outputs=[model_output, glb_high_quality_bytes_state],
     ).then(
-        lambda glb_bytes: (glb_bytes, ),
+        # Map the high-quality GLB bytes to the DownloadButton
+        lambda glb_bytes: (glb_bytes, "model_high_quality.glb"),
         inputs=[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],
     )