Update app.py

app.py

@@ -106,7 +106,6 @@ def mix_styles(image1_path, image2_path, styles_to_mix):
     mixed_image = Image.fromarray(image[0], 'RGB')
     return mixed_image
 
-# --- Step 1: Mixing and preparing the image ---
 def style_mixing_interface(image1, image2, mix_value):
     if image1 is None or image2 is None:
         return None, None
@@ -118,7 +117,6 @@ def style_mixing_interface(image1, image2, mix_value):
     buffer.seek(0)
     return mixed_img, buffer
 
-# --- Step 2: Extract user_id from JWT cookie and upload image ---
 def send_to_backend(image_buffer, request: gr.Request):
     cookie_header = request.headers.get('cookie', '')
     cookies = SimpleCookie()
@@ -153,7 +151,7 @@ def send_to_backend(image_buffer, request: gr.Request):
     except Exception as e:
         return f"⚠️ Error: {str(e)}"
 
-# --- Gradio Interface ---
+# --- Gradio UI ---
 with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
     gr.Markdown("## Style Mixing for Clothing Design\nUpload two projected clothing images and mix their styles.")
 
@@ -163,10 +161,11 @@ with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
 
     mix_slider = gr.Slider(label="Style Mixing Strength (Layers 0 to N)", minimum=0, maximum=9, step=1, value=5)
 
-    output_image = gr.Image(label="Mixed Clothing Design")
-    image_buffer = gr.State()
+    with gr.Row():
+        output_image = gr.Image(label="Mixed Clothing Design")
+        save_button = gr.Button("Download & Save to Database")
 
-    save_button = gr.Button("Download & Save to Database")
+    image_buffer = gr.State()
     save_status = gr.Textbox(label="Save Status", interactive=False)
 
     def mix_and_store(image1, image2, mix_value):