Refactor Gradio interface to encapsulate image processing logic within a dedicated function, enhancing error handling for image uploads and streamlining event chaining for improved clarity.

app.py

@@ -256,24 +256,32 @@ 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.")
 
-    # Chain the events directly
+    def generate(image, bg_choice, fg_ratio, bg_color, seed_val, guidance, steps):
+        if image is None:
+            raise gr.Error("No image uploaded!")
+        processed = preprocess_image(image, bg_choice, fg_ratio, bg_color)
+        return gen_image(processed, seed_val, guidance, steps)
+
     text_button.click(
-        fn=lambda x: gr.Error("No image uploaded!") if x is None else x,
-        inputs=[image_input],
-        outputs=[image_input]
-    ).then(
-        fn=preprocess_image,
-        inputs=[image_input, background_choice, foreground_ratio, back_groud_color],
-        outputs=[processed_image]
-    ).then(
-        fn=gen_image,
-        inputs=[processed_image, seed, guidance_scale, step],
-        outputs=[image_output, xyz_output, output_model]
+        fn=generate,
+        inputs=[
+            image_input,
+            background_choice,
+            foreground_ratio,
+            back_groud_color,
+            seed,
+            guidance_scale,
+            step
+        ],
+        outputs=[
+            image_output,
+            xyz_output,
+            output_model
+        ]
     )
 
-# Launch the interface
+# Launch the interface without share parameter since we're on Hugging Face Spaces
 demo.launch(
     show_error=True,
-    max_threads=1,
-    share=True
+    max_threads=1
 )