update interface

app.py

@@ -424,7 +424,10 @@ with gr.Blocks(css="""
                     output_video1 = gr.Video(label="Generated Video", interactive=False, loop=True, autoplay=True)
                     btn_download1 = gr.Button("Download Mesh")
 
-                    file_output1 = gr.File()
+
+                    download_1 = gr.DownloadButton(label="Download mesh", interactive=False)
+
+                    # file_output1 = gr.File()
                     
         with gr.TabItem('Image-to-3D', id='tab_image_to_3d'):
             with gr.Row():
@@ -437,37 +440,45 @@ with gr.Blocks(css="""
 
                     image_caption = gr.Textbox(value="", label="Image Caption", placeholder="caption will be generated here base on your input image. You can also edit this caption", lines=4, interactive=True)
                     
-                    output_image2 = gr.Image(label="Generated image", interactive=False)
-                    strength1 = gr.Slider(minimum=0, maximum=1.0, step=0.01, value=0.5, label="strength1")
-                    strength2 = gr.Slider(minimum=0, maximum=1.0, step=0.01, value=0.95, label="strength2")
-                    enable_redux = gr.Checkbox(label="enable redux", value=True)
-                    use_controlnet = gr.Checkbox(label="use controlnet", value=True)
+                    with gr.Accordion(label="Extra Settings", open=False):
+                        output_image2 = gr.Image(label="Generated image", interactive=False)
+                        strength1 = gr.Slider(minimum=0, maximum=1.0, step=0.01, value=0.5, label="redux strength")
+                        strength2 = gr.Slider(minimum=0, maximum=1.0, step=0.01, value=0.95, label="denoise strength")
+                        enable_redux = gr.Checkbox(label="enable redux", value=True)
+                        use_controlnet = gr.Checkbox(label="enable controlnet", value=True)
 
                     btn_img2mesh_main = gr.Button("Generate Mesh")
 
                 with gr.Column(scale=1):
 
                     # output_mesh2 = gr.Model3D(label="Generated Mesh", interactive=False)
-                    output_image3 = gr.Image(label="gen save image", interactive=False)
+                    output_image3 = gr.Image(label="Final Bundle Image", interactive=False)
                     output_video2 = gr.Video(label="Generated Video", interactive=False, loop=True, autoplay=True)
-                    btn_download2 = gr.Button("Download Mesh")
-                    file_output2 = gr.File()
+                    # btn_download2 = gr.Button("Download Mesh")
+                    download_2 = gr.DownloadButton(label="Download mesh", interactive=False)
+                    # file_output2 = gr.File()
 
     # Image2
     btn_img2mesh_preprocess.click(fn=image2mesh_preprocess_, inputs=[image, seed2], outputs=[output_image2, image_caption])
 
-    btn_img2mesh_main.click(fn=image2mesh_main_, inputs=[output_image2, image_caption, seed2, strength1, strength2, enable_redux, use_controlnet], outputs=[output_image3, output_video2])
+    btn_img2mesh_main.click(fn=image2mesh_main_, inputs=[output_image2, image_caption, seed2, strength1, strength2, enable_redux, use_controlnet], outputs=[output_image3, output_video2, download_2]).then(
+        lambda: gr.Button(interactive=True),
+        outputs=[download_2],
+    )
 
 
-    btn_download1.click(fn=save_cached_mesh, inputs=[], outputs=file_output1)
-    btn_download2.click(fn=save_cached_mesh, inputs=[], outputs=file_output2)
+    # btn_download1.click(fn=save_cached_mesh, inputs=[], outputs=file_output1)
+    # btn_download2.click(fn=save_cached_mesh, inputs=[], outputs=file_output2)
 
 
     # Button Click Events
     # Text2
     btn_text2detailed.click(fn=text_to_detailed, inputs=[prompt, seed1], outputs=detailed_prompt)
     btn_text2img.click(fn=text_to_image, inputs=[detailed_prompt, seed1], outputs=output_image1)
-    btn_gen_mesh.click(fn=bundle_image_to_mesh, inputs=[output_image1,], outputs=output_video1)
+    btn_gen_mesh.click(fn=bundle_image_to_mesh, inputs=[output_image1,], outputs=[output_video1, download_1]).then(
+        lambda: gr.Button(interactive=True),
+        outputs=[download_1],
+    )
     # btn_gen_mesh.click(fn=bundle_image_to_mesh, inputs=[output_image1, lrm_radius, isomer_radius, reconstruction_stage1_steps, reconstruction_stage2_steps], outputs=output_video1)
 
     with gr.Row():