Upload folder using huggingface_hub

app.py

@@ -34,20 +34,80 @@ from PIL import Image
 from einops import rearrange
 import pandas as pd
 
+# import sys
+# import spaces
+# import subprocess
+# from huggingface_hub import snapshot_download
+
+# def install_cuda_toolkit():
+#     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
+#     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
+#     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
+#     subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
+#     subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
+#     subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
+
+#     os.environ["CUDA_HOME"] = "/usr/local/cuda"
+#     os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
+#     os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
+#         os.environ["CUDA_HOME"],
+#         "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
+#     )
+#     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
+#     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
+        
+# def install_requirements():
+#     subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/NVlabs/nvdiffrast"])
+#     subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"])
+
+# def download_models():
+#     os.makedirs("weights", exist_ok=True)
+#     os.makedirs("weights/hunyuanDiT", exist_ok=True)
+#     os.makedirs("third_party/weights/DUSt3R_ViTLarge_BaseDecoder_512_dpt", exist_ok=True)
+#     try:
+#         snapshot_download(
+#             repo_id="tencent/Hunyuan3D-1",
+#             local_dir="./weights",
+#             resume_download=True
+#         )
+#         print("Successfully downloaded Hunyuan3D-1 model")
+#     except Exception as e:
+#         print(f"Error downloading Hunyuan3D-1: {e}")
+#     try:
+#         snapshot_download(
+#             repo_id="Tencent-Hunyuan/HunyuanDiT-v1.1-Diffusers-Distilled",
+#             local_dir="./weights/hunyuanDiT",
+#             resume_download=True
+#         )
+#         print("Successfully downloaded HunyuanDiT model")
+#     except Exception as e:
+#         print(f"Error downloading HunyuanDiT: {e}")
+#     try:
+#         snapshot_download(
+#             repo_id="naver/DUSt3R_ViTLarge_BaseDecoder_512_dpt",
+#             local_dir="./third_party/weights/DUSt3R_ViTLarge_BaseDecoder_512_dpt",
+#             resume_download=True
+#         )
+#         print("Successfully downloaded DUSt3R model")
+#     except Exception as e:
+#         print(f"Error downloading DUSt3R: {e}")
+# install_cuda_toolkit()
+# install_requirements()
+# download_models()
+
 from infer import seed_everything, save_gif
 from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer
 from third_party.check import check_bake_available
 
 try:
     from third_party.mesh_baker import MeshBaker
+    assert check_bake_available()
     BAKE_AVAILEBLE = True
 except Exception as err:
     print(err)
     print("import baking related fail, run without baking")
-    check_bake_available()
     BAKE_AVAILEBLE = False
 
-
 warnings.simplefilter('ignore', category=UserWarning)
 warnings.simplefilter('ignore', category=FutureWarning)
 warnings.simplefilter('ignore', category=DeprecationWarning)
@@ -57,17 +117,15 @@ parser.add_argument("--use_lite", default=False, action="store_true")
 parser.add_argument("--mv23d_cfg_path", default="./svrm/configs/svrm.yaml", type=str)
 parser.add_argument("--mv23d_ckt_path", default="weights/svrm/svrm.safetensors", type=str)
 parser.add_argument("--text2image_path", default="weights/hunyuanDiT", type=str)
-parser.add_argument("--save_memory", default=False, action="store_true")
+parser.add_argument("--save_memory", default=False)
 parser.add_argument("--device", default="cuda:0", type=str)
 args = parser.parse_args()
 
+
 ################################################################
 # initial setting
 ################################################################
 
-CONST_PORT = 8080
-CONST_MAX_QUEUE = 1
-CONST_SERVER = '0.0.0.0'
 
 CONST_HEADER = '''
 <h2><a href='https://github.com/tencent/Hunyuan3D-1' target='_blank'><b>Tencent Hunyuan3D-1.0: A Unified Framework for Text-to-3D and Image-to-3D Generation</b></a></h2>
@@ -127,36 +185,36 @@ worker_v23 = Views2Mesh(
 )
 worker_gif = GifRenderer(args.device)
 
-
 if BAKE_AVAILEBLE:
     worker_baker = MeshBaker()
 
 
 ### functional modules    
 
-def stage_0_t2i(text, image, seed, step):
+def gen_save_folder(max_size=30):
     os.makedirs('./outputs/app_output', exist_ok=True)
     exists = set(int(_) for _ in os.listdir('./outputs/app_output') if not _.startswith("."))
-    if len(exists) == 30: shutil.rmtree(f"./outputs/app_output/0");cur_id = 0
-    else:                 cur_id = min(set(range(30)) - exists)
-    if os.path.exists(f"./outputs/app_output/{(cur_id + 1) % 30}"):
-        shutil.rmtree(f"./outputs/app_output/{(cur_id + 1) % 30}")
+    if len(exists) == max_size: 
+        shutil.rmtree(f"./outputs/app_output/0")
+        cur_id = 0
+    else:
+        cur_id = min(set(range(max_size)) - exists)
+    if os.path.exists(f"./outputs/app_output/{(cur_id + 1) % max_size}"):
+        shutil.rmtree(f"./outputs/app_output/{(cur_id + 1) % max_size}")
     save_folder = f'./outputs/app_output/{cur_id}'
     os.makedirs(save_folder, exist_ok=True)
+    print(f"mkdir {save_folder} suceess !!!")
+    return save_folder
 
+@space.GPU(duration=120)
+def stage_0_t2i(text, seed, step, save_folder):
     dst = save_folder + '/img.png'
-    
-    if not text:
-        if image is None: 
-            return dst, save_folder
-            raise gr.Error("Upload image or provide text ...")
-        image.save(dst)
-        return dst, save_folder
-        
     image = worker_t2i(text, seed, step)
     image.save(dst)
-    dst = worker_xbg(image, save_folder)
-    return dst, save_folder
+    img_nobg = worker_xbg(image, force=True)
+    dst = save_folder + '/img_nobg.png'
+    img_nobg.save(dst)
+    return dst
 
 def stage_1_xbg(image, save_folder, force_remove): 
     if isinstance(image, str):
@@ -165,7 +223,8 @@ def stage_1_xbg(image, save_folder, force_remove):
     rgba = worker_xbg(image, force=force_remove)
     rgba.save(dst)
     return dst
-    
+
+# @space.GPU
 def stage_2_i2v(image, seed, step, save_folder):
     if isinstance(image, str):
         image = Image.open(image)
@@ -180,6 +239,7 @@ def stage_2_i2v(image, seed, step, save_folder):
     show_img = Image.fromarray(show_img) 
     return views_img, cond_img, show_img
 
+# @space.GPU
 def stage_3_v23(
     views_pil, 
     cond_pil, 
@@ -202,32 +262,32 @@ def stage_3_v23(
     obj_dst = save_folder + '/mesh_vertex_colors.obj' # gradio just only can show vertex shading
     return obj_dst, glb_dst
 
-def stage_3p_baking(save_folder, color, bake):
+# @space.GPU
+def stage_3p_baking(save_folder, color, bake, force, front, others, align_times):
     if color == "texture" and bake:
-        obj_dst = worker_baker(save_folder)
+        obj_dst = worker_baker(save_folder, force, front, others, align_times)
         glb_dst = obj_dst.replace(".obj", ".glb")
         return glb_dst
     else:
         return None
 
+# @space.GPU
 def stage_4_gif(save_folder, color, bake, render):
     if not render: return None
-    if os.path.exists(save_folder + '/view_1/bake/mesh.obj'):
-        obj_dst = save_folder + '/view_1/bake/mesh.obj'
-    elif os.path.exists(save_folder + '/view_0/bake/mesh.obj'):
-        obj_dst = save_folder + '/view_0/bake/mesh.obj'
-    elif os.path.exists(save_folder + '/mesh.obj'):
-        obj_dst = save_folder + '/mesh.obj'
-    else:
-        print(save_folder)
-        raise FileNotFoundError("mesh obj file not found")
+    baked_fld_list = sorted(glob(save_folder + '/view_*/bake/mesh.obj'))
+    obj_dst = baked_fld_list[-1] if len(baked_fld_list)>=1 else save_folder+'/mesh.obj'
+    assert os.path.exists(obj_dst), f"{obj_dst} file not found"
     gif_dst = obj_dst.replace(".obj", ".gif")
     worker_gif(obj_dst, gif_dst_path=gif_dst)
     return gif_dst
 
 
 def check_image_available(image):
-    if image.mode == "RGBA":
+    if image is None:
+        return "Please upload image", gr.update()
+    elif not hasattr(image, 'mode'):
+        return "Not support, please upload other image", gr.update()
+    elif image.mode == "RGBA":
         data = np.array(image)
         alpha_channel = data[:, :, 3]
         unique_alpha_values = np.unique(alpha_channel)
@@ -243,11 +303,29 @@ def check_image_available(image):
     else:
         raise Exception("Image Error")
     
-def update_bake_render(color):
-    if color == "vertex":
-        return gr.update(value=False, interactive=False), gr.update(value=False, interactive=False)
-    else:
-        return gr.update(interactive=True), gr.update(interactive=True)
+
+def update_mode(mode):
+    color_change = {
+        'Quick': gr.update(value='vertex'),
+        'Moderate': gr.update(value='texture'),
+        'Appearance': gr.update(value='texture')
+    }[mode]
+    bake_change = {
+        'Quick': gr.update(value=False, interactive=False, visible=False),
+        'Moderate': gr.update(value=False),
+        'Appearance': gr.update(value=BAKE_AVAILEBLE)
+    }[mode]
+    face_change = {
+        'Quick': gr.update(value=120000, maximum=300000),
+        'Moderate': gr.update(value=60000, maximum=300000),
+        'Appearance': gr.update(value=10000, maximum=60000)
+    }[mode]
+    render_change = {
+        'Quick': gr.update(value=False, interactive=False, visible=False),
+        'Moderate': gr.update(value=True),
+        'Appearance': gr.update(value=True)
+    }[mode]
+    return color_change, bake_change, face_change, render_change
     
 # ===============================================================
 # gradio display
@@ -267,35 +345,113 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     text = gr.TextArea('一只黑���相间的熊猫在白色背景上居中坐着，呈现出卡通风格和可爱氛围。', 
                                        lines=3, max_lines=20, label='Input text')
-                    with gr.Row():
-                        textgen_color = gr.Radio(choices=["vertex", "texture"], label="Color", value="texture")
-                    with gr.Row():
-                        textgen_render = gr.Checkbox(label="Do Rendering", value=True, interactive=True)
-                        if BAKE_AVAILEBLE:
-                            textgen_bake = gr.Checkbox(label="Do Baking", value=True, interactive=True)
-                        else:
-                            textgen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=False)
-                    
-                    textgen_color.change(
-                        fn=update_bake_render, 
-                        inputs=textgen_color, 
-                        outputs=[textgen_bake, textgen_render]
-                    )
 
-                    with gr.Row():
-                        textgen_seed = gr.Number(value=0, label="T2I seed", precision=0, interactive=True)
-                        textgen_step = gr.Number(value=25, label="T2I steps", precision=0, 
-                                                 minimum=10, maximum=50, interactive=True)
-                        textgen_SEED = gr.Number(value=0, label="Gen seed", precision=0, interactive=True)
-                        textgen_STEP = gr.Number(value=50, label="Gen steps", precision=0, 
-                                                 minimum=40, maximum=100, interactive=True)
-                        textgen_max_faces = gr.Number(value=90000, label="Face number", precision=0, 
-                                                      minimum=5000, maximum=1000000, interactive=True)
+                    textgen_mode = gr.Radio(
+                        choices=['Quick', 'Moderate', 'Appearance'], 
+                        label="Simple settings",
+                        value='Appearance',
+                        interactive=True
+                    )
+                    
+                    with gr.Accordion("Custom settings", open=False):
+                        textgen_color = gr.Radio(choices=["vertex", "texture"], label="Color", value="texture")
+                        
+                        with gr.Row():
+                            textgen_render = gr.Checkbox(
+                                label="Do Rendering", 
+                                value=True, 
+                                interactive=True
+                            )
+                            textgen_bake = gr.Checkbox(
+                                label="Do Baking", 
+                                value=True if BAKE_AVAILEBLE else False, 
+                                interactive=True if BAKE_AVAILEBLE else False
+                            )
+                            
+                        with gr.Row():
+                            textgen_seed = gr.Number(value=0, label="T2I seed", precision=0, interactive=True)
+                            textgen_SEED = gr.Number(value=0, label="Gen seed", precision=0, interactive=True)
+
+                        textgen_step = gr.Slider(
+                            value=25,
+                            minimum=15,
+                            maximum=50,
+                            step=1,
+                            label="T2I steps",
+                            interactive=True
+                        )
+                        textgen_STEP = gr.Slider(
+                            value=50,
+                            minimum=20,
+                            maximum=80,
+                            step=1,
+                            label="Gen steps",
+                            interactive=True
+                        )
+                        textgen_max_faces =gr.Slider(
+                            value=10000,
+                            minimum=2000,
+                            maximum=60000,
+                            step=1000,
+                            label="Face number limit",
+                            interactive=True
+                        )
+
+                        with gr.Accordion("Baking Options", open=False):
+                            textgen_force_bake = gr.Checkbox(
+                                label="Force (Ignore the degree of matching)", 
+                                value=False, 
+                                interactive=True
+                            )
+                            textgen_front_baking = gr.Radio(
+                                choices=['input image', 'multi-view front view', 'auto'], 
+                                label="Front view baking",
+                                value='auto',
+                                interactive=True,
+                                visible=True
+                            )
+                            textgen_other_views = gr.CheckboxGroup(
+                                choices=['60°', '120°', '180°', '240°', '300°'], 
+                                label="Other views Baking",
+                                value=['180°'],
+                                interactive=True,
+                                visible=True
+                            )
+                            textgen_align_times =gr.Slider(
+                                value=3,
+                                minimum=1,
+                                maximum=5,
+                                step=1,
+                                label="Number of alignment attempts per view",
+                                interactive=True
+                            )
+                        
                     with gr.Row():
                         textgen_submit = gr.Button("Generate", variant="primary")
 
                     with gr.Row():
                         gr.Examples(examples=example_ts, inputs=[text], label="Text examples", examples_per_page=10)
+                        
+                        
+            textgen_mode.change(
+                fn=update_mode,
+                inputs=textgen_mode, 
+                outputs=[textgen_color, textgen_bake, textgen_max_faces, textgen_render]
+            )
+            textgen_color.change(
+                fn=lambda x:[
+                    gr.update(value=(x=='texture'), interactive=(x=='texture'), visible=(x=='texture')),
+                    gr.update(value=(x=='texture'), interactive=(x=='texture'), visible=(x=='texture')),
+                ],
+                inputs=textgen_color, 
+                outputs=[textgen_bake, textgen_render]
+            )
+            textgen_bake.change(
+                fn= lambda x:[gr.update(visible=x)]*4+[gr.update(value=10000, minimum=2000, maximum=60000 if x else 300000)],
+                inputs=textgen_bake, 
+                outputs=[textgen_front_baking, textgen_other_views, textgen_align_times, textgen_force_bake, textgen_max_faces]
+            )
+            
                     
             ### Image iutput region
             
@@ -305,39 +461,112 @@ with gr.Blocks() as demo:
                                            image_mode="RGBA", sources="upload", interactive=True)
                 with gr.Row():
                     alert_message = gr.Markdown("")  # for warning 
-                with gr.Row():
-                    imggen_color = gr.Radio(choices=["vertex", "texture"], label="Color", value="texture")
-                with gr.Row():
-                    imggen_removebg = gr.Checkbox(label="Remove Background", value=True, interactive=True)
-                    imggen_render = gr.Checkbox(label="Do Rendering", value=True, interactive=True)
-                    if BAKE_AVAILEBLE:
-                        imggen_bake = gr.Checkbox(label="Do Baking", value=True, interactive=True)
-                    else:
-                        imggen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=False)
                     
+                imggen_mode = gr.Radio(
+                    choices=['Quick', 'Moderate', 'Appearance'], 
+                    label="Simple settings",
+                    value='Appearance',
+                    interactive=True
+                )
+                
+                with gr.Accordion("Custom settings", open=False):
+                    imggen_color = gr.Radio(choices=["vertex", "texture"], label="Color", value="texture")
+
+                    with gr.Row():
+                        imggen_removebg = gr.Checkbox(
+                            label="Remove Background", 
+                            value=True, 
+                            interactive=True
+                        )
+                        imggen_render = gr.Checkbox(
+                            label="Do Rendering", 
+                            value=True, 
+                            interactive=True
+                        )
+                        imggen_bake = gr.Checkbox(
+                            label="Do Baking", 
+                            value=True if BAKE_AVAILEBLE else False, 
+                            interactive=True if BAKE_AVAILEBLE else False
+                        )
+                    imggen_SEED = gr.Number(value=0, label="Gen seed", precision=0, interactive=True)
+
+                    imggen_STEP = gr.Slider(
+                        value=50,
+                        minimum=20,
+                        maximum=80,
+                        step=1,
+                        label="Gen steps",
+                        interactive=True
+                    )
+                    imggen_max_faces =gr.Slider(
+                        value=10000,
+                        minimum=2000,
+                        maximum=60000,
+                        step=1000,
+                        label="Face number limit",
+                        interactive=True
+                    )
+
+                    with gr.Accordion("Baking Options", open=False):
+                        imggen_force_bake = gr.Checkbox(
+                                label="Force (Ignore the degree of matching)", 
+                                value=False, 
+                                interactive=True
+                            )
+                        imggen_front_baking = gr.Radio(
+                            choices=['input image', 'multi-view front view', 'auto'], 
+                            label="Front view baking",
+                            value='auto',
+                            interactive=True,
+                            visible=True
+                        )
+                        imggen_other_views = gr.CheckboxGroup(
+                            choices=['60°', '120°', '180°', '240°', '300°'], 
+                            label="Other views Baking",
+                            value=['180°'],
+                            interactive=True,
+                            visible=True
+                        )
+                        imggen_align_times =gr.Slider(
+                            value=3,
+                            minimum=1,
+                            maximum=5,
+                            step=1,
+                            label="Number of alignment attempts per view",
+                            interactive=True
+                        )
+
                 input_image.change(
                     fn=check_image_available, 
                     inputs=input_image, 
                     outputs=[alert_message, imggen_removebg]
                 )
+                
+                imggen_mode.change(
+                    fn=update_mode,
+                    inputs=imggen_mode, 
+                    outputs=[imggen_color, imggen_bake, imggen_max_faces, imggen_render]
+                )
+                
                 imggen_color.change(
-                    fn=update_bake_render, 
+                    fn=lambda x:[gr.update(value=(x=='texture'), interactive=(x=='texture'), visible=(x=='texture'))]*2, 
                     inputs=imggen_color, 
                     outputs=[imggen_bake, imggen_render]
                 )
-                        
-                with gr.Row():
-                    imggen_SEED = gr.Number(value=0, label="Gen seed", precision=0, interactive=True)
-                    imggen_STEP = gr.Number(value=50, label="Gen steps", precision=0, 
-                                            minimum=40, maximum=100, interactive=True)
-                    imggen_max_faces = gr.Number(value=90000, label="Face number", precision=0, 
-                                                     minimum=5000, maximum=1000000, interactive=True)
+                
+                imggen_bake.change(
+                    fn= lambda x:[gr.update(visible=x)]*4+[gr.update(value=120000, minimum=2000, maximum=60000 if x else 300000)],
+                    inputs=imggen_bake, 
+                    outputs=[imggen_front_baking, imggen_other_views, imggen_align_times, imggen_force_bake, imggen_max_faces]
+                )
+
                 with gr.Row():
                     imggen_submit = gr.Button("Generate", variant="primary")      
 
                 with gr.Row():
                     gr.Examples(examples=example_is,  inputs=[input_image], 
                         label="Img examples", examples_per_page=10)
+            
                     
             gr.Markdown(CONST_NOTE)
                     
@@ -359,33 +588,32 @@ with gr.Blocks() as demo:
                         interactive=False
                     )
             
-            with gr.Row():          
-                result_3dobj = gr.Model3D(
-                    clear_color=[0.0, 0.0, 0.0, 0.0],
-                    label="OBJ vertex color",
-                    show_label=True,
-                    visible=True,
-                    camera_position=[90, 90, None],
-                    interactive=False
-                )
-                result_gif = gr.Image(label="GIF", interactive=False)
+            result_3dobj = gr.Model3D(
+                clear_color=[0.0, 0.0, 0.0, 0.0],
+                label="OBJ vertex color",
+                show_label=True,
+                visible=True,
+                camera_position=[90, 90, None],
+                interactive=False
+            )
                 
-            with gr.Row():
-                result_3dglb_texture = gr.Model3D(
-                    clear_color=[0.0, 0.0, 0.0, 0.0],
-                    label="GLB texture color",
-                    show_label=True,
-                    visible=True,
-                    camera_position=[90, 90, None],
-                    interactive=False)
-
-                result_3dglb_baked = gr.Model3D(
-                    clear_color=[0.0, 0.0, 0.0, 0.0],
-                    label="GLB baked color",
-                    show_label=True,
-                    visible=True,
-                    camera_position=[90, 90, None],
-                    interactive=False)
+            result_3dglb_texture = gr.Model3D(
+                clear_color=[0.0, 0.0, 0.0, 0.0],
+                label="GLB texture color",
+                show_label=True,
+                visible=True,
+                camera_position=[90, 90, None],
+                interactive=False)
+
+            result_3dglb_baked = gr.Model3D(
+                clear_color=[0.0, 0.0, 0.0, 0.0],
+                label="GLB baked color",
+                show_label=True,
+                visible=True,
+                camera_position=[90, 90, None],
+                interactive=False)
+            
+            result_gif = gr.Image(label="GIF", interactive=False)
                 
             with gr.Row():
                 gr.Markdown(
@@ -399,17 +627,23 @@ with gr.Blocks() as demo:
     # gradio running code
     #===============================================================
     
-    none = gr.State(None)
     save_folder = gr.State()
     cond_image = gr.State()
     views_image = gr.State()
-    text_image = gr.State()
-    
     
+    def handle_click(save_folder):
+        if save_folder is None:
+            save_folder = gen_save_folder()
+        return save_folder
+
     textgen_submit.click(
+        fn=handle_click,
+        inputs=[save_folder],
+        outputs=[save_folder]
+    ).success(
         fn=stage_0_t2i, 
-        inputs=[text, none, textgen_seed, textgen_step], 
-        outputs=[rem_bg_image, save_folder],
+        inputs=[text, textgen_seed, textgen_step, save_folder], 
+        outputs=[rem_bg_image],
     ).success(
         fn=stage_2_i2v, 
         inputs=[rem_bg_image, textgen_SEED, textgen_STEP, save_folder], 
@@ -420,7 +654,8 @@ with gr.Blocks() as demo:
         outputs=[result_3dobj, result_3dglb_texture],
     ).success(
         fn=stage_3p_baking, 
-        inputs=[save_folder, textgen_color, textgen_bake], 
+        inputs=[save_folder, textgen_color, textgen_bake,
+               textgen_force_bake, textgen_front_baking, textgen_other_views, textgen_align_times], 
         outputs=[result_3dglb_baked],
     ).success(
         fn=stage_4_gif, 
@@ -430,12 +665,12 @@ with gr.Blocks() as demo:
 
     
     imggen_submit.click(
-        fn=stage_0_t2i, 
-        inputs=[none, input_image, textgen_seed, textgen_step], 
-        outputs=[text_image, save_folder],
+        fn=handle_click,
+        inputs=[save_folder],
+        outputs=[save_folder]
     ).success(
         fn=stage_1_xbg, 
-        inputs=[text_image, save_folder, imggen_removebg], 
+        inputs=[input_image, save_folder, imggen_removebg], 
         outputs=[rem_bg_image],
     ).success(
         fn=stage_2_i2v, 
@@ -447,7 +682,8 @@ with gr.Blocks() as demo:
         outputs=[result_3dobj, result_3dglb_texture],
     ).success(
         fn=stage_3p_baking, 
-        inputs=[save_folder, imggen_color, imggen_bake], 
+        inputs=[save_folder, imggen_color, imggen_bake, 
+                imggen_force_bake, imggen_front_baking, imggen_other_views, imggen_align_times], 
         outputs=[result_3dglb_baked],
     ).success(
         fn=stage_4_gif, 
@@ -458,7 +694,11 @@ with gr.Blocks() as demo:
     #===============================================================
     # start gradio server
     #===============================================================
+    CONST_PORT = 8080
+    CONST_MAX_QUEUE = 1
+    CONST_SERVER = '0.0.0.0'
 
     demo.queue(max_size=CONST_MAX_QUEUE)
     demo.launch(server_name=CONST_SERVER, server_port=CONST_PORT)
-
+    # demo.launch()
+    

app_hg.py

@@ -700,6 +700,6 @@ with gr.Blocks() as demo:
     CONST_MAX_QUEUE = 1
     CONST_SERVER = '0.0.0.0'
 
-    demo.queue()
+    demo.queue(max_size=CONST_MAX_QUEUE)
     demo.launch()
     

main.py

@@ -195,14 +195,10 @@ if __name__ == "__main__":
     if args.do_render:
         if mesh_file_for_render and os.path.exists(mesh_file_for_render):
             mesh_file_for_render = mesh_file_for_render
-        elif os.path.exists(os.path.join(args.save_folder, 'view_1/bake/mesh.obj')):
-            mesh_file_for_render = os.path.join(args.save_folder, 'view_1/bake/mesh.obj')
-        elif os.path.exists(os.path.join(args.save_folder, 'view_0/bake/mesh.obj')):
-            mesh_file_for_render = os.path.join(args.save_folder, 'view_0/bake/mesh.obj')
-        elif os.path.exists(os.path.join(args.save_folder, 'mesh.obj')):
-            mesh_file_for_render = os.path.join(args.save_folder, 'mesh.obj')
         else:
-            raise FileNotFoundError("mesh_file_for_render not found")
+            baked_fld_list = sorted(glob(args.save_folder + '/view_*/bake/mesh.obj'))
+            mesh_file_for_render = baked_fld_list[-1] if len(baked_fld_list)>=1 else args.save_folder+'/mesh.obj'
+            assert os.path.exists(mesh_file_for_render), f"{mesh_file_for_render} file not found"
             
         print("Rendering 3d file:", mesh_file_for_render)