优化界面和交互

app.py

@@ -12,6 +12,16 @@ def get_example_img_list():
     return sorted(glob('./assets/example_images/**/*.png', recursive=True))
 example_imgs = get_example_img_list()
 
+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>
+    </div>
+</div>
+"""
+MAX_SEED = 1e7
+
 title = "## Image to 3D"
 description = "A lightweight image to 3D converter"
 
@@ -22,8 +32,34 @@ with gr.Blocks().queue() as demo:
         with gr.Column(scale=3):
             gr.Markdown("#### Image Prompt")
             image = gr.Image(sources=["upload"], label='Image', type='pil', image_mode='RGBA', height=290)
+            with gr.Column():
+                seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=1234,
+                    min_width=100,
+                )
+                with gr.Column():
+                    num_steps = gr.Slider(maximum=100,
+                                            minimum=1,
+                                            value=5,
+                                            step=1, label='Inference Steps')
+                    octree_resolution = gr.Slider(maximum=512, minimum=16, value=256, label='Octree Resolution')
+                with gr.Column():
+                    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.Column(scale=6):
             gr.Markdown("#### Generated Mesh")
+            with gr.Tabs(selected='gen_mesh_panel') as tabs_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.Column(scale=3):
             gr.Markdown("#### Image Examples")
             gr.Examples(examples=example_imgs, inputs=[image],