Streamline Gradio interface by chaining event actions directly, improving error handling for image uploads, and maintaining API endpoint only for the final image generation step.

app.py

@@ -256,40 +256,24 @@ with gr.Blocks() as demo:
             )
             gr.Markdown("Note: Ensure that the input image is correctly pre-processed into a grey background, otherwise the results will be unpredictable.")
 
-    inputs = [
-        processed_image,
-        seed,
-        guidance_scale,
-        step,
-    ]
-    outputs = [
-        image_output,
-        xyz_output,
-        output_model,
-    ]
-
+    # Chain the events directly
     text_button.click(
-        fn=check_input_image, 
+        fn=lambda x: gr.Error("No image uploaded!") if x is None else x,
         inputs=[image_input],
-        concurrency_limit=1,
-        api_name=None  # Disable API endpoint for this step
-    ).success(
+        outputs=[image_input]
+    ).then(
         fn=preprocess_image,
         inputs=[image_input, background_choice, foreground_ratio, back_groud_color],
-        outputs=[processed_image],
-        concurrency_limit=1,
-        api_name=None  # Disable API endpoint for this step
-    ).success(
+        outputs=[processed_image]
+    ).then(
         fn=gen_image,
-        inputs=inputs,
-        outputs=outputs,
-        concurrency_limit=1,
-        api_name="generate"  # Only enable API for the final generation step
+        inputs=[processed_image, seed, guidance_scale, step],
+        outputs=[image_output, xyz_output, output_model]
     )
 
 # Launch the interface
 demo.launch(
     show_error=True,
     max_threads=1,
-    share=True  # Required for Hugging Face Spaces
+    share=True
 )