Update app.py

app.py

@@ -62,7 +62,6 @@
 
 
 # iface.launch()
-
 import gradio as gr
 import torch
 import numpy as np
@@ -72,7 +71,6 @@ import legacy
 import torch_utils
 import requests
 import io
-from http.cookies import SimpleCookie
 import jwt  # PyJWT
 import warnings
 
@@ -86,6 +84,9 @@ 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 = {}
+
 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,18 +119,11 @@ def style_mixing_interface(image1, image2, mix_value):
     return mixed_img, buffer
 
 def send_to_backend(image_buffer, request: gr.Request):
-    cookie_header = request.headers.get('cookie', '')
-    cookies = SimpleCookie()
-    cookies.load(cookie_header)
-
-    jwt_token_cookie = cookies.get("access_token")
-    if not jwt_token_cookie:
-        return "❌ JWT token not found in cookies."
-
-    token = jwt_token_cookie.value
+    token = user_token.get(request.client.host)
+    if not token:
+        return "❌ JWT token not found in URL or session."
 
     try:
-        # Decode JWT (no verification assuming it's a trusted local token for this case)
         payload = jwt.decode(token, options={"verify_signature": False})
         user_id = payload.get("user_id")
         if not user_id:
@@ -153,6 +147,14 @@ def send_to_backend(image_buffer, request: gr.Request):
 
 # --- Gradio UI ---
 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
+        return
+
     gr.Markdown("## Style Mixing for Clothing Design\nUpload two projected clothing images and mix their styles.")
 
     with gr.Row():
@@ -179,3 +181,4 @@ iface.launch()
 
 
 
+