Update app.py

app.py

@@ -179,10 +179,6 @@
 
 
 
-
-
-
-
 import gradio as gr
 import torch
 import numpy as np
@@ -239,14 +235,16 @@ def style_mixing_interface(image1, image2, mix_value):
 def send_to_backend(image_buffer, user_id):
     if not user_id:
         return "❌ user_id not found."
+    if image_buffer is None:
+        return "⚠️ No image generated. Please mix styles first."
 
     try:
-        # ✅ Convert BytesIO to raw bytes before sending
+        # Convert BytesIO to raw bytes before sending
         file_bytes = image_buffer.getvalue()
 
         files = {'file': ('generated_image.png', file_bytes, 'image/png')}
 
-        # ✅ Update with your actual ngrok or server URL
+        # Update with your actual ngrok or server URL
         url = f"https://5a4d-103-40-74-78.ngrok-free.app/customisation/upload/{user_id}"
 
         response = requests.post(url, files=files)
@@ -288,7 +286,8 @@ with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
         result_image, buffer = style_mixing_interface(image1, image2, mix_value)
         return result_image, buffer
 
-    mix_slider.change(mix_and_store, inputs=[image1_input, image2_input, mix_slider], outputs=[output_image, image_buffer])
+    mix_button = gr.Button("Mix Styles")
+    mix_button.click(mix_and_store, inputs=[image1_input, image2_input, mix_slider], outputs=[output_image, image_buffer])
     save_button.click(send_to_backend, inputs=[image_buffer, user_id_state], outputs=[save_status])
 
 iface.launch()