Update app.py

app.py

@@ -229,10 +229,9 @@ with gr.Blocks(title="🌍 ZenCtrl") as demo:
                             label="Width",
                         )
 
-                def on_generate_click(
-                    model_name, prompt, steps, strength, height, width, remove_bg, image
-                ):
-                    return call_baseten_generate(
+                def on_generate_click(model_name, prompt, steps, strength, height, width, remove_bg, image):
+                    print("🚀 on_generate_click triggered")
+                    result = call_baseten_generate(
                         image,
                         prompt,
                         steps,
@@ -242,6 +241,8 @@ with gr.Blocks(title="🌍 ZenCtrl") as demo:
                         model_name,
                         remove_bg,
                     )
+                    print("🎯 on_generate_click returning:", "Image OK" if result else "None")
+                    return result
 
                 generate_button.click(
                     fn=on_generate_click,