update app.py

app.py

@@ -372,6 +372,7 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
 
         dump_video_path = os.path.join(working_dir.name, "output.mp4")
         dump_image_path = os.path.join(working_dir.name, "output.png")
+        dump_model_path = os.path.join(working_dir.name, "output.ply")
 
 
         # prepare dump paths
@@ -513,6 +514,10 @@ 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)
+
         # rendering !!!!
         start_time = time.time()
         batch_dict = dict()
@@ -597,6 +602,8 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
             verbose=True,
         )
 
+
+
         return dump_image_path, dump_video_path
         # return rgb, dump_image_path, dump_video_path
 
@@ -687,7 +694,6 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
                         inputs=[input_image], 
                         examples_per_page=20,
                     )
-
             with gr.Column():
                 with gr.Tabs(elem_id="openlrm_input_video"):
                     with gr.TabItem('Input Video'):
@@ -719,17 +725,24 @@ def demo_lhm(pose_estimator, face_detector, lhm, cfg):
                         with gr.Row():
                             processed_image = gr.Image(label="Processed Image", image_mode="RGBA", type="filepath", elem_id="processed_image", height=480, width=270, interactive=False)
 
-            with gr.Column(variant='panel', scale=1):
-                with gr.Tabs(elem_id="openlrm_render_video"):
-                    with gr.TabItem('Rendered Video'):
-                        with gr.Row():
-                            output_video = gr.Video(label="Rendered Video", format="mp4", height=480, width=270, autoplay=True)
-
         # SETTING
         with gr.Row():
             with gr.Column(variant='panel', scale=1):
                 submit = gr.Button('Generate', elem_id="openlrm_generate", variant='primary')
 
+        # 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_video"):
+                    with gr.TabItem('Rendered Video'):
+                        with gr.Row():
+                            output_video = gr.Video(label="Rendered Video", format="mp4", height=480, width=270, autoplay=True)
 
         working_dir = gr.State()
         submit.click(
@@ -785,6 +798,6 @@ def launch_gradio_app():
     demo_lhm(pose_estimator, facedetector, lhm, cfg)
 
 if __name__ == '__main__':
-    launch_pretrained()
+    # launch_pretrained()
     # launch_env_not_compile_with_cuda()
     launch_gradio_app()