Upload app_hg.py with huggingface_hub

app_hg.py

@@ -21,7 +21,7 @@
 # optimizer states), machine-learning model code, inference-enabling code, training-enabling code, 
 # fine-tuning enabling code and other elements of the foregoing made publicly available 
 # by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
-import spaces
+
 import os
 import warnings
 import argparse
@@ -33,56 +33,35 @@ import numpy as np
 from PIL import Image
 from einops import rearrange
 import pandas as pd
-from huggingface_hub import snapshot_download
 
 import sys
+import spaces
 import subprocess
-from glob import glob
+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_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"
         
-# install_cuda_toolkit()
-
 def install_requirements():
-    # Install the packages listed in requirements.txt
     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"])
-    
-install_requirements()
 
-
-
-from infer import seed_everything, save_gif
-from infer import Text2Image, Removebg, Image2Views, Views2Mesh, GifRenderer
-from third_party.check import check_bake_available
-
-warnings.simplefilter('ignore', category=UserWarning)
-warnings.simplefilter('ignore', category=FutureWarning)
-warnings.simplefilter('ignore', category=DeprecationWarning)
-
-parser = argparse.ArgumentParser()
-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)
-parser.add_argument("--device", default="cuda:0", type=str)
-args = parser.parse_args()
+# install_cuda_toolkit()
+install_requirements()
 
 def download_models():
     os.makedirs("weights", exist_ok=True)
@@ -116,7 +95,10 @@ def download_models():
     except Exception as e:
         print(f"Error downloading DUSt3R: {e}")
         
-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
@@ -124,9 +106,23 @@ try:
     BAKE_AVAILEBLE = True
 except Exception as err:
     print(err)
-    print("import baking related files fail, running without baking")
+    print("import baking related fail, run without baking")
     BAKE_AVAILEBLE = False
 
+warnings.simplefilter('ignore', category=UserWarning)
+warnings.simplefilter('ignore', category=FutureWarning)
+warnings.simplefilter('ignore', category=DeprecationWarning)
+
+parser = argparse.ArgumentParser()
+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)
+parser.add_argument("--device", default="cuda:0", type=str)
+args = parser.parse_args()
+
+download_models() ### download weights !!!! 
 
 ################################################################
 # initial setting
@@ -170,13 +166,13 @@ example_ts = get_example_txt_list()
 # initial models
 ################################################################
 
+worker_xbg = Removebg()
 print(f"loading {args.text2image_path}")
 worker_t2i = Text2Image(
     pretrain = args.text2image_path, 
     device = args.device, 
     save_memory = args.save_memory
 )
-worker_xbg = Removebg()
 worker_i2v = Image2Views(
     use_lite = args.use_lite, 
     device = args.device,
@@ -189,38 +185,38 @@ worker_v23 = Views2Mesh(
     device = args.device,
     save_memory = args.save_memory
 )
-worker_gif = GifRenderer(device=args.device)
-
+worker_gif = GifRenderer(args.device)
 
 if BAKE_AVAILEBLE:
-    worker_baker = MeshBaker(device=args.device)
+    worker_baker = MeshBaker()
 
 
 ### functional modules    
-@spaces.GPU(duration=140)
-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):
@@ -230,7 +226,7 @@ def stage_1_xbg(image, save_folder, force_remove):
     rgba.save(dst)
     return dst
 
-@spaces.GPU(duration=110)
+@space.GPU
 def stage_2_i2v(image, seed, step, save_folder):
     if isinstance(image, str):
         image = Image.open(image)
@@ -245,7 +241,7 @@ def stage_2_i2v(image, seed, step, save_folder):
     show_img = Image.fromarray(show_img) 
     return views_img, cond_img, show_img
 
-@spaces.GPU(duration=70)
+@space.GPU
 def stage_3_v23(
     views_pil, 
     cond_pil, 
@@ -268,34 +264,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
 
-@spaces.GPU(duration=40)
-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
-    
-@spaces.GPU(duration=20)
+
+@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)
@@ -311,11 +305,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
@@ -335,31 +347,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="vertex")
-                    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=False, 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=60000, 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
             
@@ -369,31 +463,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=False, interactive=True)
-                    else:
-                        imggen_bake = gr.Checkbox(label="Do Baking", value=False, interactive=False)
                     
-                input_image.change(fn=check_image_available, inputs=input_image, outputs=[alert_message, imggen_removebg])
-                imggen_color.change(fn=update_bake_render, inputs=imggen_color, outputs=[imggen_bake, imggen_render])
-                        
-                with gr.Row():
+                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.Number(value=50, label="Gen steps", precision=0, 
-                                            minimum=40, maximum=100, interactive=True)
-                    imggen_max_faces = gr.Number(value=60000, label="Face number", precision=0, 
-                                                     minimum=5000, maximum=1000000, 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=lambda x:[gr.update(value=(x=='texture'), interactive=(x=='texture'), visible=(x=='texture'))]*2, 
+                    inputs=imggen_color, 
+                    outputs=[imggen_bake, imggen_render]
+                )
+                
+                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)
                     
@@ -423,7 +598,7 @@ with gr.Blocks() as demo:
                 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",
@@ -454,17 +629,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], 
@@ -475,7 +656,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, 
@@ -485,12 +667,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, 
@@ -502,7 +684,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, 
@@ -513,7 +696,10 @@ with gr.Blocks() as demo:
     #===============================================================
     # start gradio server
     #===============================================================
+    CONST_PORT = 8080
+    CONST_MAX_QUEUE = 1
+    CONST_SERVER = '0.0.0.0'
 
     demo.queue()
     demo.launch()
-
+