Refactor Gradio interface to utilize Blocks for improved layout and organization, enhancing user interaction with dedicated input and output sections while maintaining minimal launch settings.

app.py

@@ -212,48 +212,61 @@ def generate(image, bg_choice, fg_ratio, bg_color, seed_val, guidance, steps):
     processed = preprocess_image(image, bg_choice, fg_ratio, bg_color)
     return gen_image(processed, seed_val, guidance, steps)
 
-# Create a simple Interface instead of Blocks
-demo = gr.Interface(
-    fn=generate,
-    inputs=[
-        gr.Image(
-            label="Image input",
-            image_mode="RGBA",
-            sources="upload",
-            type="pil",
-        ),
-        gr.Radio(
-            ["Alpha as mask", "Auto Remove background"],
-            value="Auto Remove background",
-            label="background choice"
-        ),
-        gr.Slider(
-            label="Foreground Ratio",
-            minimum=0.5,
-            maximum=1.0,
-            value=1.0,
-            step=0.05,
-        ),
-        gr.ColorPicker(label="Background Color", value="#7F7F7F", interactive=False),
-        gr.Number(value=1234, label="seed", precision=0),
-        gr.Number(value=5.5, minimum=3, maximum=10, label="guidance_scale"),
-        gr.Number(value=30, minimum=30, maximum=100, label="sample steps", precision=0)
-    ],
-    outputs=[
-        gr.Image(interactive=False, label="Output RGB image"),
-        gr.Image(interactive=False, label="Output CCM image"),
-        gr.Model3D(label="Output GLB", interactive=False)
-    ],
-    title="CRM: Single Image to 3D Textured Mesh with Convolutional Reconstruction Model",
-    description=_DESCRIPTION,
-    examples=[[os.path.join("examples", i)] for i in os.listdir("examples")],
-    concurrency_limit=1,
-    analytics_enabled=False
-)
+# Create a Blocks interface
+with gr.Blocks(analytics_enabled=False) as demo:
+    gr.Markdown("# CRM: Single Image to 3D Textured Mesh with Convolutional Reconstruction Model")
+    gr.Markdown(_DESCRIPTION)
+    
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(
+                label="Image input",
+                image_mode="RGBA",
+                sources="upload",
+                type="pil",
+            )
+            bg_choice = gr.Radio(
+                ["Alpha as mask", "Auto Remove background"],
+                value="Auto Remove background",
+                label="background choice"
+            )
+            fg_ratio = gr.Slider(
+                label="Foreground Ratio",
+                minimum=0.5,
+                maximum=1.0,
+                value=1.0,
+                step=0.05,
+            )
+            bg_color = gr.ColorPicker(label="Background Color", value="#7F7F7F", interactive=False)
+            seed_val = gr.Number(value=1234, label="seed", precision=0)
+            guidance = gr.Number(value=5.5, minimum=3, maximum=10, label="guidance_scale")
+            steps = gr.Number(value=30, minimum=30, maximum=100, label="sample steps", precision=0)
+            generate_btn = gr.Button("Generate")
+        
+        with gr.Column():
+            output_rgb = gr.Image(interactive=False, label="Output RGB image")
+            output_ccm = gr.Image(interactive=False, label="Output CCM image")
+            output_glb = gr.Model3D(label="Output GLB", interactive=False)
+    
+    # Connect the button click event
+    generate_btn.click(
+        fn=generate,
+        inputs=[input_image, bg_choice, fg_ratio, bg_color, seed_val, guidance, steps],
+        outputs=[output_rgb, output_ccm, output_glb],
+        concurrency_limit=1
+    )
+    
+    # Add examples
+    gr.Examples(
+        examples=[[os.path.join("examples", i)] for i in os.listdir("examples")],
+        inputs=input_image
+    )
 
-# Launch with minimal parameters
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=7860,
-    show_error=True
-)
+# Launch with minimal settings
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True,
+        max_threads=1
+    )