Update app.py

app.py

@@ -62,7 +62,6 @@
 
 
 # iface.launch()
-
 import gradio as gr
 import torch
 import numpy as np
@@ -84,9 +83,6 @@ model_path = 'dress_model.pkl'
 with open(model_path, 'rb') as f:
     G = legacy.load_network_pkl(f)['G_ema'].to(device)
 
-# Global dictionary to store user_id based on IP
-user_ids = {}
-
 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]
@@ -118,14 +114,13 @@ def style_mixing_interface(image1, image2, mix_value):
     buffer.seek(0)
     return mixed_img, buffer
 
-def send_to_backend(image_buffer, request: gr.Request):
-    user_id = user_ids.get(request.client.host)
+def send_to_backend(image_buffer, user_id):
     if not user_id:
         return "❌ user_id not found in URL."
 
     try:
         files = {'file': ('generated_image.png', image_buffer, 'image/png')}
-        url = f"https://361d-103-40-74-78.ngrok-free.app/customisation/upload/74"
+        url = f"https://361d-103-40-74-78.ngrok-free.app/customisation/upload/{user_id}"
 
         response = requests.post(url, files=files)
 
@@ -139,13 +134,13 @@ def send_to_backend(image_buffer, request: gr.Request):
 
 # --- Gradio UI ---
 with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
+    
+    user_id_state = gr.State()
 
     @iface.load
     def on_load(request: gr.Request):
-        user_id = request.query_params.get('user_id')
-        if user_id:
-            user_ids[request.client.host] = user_id
-        return
+        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.")
 
@@ -167,9 +162,10 @@ with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
         return result_image, buffer
 
     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], outputs=[save_status])
+    save_button.click(send_to_backend, inputs=[image_buffer, user_id_state], outputs=[save_status])
 
 iface.launch()
 
 
 
+