Update app.py

app.py

@@ -136,44 +136,72 @@ def split_image(image: Image.Image) -> List[Image.Image]:
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
     ## Image to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
-    * Upload an image and click "Generate" to create a 3D asset. If the image has alpha channel, it be used as the mask. Otherwise, we use `rembg` to remove the background.
-    * If you find the generated 3D asset satisfactory, click "Extract GLB" to extract the GLB file and download it.
-    
+    * Upload an image and click "Generate" to create a 3D asset. If the image has alpha channel, it will be used as the mask. Otherwise, we use `rembg` to remove the background.
+    * If satisfied, click "Extract GLB" to download the 3D model.
     ✨New: Experimental multi-image support.
     """)
     
-    with gr.Row():
-        with gr.Column():
-            with gr.Tabs() as input_tabs:
-                with gr.Tab(label="Single Image", id=0) as single_image_input_tab:
-                    image_prompt = gr.Image(label="Image Prompt", format="png", image_mode="RGBA", type="pil", height=300)
+    with gr.Row(equal_height=False):
+        # Left column (Controls)
+        with gr.Column(scale=2, min_width=400):
+            with gr.Tabs():
+                with gr.Tab(label="Input Image"):
+                    image_prompt = gr.Image(
+                        label="Image Prompt", 
+                        format="png", 
+                        image_mode="RGBA", 
+                        type="pil", 
+                        height=300,
+                        show_label=False
+                    )
+            
+            with gr.Accordion(".Generation Settings", open=False):
+                with gr.Column(variant="panel"):
+                    seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
+                    randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
+                    
+                    with gr.Group():
+                        gr.Markdown("#### Stage 1: Structure")
+                        ss_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance", value=7.5, step=0.1)
+                        ss_sampling_steps = gr.Slider(1, 50, label="Steps", value=12, step=1)
                     
-            with gr.Accordion(label="Generation Settings", open=False):
-                seed = gr.Slider(0, MAX_SEED, label="Seed", value=0, step=1)
-                randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
-                gr.Markdown("Stage 1: Sparse Structure Generation")
-                with gr.Row():
-                    ss_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=7.5, step=0.1)
-                    ss_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
-                gr.Markdown("Stage 2: Structured Latent Generation")
-                with gr.Row():
-                    slat_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance Strength", value=3.0, step=0.1)
-                    slat_sampling_steps = gr.Slider(1, 50, label="Sampling Steps", value=12, step=1)
-                
-            generate_btn = gr.Button("Generate")
+                    with gr.Group():
+                        gr.Markdown("#### Stage 2: Detail")
+                        slat_guidance_strength = gr.Slider(0.0, 10.0, label="Guidance", value=3.0, step=0.1)
+                        slat_sampling_steps = gr.Slider(1, 50, label="Steps", value=12, step=1)
             
-            with gr.Accordion(label="GLB Extraction Settings", open=False):
-                mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)
-                texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
+            generate_btn = gr.Button("Generate 3D Asset", variant="primary", size="lg")
             
-            extract_glb_btn = gr.Button("Extract GLB", interactive=False)
+            with gr.Accordion("GLB Export Settings", open=False):
+                with gr.Column(variant="panel"):
+                    mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify Mesh", value=0.95, step=0.01)
+                    texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
+                    extract_glb_btn = gr.Button("Export GLB", interactive=False, size="lg")
 
-        with gr.Column():
-            video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
-            model_output = LitModel3D(label="Extracted GLB", exposure=10.0, height=300)
+        # Right column (Outputs)
+        with gr.Column(scale=3, min_width=600):
+            with gr.Group():
+                video_output = gr.Video(
+                    label="3D Preview", 
+                    autoplay=True, 
+                    loop=True, 
+                    height=300,
+                    show_label=False
+                )
+                model_output = LitModel3D(
+                    label="3D Model Viewer", 
+                    exposure=10.0, 
+                    height=400,
+                    clear_color=[1.0, 1.0, 1.0, 1.0]
+                )
             
             with gr.Row():
-                download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
+                download_glb = gr.DownloadButton(
+                    label="Download GLB File", 
+                    interactive=False,
+                    variant="secondary",
+                    size="lg"
+                )
 
     output_buf = gr.State()