new

app.py

@@ -96,7 +96,7 @@ from pipeline.kiss3d_wrapper import init_wrapper_from_config, run_text_to_3d, ru
 # Add logo file path and hyperlinks
 LOGO_PATH = "app_assets/logo_temp_.png"  # Update this to the actual path of your logo
 ARXIV_LINK = "https://arxiv.org/abs/example"
-GITHUB_LINK = "https://github.com/example"
+GITHUB_LINK = "https://github.com/EnVision-Research/Kiss3DGen"
 
 
 k3d_wrapper = init_wrapper_from_config('./pipeline/pipeline_config/default.yaml')
@@ -251,15 +251,17 @@ def bundle_image_to_mesh(
     else:
         return recon_mesh_path, mesh_cache
 
-_HEADER_=f"""
-<img src="{LOGO_PATH}">
-    <h2><b>Official 🤗 Gradio Demo</b></h2><h2>
-    <b>Kiss3DGen: Repurposing Image Diffusion Models for 3D Asset Generation</b></a></h2>
+# _HEADER_=f"""
+# <img src="{LOGO_PATH}">
+#     <h2><b>Official 🤗 Gradio Demo</b></h2>
+#     <h2><b>Kiss3DGen: Repurposing Image Diffusion Models for 3D Asset Generation</b></h2>
+#     <h2>Try our demo:Please click the buttons in sequence. Feel free to redo some steps multiple times until you get a </h2>
 
-<p>**Kiss3DGen** is xxxxxxxxx</p>
 
-[![arXiv](https://img.shields.io/badge/arXiv-Link-red)]({ARXIV_LINK})  [![GitHub](https://img.shields.io/badge/GitHub-Repo-blue)]({GITHUB_LINK})
-"""
+
+# [![arXiv](https://img.shields.io/badge/arXiv-Link-red)]({ARXIV_LINK})  [![GitHub](https://img.shields.io/badge/GitHub-Repo-blue)]({GITHUB_LINK})
+
+# """
 
 _CITE_ = r"""
 <h2>If Kiss3DGen is helpful, please help to ⭐ the <a href='{""" + GITHUB_LINK + r"""}' target='_blank'>Github Repo</a>. Thanks!</h2>
@@ -392,7 +394,9 @@ with gr.Blocks(css="""
             #     gr.Markdown(f"[![GitHub](https://img.shields.io/badge/GitHub-Repo-blue)]({GITHUB_LINK})")
 
     # Tabs Section
-    with gr.Tabs(selected='tab_text_to_3d', elem_id="content-container") as main_tabs:
+    # with gr.Tabs(selected='tab_text_to_3d', elem_id="content-container") as main_tabs:
+    if True:
+        gr.Markdown("Explore our Text-to-3D demo! Click the buttons in order (1 to 4). Adjust the seed or input to refine each step until you're satisfied before moving to the next.")
         with gr.TabItem('Text-to-3D', id='tab_text_to_3d'):
             with gr.Row():
                 with gr.Column(scale=1):
@@ -410,9 +414,9 @@ with gr.Blocks(css="""
                             inputs=[prompt],  # 将选中的示例填入 prompt 文本框
                             label="Example Prompts"
                         )
-                    btn_text2detailed = gr.Button("Refine to detailed prompt")
+                    btn_text2detailed = gr.Button("1. Refine to detailed prompt")
                     detailed_prompt = gr.Textbox(value="", label="Detailed Prompt", placeholder="detailed prompt will be generated here base on your input prompt. You can also edit this prompt", lines=4, interactive=True)
-                    btn_text2img = gr.Button("Generate Images")
+                    btn_text2img = gr.Button("2. Generate Images")
 
                 with gr.Column(scale=1):
                     output_image1 = gr.Image(label="Generated image", interactive=False)
@@ -423,12 +427,12 @@ with gr.Blocks(css="""
                     # reconstruction_stage1_steps = gr.Number(value=10, label="reconstruction_stage1_steps")
                     # reconstruction_stage2_steps = gr.Number(value=50, label="reconstruction_stage2_steps")
 
-                    btn_gen_mesh = gr.Button("Generate Mesh")
+                    btn_gen_mesh = gr.Button("3. Generate Mesh")
                     output_video1 = gr.Video(label="Render Video", interactive=False, loop=True, autoplay=True)
                     # btn_download1 = gr.Button("Download Mesh")
 
 
-                    download_1 = gr.DownloadButton(label="Download mesh", interactive=False)
+                    download_1 = gr.DownloadButton(label="4. Download mesh", interactive=False)
 
                     # file_output1 = gr.File()
                     
@@ -476,54 +480,13 @@ with gr.Blocks(css="""
 
     # Button Click Events
     # Text2
-    # btn_text2detailed.click(fn=text_to_detailed, inputs=[prompt, seed1], outputs=detailed_prompt)
-    # btn_text2img.click(fn=text_to_image, inputs=[detailed_prompt, seed1], outputs=output_image1)
-    # btn_gen_mesh.click(fn=bundle_image_to_mesh, inputs=[output_image1,], outputs=[output_video1, download_1]).then(
-    #     lambda: gr.Button(interactive=True),
-    #     outputs=[download_1],
-    # )
-
-
-
-    # Step 1: 点击 "Refine to detailed prompt" 按钮时
-    def refine_to_detailed(prompt, seed):
-        global generated_detailed_prompt
-        detailed_prompt_value = text_to_detailed(prompt, seed)
-        generated_detailed_prompt = True
-        return detailed_prompt_value
-
-    # Step 2: 点击 "Generate Image" 按钮时
-    def generate_image(prompt, seed, detailed_prompt_value):
-        global generated_image, generated_detailed_prompt
-        # 如果没有生成过详细描述，先执行 refine-to-detailed
-        if not generated_detailed_prompt:
-            detailed_prompt_value = text_to_detailed(prompt, seed)
-            generated_detailed_prompt = True
-        else:
-            detailed_prompt_value = f"Already refined: {prompt} (Seed: {seed})"
-
-        generated_image = True
-        return detailed_prompt_value, text_to_image(detailed_prompt_value, seed)
-
-    # Step 3: 点击 "Generate Mesh" 按钮时
-    def generate_mesh(image):
-        global generated_detailed_prompt, generated_image
-        # 如果没有生成图像，先执行生成图像
-        if not generated_image:
-            detailed_prompt_value = text_to_detailed(prompt, seed1)
-            image = text_to_image(detailed_prompt_value, seed1)
-            generated_image = True
-        generated_detailed_prompt, generated_image = False, False
-        # 生成网格
-        return bundle_image_to_mesh(image)
-    # Button Click Events
-    btn_text2detailed.click(fn=refine_to_detailed, inputs=[prompt, seed1], outputs=detailed_prompt)
-
-    # Generate Image
-    btn_text2img.click(fn=generate_image, inputs=[prompt, seed1, detailed_prompt], outputs=[detailed_prompt, output_image1])
+    btn_text2detailed.click(fn=text_to_detailed, inputs=[prompt, seed1], outputs=detailed_prompt)
+    btn_text2img.click(fn=text_to_image, inputs=[detailed_prompt, seed1], outputs=output_image1)
+    btn_gen_mesh.click(fn=bundle_image_to_mesh, inputs=[output_image1,], outputs=[output_video1, download_1]).then(
+        lambda: gr.Button(interactive=True),
+        outputs=[download_1],
+    )
 
-    # Generate Mesh
-    btn_gen_mesh.click(fn=generate_mesh, inputs=[output_image1], outputs=[output_video1, download_1])
         
     with gr.Row():
         pass