Upload gradio_app.py

gradio_app.py

@@ -137,7 +137,21 @@ def build_model_viewer_html(save_folder, height=660, width=790, textured=False):
         template_html = template_html.replace('#src#', f'{related_path}/')
         f.write(template_html)
 
-    mesh_file = os.path.join(save_folder, "white_mesh.html")
+    #iframe_tag = f'<iframe src="file={output_html_path}" height="{height}" width="100%" frameborder="0"></iframe>'
+    #print(
+    #    f'Find html file {output_html_path}, {os.path.exists(output_html_path)}, relative HTML path is /static/{rel_path}')
+
+    #return f"""
+    #    <div style='height: {height}; width: 100%;'>
+    #    {iframe_tag}
+    #    </div>
+    #"""
+
+    if textured:
+        mesh_file = os.path.join(save_folder, "textured_mesh.glb")
+    else: 
+        mesh_file = os.path.join(save_folder, "white_mesh.glb")
+
     # Encode the GLB file as base64 to embed directly in the HTML
     with open(mesh_file, "rb") as f:
         glb_data = f.read()
@@ -403,15 +417,15 @@ def build_app():
         with gr.Row():
             with gr.Column(scale=3):
                 with gr.Tabs(selected='tab_img_prompt') as tabs_prompt:
-                    with gr.Tab('Image Prompt', id='tab_img_prompt', visible=not MV_MODE) as tab_ip:
+                    with gr.Tab('Image Prompt', id='tab_img_prompt', visible=False) as tab_ip:
                         image = gr.Image(label='Image', type='pil', image_mode='RGBA', height=290)
 
-                    with gr.Tab('Text Prompt', id='tab_txt_prompt', visible=HAS_T2I and not MV_MODE) as tab_tp:
+                    with gr.Tab('Text Prompt', id='tab_txt_prompt', visible=False) as tab_tp:
                         caption = gr.Textbox(label='Text Prompt',
                                              placeholder='HunyuanDiT will be used to generate image.',
                                              info='Example: A 3D model of a cute cat, white background')
 
-                    with gr.Tab('MultiView Prompt', visible=MV_MODE) as tab_mv:
+                    with gr.Tab('MultiView Prompt', visible=True) as tab_mv:
                         # gr.Label('Please upload at least one front image.')
                         with gr.Row():
                             mv_image_front = gr.Image(label='Front', type='pil', image_mode='RGBA', height=140,
@@ -480,7 +494,7 @@ def build_app():
                             file_export = gr.DownloadButton(label="Download", variant='primary',
                                                             interactive=False, min_width=100)
 
-            with gr.Column(scale=6):
+            with gr.Column(scale=9):
                 with gr.Tabs(selected='gen_mesh_panel') as tabs_output:
                     with gr.Tab('Generated Mesh', id='gen_mesh_panel'):
                         html_gen_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
@@ -489,22 +503,6 @@ def build_app():
                     with gr.Tab('Mesh Statistic', id='stats_panel'):
                         stats = gr.Json({}, label='Mesh Stats')
 
-            with gr.Column(scale=3 if MV_MODE else 2):
-                with gr.Tabs(selected='tab_img_gallery') as gallery:
-                    with gr.Tab('Image to 3D Gallery', id='tab_img_gallery', visible=not MV_MODE) as tab_gi:
-                        with gr.Row():
-                            gr.Examples(examples=example_is, inputs=[image],
-                                        label=None, examples_per_page=18)
-
-                    with gr.Tab('Text to 3D Gallery', id='tab_txt_gallery', visible=HAS_T2I and not MV_MODE) as tab_gt:
-                        with gr.Row():
-                            gr.Examples(examples=example_ts, inputs=[caption],
-                                        label=None, examples_per_page=18)
-                    with gr.Tab('MultiView to 3D Gallery', id='tab_mv_gallery', visible=MV_MODE) as tab_mv:
-                        with gr.Row():
-                            gr.Examples(examples=example_mvs,
-                                        inputs=[mv_image_front, mv_image_back, mv_image_left, mv_image_right],
-                                        label=None, examples_per_page=6)
 
         tab_ip.select(fn=lambda: gr.update(selected='tab_img_gallery'), outputs=gallery)
         if HAS_T2I:
@@ -739,4 +737,4 @@ if __name__ == '__main__':
     if args.low_vram_mode:
         torch.cuda.empty_cache()
     demo = build_app()
-    demo.launch()
+    demo.launch(share=True)