Update app.py

app.py

@@ -639,8 +639,8 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
         )
 
         # # export ply model
-        # print(dump_model_path)
-        # gs_model_list[0].save_ply(dump_model_path)
+        print(dump_model_path)
+        gs_model_list[0].save_ply(dump_model_path)
 
         # rendering !!!!
         start_time = time.time()
@@ -727,8 +727,8 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
         )
 
 
-
-        return dump_image_path, dump_video_path
+        return dump_image_path, dump_video_path, dump_model_path
+        # return dump_image_path, dump_video_path
         # return rgb, dump_image_path, dump_video_path
 
     # def core_fn_export(image, video_params, working_dir):
@@ -859,11 +859,11 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
 
         # show video && ply model
         with gr.Row():
-        #     with gr.Column(variant='panel', scale=1):
-        #         with gr.Tabs(elem_id="openlrm_render_model"):
-        #             with gr.TabItem('Rendered 3D Model'):
-        #                 with gr.Row():
-        #                     output_model = gr.Model3D(label="Rendered 3D Model")
+            with gr.Column(variant='panel', scale=1):
+                with gr.Tabs(elem_id="openlrm_render_model"):
+                    with gr.TabItem('Rendered 3D Model'):
+                        with gr.Row():
+                            output_model = gr.Model3D(label="Rendered 3D Model")
 
             with gr.Column(variant='panel', scale=1):
                 with gr.Tabs(elem_id="openlrm_render_video"):
@@ -871,6 +871,16 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
                         with gr.Row():
                             output_video = gr.Video(label="Rendered Video", format="mp4", height=480, width=270, autoplay=True)
 
+        with gr.Row():
+            download_btn = gr.Button("Download 3D Model")
+            model_download_link = gr.File(label="3D Model File", interactive=False)
+        
+        download_btn.click(
+            fn=lambda x: x if x else None,
+            inputs=[output_model],
+            outputs=[model_download_link]
+        )
+
         working_dir = gr.State()
         submit.click(
             fn=assert_input_image,
@@ -883,7 +893,7 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
         ).success(
             fn=core_fn,
             inputs=[input_image, video_input, working_dir], # video_params refer to smpl dir
-            outputs=[processed_image, output_video],
+            outputs=[processed_image, output_video, output_model],
         )
 
         demo.queue(max_size=1)