Upload gradio_app.py

gradio_app.py

@@ -347,11 +347,11 @@ def build_app():
         with gr.Row():
             with gr.Column(scale=3):
                 with gr.Tabs(selected='tab_mv'):
-                    with gr.Tab("Imágenes", id='tab_mv', visible=TURBO_MODE):
+                    with gr.Tab('Imágenes', id='tab_mv'):
                         with gr.Row():
-                            mv_image_front = gr.Image(label='Parte Frontal', type='pil', image_mode='RGBA', height=140,
+                            mv_image_front = gr.Image(label='Frente', type='pil', image_mode='RGBA', height=140,
                                                       min_width=100, elem_classes='mv-image')
-                            mv_image_back = gr.Image(label='Parte Trasera', type='pil', image_mode='RGBA', height=140,
+                            mv_image_back = gr.Image(label='Atrás', type='pil', image_mode='RGBA', height=140,
                                                      min_width=100, elem_classes='mv-image')
                         with gr.Row():
                             mv_image_left = gr.Image(label='Izquierda', type='pil', image_mode='RGBA', height=140,
@@ -360,22 +360,24 @@ def build_app():
                                                       min_width=100, elem_classes='mv-image')
 
                 with gr.Row():
-                    btn = gr.Button(value='Generar Modelo', variant='primary', min_width=100)
+                    btn = gr.Button(value='Generar modelo', variant='primary', min_width=100)
+                    btn_all = gr.Button(value='Gen Textured Shape',
+                                        variant='primary',
+                                        visible=False,
+                                        min_width=100)
 
                 with gr.Group():
                     file_out = gr.File(label="File", visible=False)
                     file_out2 = gr.File(label="File", visible=False)
 
                 with gr.Tabs(selected='tab_options' if TURBO_MODE else 'tab_export'):
-                    with gr.Tab("Options", id='tab_options', visible=TURBO_MODE):
-                        gen_mode = gr.Radio(label='Generation Mode',
-                                            info='Recommendation: Turbo for most cases, Fast for very complex cases, Standard seldom use.',
+                    with gr.Tab("Opciones", id='tab_options', visible=TURBO_MODE):
+                        gen_mode = gr.Radio(label='Modo de Generación',
                                             choices=['Turbo', 'Fast', 'Standard'], value='Turbo')
-                        decode_mode = gr.Radio(label='Decoding Mode',
-                                               info='The resolution for exporting mesh from generated vectset',
+                        decode_mode = gr.Radio(label='Mode de decripción',
                                                choices=['Low', 'Standard', 'High'],
                                                value='Standard')
-                    with gr.Tab('Advanced Options', id='tab_advanced_options'):
+                    with gr.Tab('Advanced Options', id='tab_advanced_options', visible=False):
                         with gr.Row():
                             check_box_rembg = gr.Checkbox(value=True, label='Remove Background', min_width=100)
                             randomize_seed = gr.Checkbox(label="Randomize seed", value=True, min_width=100)
@@ -397,28 +399,28 @@ def build_app():
                             cfg_scale = gr.Number(value=5.0, label='Guidance Scale', min_width=100)
                             num_chunks = gr.Slider(maximum=5000000, minimum=1000, value=8000,
                                                    label='Number of Chunks', min_width=100)
-                    with gr.Tab("Export", id='tab_export'):
+                    with gr.Tab("Exportar", id='tab_export'):
                         with gr.Row():
-                            file_type = gr.Dropdown(label='File Type', choices=SUPPORTED_FORMATS,
+                            file_type = gr.Dropdown(label='Archivo', choices=SUPPORTED_FORMATS,
                                                     value='glb', min_width=100)
-                            reduce_face = gr.Checkbox(label='Simplify Mesh', value=False, min_width=100)
+                            reduce_face = gr.Checkbox(label='Modelo Simpl.', value=False, min_width=100)
                             export_texture = gr.Checkbox(label='Include Texture', value=False,
                                                          visible=False, min_width=100)
                         target_face_num = gr.Slider(maximum=1000000, minimum=100, value=10000,
-                                                    label='Target Face Number')
+                                                    label='Numero de frentes objetivo')
                         with gr.Row():
-                            confirm_export = gr.Button(value="Transform", min_width=100)
-                            file_export = gr.DownloadButton(label="Download", variant='primary',
+                            confirm_export = gr.Button(value="Transformar", min_width=100)
+                            file_export = gr.DownloadButton(label="Descargar", variant='primary',
                                                             interactive=False, min_width=100)
 
             with gr.Column(scale=9):
                 with gr.Tabs(selected='gen_mesh_panel') as tabs_output:
                     with gr.Tab('Modelo Generado', id='gen_mesh_panel'):
                         html_gen_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
-                    #with gr.Tab('Exporting Mesh', id='export_mesh_panel'):
-                    #    html_export_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output')
-                    #with gr.Tab('Mesh Statistic', id='stats_panel'):
-                    #    stats = gr.Json({}, label='Mesh Stats')
+                    with gr.Tab('Exporting Mesh', id='export_mesh_panel', visible=False):
+                        html_export_mesh = gr.HTML(HTML_OUTPUT_PLACEHOLDER, label='Output', visible=False)
+                    with gr.Tab('Mesh Statistic', id='stats_panel'):
+                        stats = gr.Json({}, label='Mesh Stats')
 
         btn.click(
             shape_generation,
@@ -445,6 +447,31 @@ def build_app():
             outputs=[tabs_output],
         )
 
+        btn_all.click(
+            generation_all,
+            inputs=[
+                mv_image_front,
+                mv_image_back,
+                mv_image_left,
+                mv_image_right,
+                num_steps,
+                cfg_scale,
+                seed,
+                octree_resolution,
+                check_box_rembg,
+                num_chunks,
+                randomize_seed,
+            ],
+            outputs=[file_out, file_out2, html_gen_mesh, stats, seed]
+        ).then(
+            lambda: (gr.update(visible=True, value=True), gr.update(interactive=False), gr.update(interactive=True),
+                     gr.update(interactive=False)),
+            outputs=[export_texture, reduce_face, confirm_export, file_export],
+        ).then(
+            lambda: gr.update(selected='gen_mesh_panel'),
+            outputs=[tabs_output],
+        )
+
         def on_gen_mode_change(value):
             if value == 'Turbo':
                 return gr.update(value=5)
@@ -471,20 +498,30 @@ def build_app():
 
             print(f'exporting {file_out}')
             print(f'reduce face to {target_face_num}')
-            
-            mesh = trimesh.load(file_out)
-            mesh = floater_remove_worker(mesh)
-            mesh = degenerate_face_remove_worker(mesh)
-            if reduce_face:
-                mesh = face_reduce_worker(mesh, target_face_num)
-            save_folder = gen_save_folder()
-            path = export_mesh(mesh, save_folder, textured=False, type=file_type)
-
-            # for preview
-            save_folder = gen_save_folder()
-            _ = export_mesh(mesh, save_folder, textured=False)
-            model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH,
-                                                        textured=False)
+            if export_texture:
+                mesh = trimesh.load(file_out2)
+                save_folder = gen_save_folder()
+                path = export_mesh(mesh, save_folder, textured=True, type=file_type)
+
+                # for preview
+                save_folder = gen_save_folder()
+                _ = export_mesh(mesh, save_folder, textured=True)
+                model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH,
+                                                            textured=True)
+            else:
+                mesh = trimesh.load(file_out)
+                mesh = floater_remove_worker(mesh)
+                mesh = degenerate_face_remove_worker(mesh)
+                if reduce_face:
+                    mesh = face_reduce_worker(mesh, target_face_num)
+                save_folder = gen_save_folder()
+                path = export_mesh(mesh, save_folder, textured=False, type=file_type)
+
+                # for preview
+                save_folder = gen_save_folder()
+                _ = export_mesh(mesh, save_folder, textured=False)
+                model_viewer_html = build_model_viewer_html(save_folder, height=HTML_HEIGHT, width=HTML_WIDTH,
+                                                            textured=False)
             print(f'export to {path}')
             return model_viewer_html, gr.update(value=path, interactive=True)