Upload gradio_app.py

gradio_app.py

@@ -31,10 +31,9 @@ from hy3dgen.shapegen.utils import logger
 
 MAX_SEED = 1e7
 
-import spaces
-
 if True:
-    import os    
+    import os
+    import spaces
     import subprocess
     import sys
     import shlex
@@ -43,38 +42,6 @@ if True:
     print('install custom')
     subprocess.run(shlex.split("pip install custom_rasterizer-0.1-cp310-cp310-linux_x86_64.whl"), check=True)
 
-@spaces.GPU
-def my_gpu_function():
-    pass
-
-def get_example_img_list():
-    print('Loading example img list ...')
-    return sorted(glob('./assets/example_images/**/*.png', recursive=True))
-
-
-def get_example_txt_list():
-    print('Loading example txt list ...')
-    txt_list = list()
-    for line in open('./assets/example_prompts.txt', encoding='utf-8'):
-        txt_list.append(line.strip())
-    return txt_list
-
-
-def get_example_mv_list():
-    print('Loading example mv list ...')
-    mv_list = list()
-    root = './assets/example_mv_images'
-    for mv_dir in os.listdir(root):
-        view_list = []
-        for view in ['front', 'back', 'left', 'right']:
-            path = os.path.join(root, mv_dir, f'{view}.png')
-            if os.path.exists(path):
-                view_list.append(path)
-            else:
-                view_list.append(None)
-        mv_list.append(view_list)
-    return mv_list
-
 
 def gen_save_folder(max_size=200):
     os.makedirs(SAVE_DIR, exist_ok=True)
@@ -366,28 +333,10 @@ def shape_generation(
 
 
 def build_app():
-    title = 'Hunyuan3D-2: High Resolution Textured 3D Assets Generation'
-    if MV_MODE:
-        title = 'Hunyuan3D-2mv: Image to 3D Generation with 1-4 Views'
-    if 'mini' in args.subfolder:
-        title = 'Hunyuan3D-2mini: Strong 0.6B Image to Shape Generator'
-    if TURBO_MODE:
-        title = title.replace(':', '-Turbo: Fast ')
 
     title_html = f"""
     <div style="font-size: 2em; font-weight: bold; text-align: center; margin-bottom: 5px">
-
-    {title}
-    </div>
-    <div align="center">
-    Tencent Hunyuan3D Team
-    </div>
-    <div align="center">
-      <a href="https://github.com/tencent/Hunyuan3D-2">Github</a> &ensp; 
-      <a href="http://3d-models.hunyuan.tencent.com">Homepage</a> &ensp;
-      <a href="https://3d.hunyuan.tencent.com">Hunyuan3D Studio</a> &ensp;
-      <a href="#">Technical Report</a> &ensp;
-      <a href="https://huggingface.co/Tencent/Hunyuan3D-2"> Pretrained Models</a> &ensp;
+    Genera tu modelo 3D desde fotos
     </div>
     """
     custom_css = """
@@ -401,7 +350,6 @@ def build_app():
     .mv-image .icon-wrap {
         width: 20px;
     }
-
     """
 
     with gr.Blocks(theme=gr.themes.Base(), title='Hunyuan-3D-2.0', analytics_enabled=False, css=custom_css) as demo:
@@ -417,45 +365,42 @@ def build_app():
                         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:
                         # 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,
+                            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='Back', type='pil', image_mode='RGBA', height=140,
+                            mv_image_back = gr.Image(label='Parte trasera', type='pil', image_mode='RGBA', height=140,
                                                      min_width=100, elem_classes='mv-image')
                         with gr.Row():
-                            mv_image_left = gr.Image(label='Left', type='pil', image_mode='RGBA', height=140,
+                            mv_image_left = gr.Image(label='Izquierda', type='pil', image_mode='RGBA', height=140,
                                                      min_width=100, elem_classes='mv-image')
-                            mv_image_right = gr.Image(label='Right', type='pil', image_mode='RGBA', height=140,
+                            mv_image_right = gr.Image(label='Derecha', type='pil', image_mode='RGBA', height=140,
                                                       min_width=100, elem_classes='mv-image')
 
                 with gr.Row():
-                    btn = gr.Button(value='Gen Shape', variant='primary', min_width=100)
-                    btn_all = gr.Button(value='Gen Textured Shape',
-                                        variant='primary',
-                                        visible=HAS_TEXTUREGEN,
-                                        min_width=100)
+                    btn = gr.Button(value='Generar modelo', variant='primary', 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 generacioón',
+                                            info='Recomendacioón: Turbo por la mayoria, Fast para modelos complejos, Standard para uso cualquiera.',
                                             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='Modo de decoding',
+                                               info='La resolución para la exportación de la malla desde el vectset generado',
                                                choices=['Low', 'Standard', 'High'],
                                                value='Standard')
-                    with gr.Tab('Advanced Options', id='tab_advanced_options'):
+                    with gr.Tab('Opciones avanzadas', id='tab_advanced_options'):
                         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)
+                            check_box_rembg = gr.Checkbox(value=True, label='Eliminar fondo', min_width=100)
+                            randomize_seed = gr.Checkbox(label="Generar semilla aleatoria", value=True, min_width=100)
                         seed = gr.Slider(
-                            label="Seed",
+                            label="Semmilla",
                             minimum=0,
                             maximum=MAX_SEED,
                             step=1,
@@ -486,7 +431,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')
@@ -495,48 +440,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)
-
-        gr.HTML(f"""
-        <div align="center">
-        Activated Model - Shape Generation ({args.model_path}/{args.subfolder}) ; Texture Generation ({'Hunyuan3D-2' if HAS_TEXTUREGEN else 'Unavailable'})
-        </div>
-        """)
-        if not HAS_TEXTUREGEN:
-            gr.HTML("""
-            <div style="margin-top: 5px;"  align="center">
-                <b>Warning: </b>
-                Texture synthesis is disable due to missing requirements,
-                 please install requirements following <a href="https://github.com/Tencent/Hunyuan3D-2?tab=readme-ov-file#install-requirements">README.md</a>to activate it.
-            </div>
-            """)
-        if not args.enable_t23d:
-            gr.HTML("""
-            <div style="margin-top: 5px;"  align="center">
-                <b>Warning: </b>
-                Text to 3D is disable. To activate it, please run `python gradio_app.py --enable_t23d`.
-            </div>
-            """)
-
-        tab_ip.select(fn=lambda: gr.update(selected='tab_img_gallery'), outputs=gallery)
-        if HAS_T2I:
-            tab_tp.select(fn=lambda: gr.update(selected='tab_txt_gallery'), outputs=gallery)
-
 
         btn.click(
             shape_generation,
@@ -565,33 +468,6 @@ def build_app():
             outputs=[tabs_output],
         )
 
-        btn_all.click(
-            generation_all,
-            inputs=[
-                caption,
-                image,
-                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)
@@ -676,12 +552,6 @@ if __name__ == '__main__':
     parser.add_argument('--low_vram_mode', action='store_true')
     args = parser.parse_args()
 
-    try:
-        port = int(args.port)
-    except ValueError:
-        print(f"Invalid port argument detected: {args.port} — using default 7860")
-        port = 7860
-
     args.enable_flashvdm = True
     SAVE_DIR = args.cache_path
     os.makedirs(SAVE_DIR, exist_ok=True)
@@ -695,8 +565,7 @@ if __name__ == '__main__':
     HTML_OUTPUT_PLACEHOLDER = f"""
     <div style='height: {650}px; width: 100%; border-radius: 8px; border-color: #e5e7eb; border-style: solid; border-width: 1px; display: flex; justify-content: center; align-items: center;'>
       <div style='text-align: center; font-size: 16px; color: #6b7280;'>
-        <p style="color: #8d8d8d;">Welcome to Hunyuan3D!</p>
-        <p style="color: #8d8d8d;">No mesh here.</p>
+        <p style="color: #8d8d8d;">Carga tus fotos para generar el modelo 3D</p>
       </div>
     </div>
     """
@@ -706,14 +575,7 @@ if __name__ == '__main__':
     border-color: #e5e7eb; order-style: solid; border-width: 1px;'>
     </div>
     """
-
-    #demo = gr.Interface(fn=my_gpu_function, inputs=[], outputs="text")
-    #demo.launch()
-
-    example_is = get_example_img_list()
-    example_ts = get_example_txt_list()
-    example_mvs = get_example_mv_list()
-
+ 
     SUPPORTED_FORMATS = ['glb', 'obj', 'ply', 'stl']
 
     HAS_TEXTUREGEN = False
@@ -768,17 +630,15 @@ if __name__ == '__main__':
 
     # https://discuss.huggingface.co/t/how-to-serve-an-html-file/33921/2
     # create a FastAPI app
-    #app = FastAPI()
+    app = FastAPI()
     # create a static directory to store the static files
     static_dir = Path(SAVE_DIR).absolute()
     static_dir.mkdir(parents=True, exist_ok=True)
-    #app.mount("/static", StaticFiles(directory=static_dir, html=True), name="static")
+    app.mount("/static", StaticFiles(directory=static_dir, html=True), name="static")
     shutil.copytree('./assets/env_maps', os.path.join(static_dir, 'env_maps'), dirs_exist_ok=True)
 
-
     if args.low_vram_mode:
         torch.cuda.empty_cache()
     demo = build_app()
-    demo.launch()
-    #app = gr.mount_gradio_app(app, demo, path="/")
-    #uvicorn.run(app, host=args.host, port=args.port)
+    app = gr.mount_gradio_app(app, demo, path="/")
+    uvicorn.run(app, host=args.host, port=args.port)