Enhance app.py UI components by adding show_label and show_download_button options for better user experience. Refactor background choice and color picker for improved readability. Update demo launch settings to enable API access and error display.

app.py

@@ -190,44 +190,80 @@ with gr.Blocks() as demo:
                 image_input = gr.Image(
                     label="Image input",
                     image_mode="RGBA",
-                    sources="upload",
+                    sources=["upload"],
                     type="pil",
+                    show_label=True,
+                    show_download_button=False
+                )
+                processed_image = gr.Image(
+                    label="Processed Image", 
+                    interactive=False, 
+                    type="pil", 
+                    image_mode="RGB",
+                    show_label=True,
+                    show_download_button=False
                 )
-                processed_image = gr.Image(label="Processed Image", interactive=False, type="pil", image_mode="RGB")
             with gr.Row():
                 with gr.Column():
                     with gr.Row():
-                        background_choice = gr.Radio([
-                            "Alpha as mask",
-                            "Auto Remove background"
-                        ], value="Auto Remove background",
-                        label="Background choice")
-                    back_ground_color = gr.ColorPicker(label="Background Color", value="#7F7F7F")
+                        background_choice = gr.Radio(
+                            choices=["Alpha as mask", "Auto Remove background"],
+                            value="Auto Remove background",
+                            label="Background choice",
+                            show_label=True
+                        )
+                    back_ground_color = gr.ColorPicker(
+                        label="Background Color", 
+                        value="#7F7F7F",
+                        show_label=True
+                    )
                     foreground_ratio = gr.Slider(
                         label="Foreground Ratio",
                         minimum=0.5,
                         maximum=1.0,
                         value=1.0,
                         step=0.05,
+                        show_label=True
                     )
 
                 with gr.Column():
-                    seed = gr.Number(value=1234, label="Seed", precision=0)
-                    guidance_scale = gr.Number(value=5.5, minimum=3, maximum=10, label="Guidance scale")
-                    step = gr.Number(value=30, minimum=30, maximum=100, label="Sample steps", precision=0)
+                    seed = gr.Number(
+                        value=1234, 
+                        label="Seed", 
+                        precision=0,
+                        show_label=True
+                    )
+                    guidance_scale = gr.Number(
+                        value=5.5, 
+                        minimum=3.0, 
+                        maximum=10.0, 
+                        label="Guidance scale",
+                        show_label=True
+                    )
+                    step = gr.Number(
+                        value=30, 
+                        minimum=30, 
+                        maximum=100, 
+                        label="Sample steps", 
+                        precision=0,
+                        show_label=True
+                    )
             text_button = gr.Button("Generate 3D shape")
-            if os.path.exists("examples") and os.listdir("examples"):
-                gr.Examples(
-                    examples=[os.path.join("examples", i) for i in os.listdir("examples") if i.lower().endswith(('.png', '.jpg', '.jpeg'))],
-                    inputs=[image_input],
-                    examples_per_page=20,
-                )
+            # if os.path.exists("examples") and os.listdir("examples"):
+            #     gr.Examples(
+            #         examples=[os.path.join("examples", i) for i in os.listdir("examples") if i.lower().endswith(('.png', '.jpg', '.jpeg'))],
+            #         inputs=[image_input],
+            #         examples_per_page=20,
+            #     )
         with gr.Column():
             image_output = gr.Image(interactive=False, label="Output RGB image")
             xyz_output = gr.Image(interactive=False, label="Output CCM image")
             output_model = gr.Model3D(
                 label="Output GLB",
                 interactive=False,
+                type="filepath",
+                show_label=True,
+                show_download_button=True
             )
             gr.Markdown("Note: Ensure that the input image is correctly pre-processed into a grey background, otherwise the results will be unpredictable.")
 
@@ -254,4 +290,11 @@ with gr.Blocks() as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    # Ensure API endpoints are properly launched
+    demo.launch(
+        enable_queue=True,
+        api_open=True,  # Enable API access
+        show_api=True,  # Show API documentation
+        show_error=True,
+        allowed_paths=["examples"]  # Allow access to examples directory
+    )