Update app.py

app.py

@@ -63,14 +63,13 @@
 
 # iface.launch()
 
-
 import gradio as gr
 import torch
 import numpy as np
 from PIL import Image
 import os
-import legacy  # your local module
-import torch_utils  # your local module
+import legacy
+import torch_utils
 import requests
 import io
 import warnings
@@ -78,12 +77,10 @@ import warnings
 warnings.filterwarnings("ignore")
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
-# Load StyleGAN model
 model_path = 'dress_model.pkl'
 with open(model_path, 'rb') as f:
     G = legacy.load_network_pkl(f)['G_ema'].to(device)
 
-# Function to perform style mixing
 def mix_styles(image1_path, image2_path, styles_to_mix):
     image1_name = os.path.splitext(os.path.basename(image1_path))[0]
     image2_name = os.path.splitext(os.path.basename(image2_path))[0]
@@ -104,7 +101,6 @@ def mix_styles(image1_path, image2_path, styles_to_mix):
     mixed_image = Image.fromarray(image[0], 'RGB')
     return mixed_image
 
-# Gradio-friendly wrapper
 def style_mixing_interface(image1, image2, mix_value):
     if image1 is None or image2 is None:
         return None, None
@@ -116,14 +112,14 @@ def style_mixing_interface(image1, image2, mix_value):
     buffer.seek(0)
     return mixed_img, buffer
 
-# Upload to backend
 def send_to_backend(image_buffer, user_id):
     if not user_id:
-        return "❌ user_id not found in URL."
+        return "❌ user_id not found."
 
     try:
         files = {'file': ('generated_image.png', image_buffer, 'image/png')}
         url = f"https://361d-103-40-74-78.ngrok-free.app/customisation/upload/{user_id}"
+
         response = requests.post(url, files=files)
 
         if response.status_code == 201:
@@ -134,10 +130,15 @@ def send_to_backend(image_buffer, user_id):
     except Exception as e:
         return f"⚠️ Error: {str(e)}"
 
-# Gradio Interface
+# --- Gradio UI ---
 with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
+
     user_id_state = gr.State()
-    image_buffer = gr.State()
+
+    @iface.load(inputs=None, outputs=[user_id_state])
+    def on_load(request: gr.Request):
+        user_id = request.query_params.get('user_id', '')
+        return user_id
 
     gr.Markdown("## Style Mixing for Clothing Design\nUpload two projected clothing images and mix their styles.")
 
@@ -151,30 +152,15 @@ with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
         output_image = gr.Image(label="Mixed Clothing Design")
         save_button = gr.Button("Download & Save to Database")
 
+    image_buffer = gr.State()
     save_status = gr.Textbox(label="Save Status", interactive=False)
 
-    # Mix and update preview image + store in memory buffer
     def mix_and_store(image1, image2, mix_value):
         result_image, buffer = style_mixing_interface(image1, image2, mix_value)
         return result_image, buffer
 
-    mix_slider.change(
-        fn=mix_and_store,
-        inputs=[image1_input, image2_input, mix_slider],
-        outputs=[output_image, image_buffer]
-    )
-
-    save_button.click(
-        fn=send_to_backend,
-        inputs=[image_buffer, user_id_state],
-        outputs=[save_status]
-    )
-
-    # Extract user_id from URL query
-    @iface.load
-    def init_fn(request: gr.Request):
-        user_id = request.query_params.get("user_id", "")
-        return {user_id_state: user_id}
+    mix_slider.change(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()