Update app.py

app.py

@@ -98,7 +98,7 @@ bg_remover = rembg.new_session()
 
 # process function
 # @spaces.GPU
-def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_steps=30, input_seed=42):
+def generate(input_image, prompt, prompt_neg='', input_elevation=0, input_num_steps=30, input_seed=42):
 
     # seed
     kiui.seed_everything(input_seed)
@@ -193,7 +193,12 @@ def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_ste
         images = np.concatenate(images, axis=0)
         imageio.mimwrite(output_video_path, images, fps=30)
 
+    return mv_image_grid, output_video_path, output_ply_path
 
+def convert(output_ply_path):
+    if not os.path.exists(output_ply_path):
+        gr.Warning("PLY file not found please upload or generate 3D model")
+        
     # load a saved ply and convert to mesh
     opt.test_path = output_ply_path
     opt.force_cuda_rast = True # container compatability
@@ -203,9 +208,8 @@ def process(input_image, prompt, prompt_neg='', input_elevation=0, input_num_ste
     converter.fit_mesh()
     converter.fit_mesh_uv()
     converter.export_mesh(opt.test_path.replace('.ply', '.glb'))
-
-    return mv_image_grid, output_video_path, output_glb_path
-
+    return output_glb_path
+    
 # gradio UI
 
 _TITLE = '''LGM: Large Multi-View Gaussian Model for High-Resolution 3D Content Creation'''
@@ -247,7 +251,7 @@ with block:
 
 
         with gr.Column(scale=1):
-            with gr.Tab("Video"):
+            with gr.Tab("Video and File"):
                 # final video results
                 output_video = gr.Video(label="video")
                 # ply file
@@ -255,9 +259,14 @@ with block:
             with gr.Tab("Multi-view Image"):
                 # multi-view results
                 output_image = gr.Image(interactive=False, show_label=False)
+            with gr.Tab("Convert Format"):
+                # multi-view results
+                output_glb = gr.File(label="3D Mesh (glb format)")
+                convert_button = gr.Button("Convert to Mesh")
 
-        button_gen.click(process, inputs=[input_image, input_text, input_neg_text, input_elevation, input_num_steps, input_seed], outputs=[output_image, output_video, output_file])
-
+        button_gen.click(generate, inputs=[input_image, input_text, input_neg_text, input_elevation, input_num_steps, input_seed], outputs=[output_image, output_video, output_file])
+        button_con.click(convert, inputs=output_file, outputs=output_glb)
+    
     gr.Examples(
         examples=[
             "data_test/frog_sweater.jpg",
@@ -269,7 +278,7 @@ with block:
         ],
         inputs=[input_image],
         outputs=[output_image, output_video, output_file],
-        fn=lambda x: process(input_image=x, prompt=''),
+        fn=lambda x: generate(input_image=x, prompt=''),
         cache_examples=True,
         label='Image-to-3D Examples'
     )
@@ -286,7 +295,7 @@ with block:
         ],
         inputs=[input_text],
         outputs=[output_image, output_video, output_file],
-        fn=lambda x: process(input_image=None, prompt=x),
+        fn=lambda x: generate(input_image=None, prompt=x),
         cache_examples=True,
         label='Text-to-3D Examples'
     )