Update app.py

app.py

@@ -62,6 +62,9 @@
 
 
 # iface.launch()
+
+
+
 import gradio as gr
 import torch
 import numpy as np
@@ -71,7 +74,6 @@ import legacy
 import torch_utils
 import requests
 import io
-import jwt  # PyJWT
 import warnings
 
 # Suppress deprecated torch warnings
@@ -84,8 +86,8 @@ model_path = 'dress_model.pkl'
 with open(model_path, 'rb') as f:
     G = legacy.load_network_pkl(f)['G_ema'].to(device)
 
-# Global variable to store token (Not recommended for multi-user prod use)
-user_token = {}
+# Global variable to store user_id (per IP/session) — not ideal for production
+user_ids = {}
 
 def mix_styles(image1_path, image2_path, styles_to_mix):
     image1_name = os.path.splitext(os.path.basename(image1_path))[0]
@@ -119,17 +121,11 @@ def style_mixing_interface(image1, image2, mix_value):
     return mixed_img, buffer
 
 def send_to_backend(image_buffer, request: gr.Request):
-    token = user_token.get(request.client.host)
-    if not token:
-        return "❌ JWT token not found in URL or session."
+    user_id = user_ids.get(request.client.host)
+    if not user_id:
+        return "❌ user_id not found in URL."
 
     try:
-        payload = jwt.decode(token, options={"verify_signature": False})
-        user_id = payload.get("user_id")
-        if not user_id:
-            return "❌ user_id not found in JWT."
-
-        # Prepare image upload
         files = {'file': ('generated_image.png', image_buffer, 'image/png')}
         url = f"http://localhost:3000/customisation/upload/{user_id}"
 
@@ -140,8 +136,6 @@ def send_to_backend(image_buffer, request: gr.Request):
         else:
             return f"❌ Upload failed: {response.status_code} - {response.text}"
 
-    except jwt.DecodeError:
-        return "❌ Invalid JWT token."
     except Exception as e:
         return f"⚠️ Error: {str(e)}"
 
@@ -150,9 +144,9 @@ with gr.Blocks(title="Style Mixing for Clothing Design") as iface:
 
     @iface.load
     def on_load(request: gr.Request):
-        token = request.query_params.get('token')
-        if token:
-            user_token[request.client.host] = token
+        user_id = request.query_params.get('user_id')
+        if user_id:
+            user_ids[request.client.host] = user_id
         return
 
     gr.Markdown("## Style Mixing for Clothing Design\nUpload two projected clothing images and mix their styles.")