authentication

app.py

@@ -57,10 +57,12 @@ def generate_signature(key, did, timestamp):
     return h.hexdigest()
 
 
-def url_to_image(url, ip):
+def url_to_image(url, ip, user_id, user_name):
     headers = {
         "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
-        'X-Forwarded-For': ip
+        'X-Forwarded-For': ip,
+        'X-HfUid': user_id,
+        'X-HfUserName': user_name,
     }
     try:
         response = requests.get(url, headers=headers, timeout=30)
@@ -72,7 +74,7 @@ def url_to_image(url, ip):
     return None
 
 
-def start_task(task_id, did, ip):
+def start_task(task_id, did, ip, user_id, user_name):
     timestamp = str(int(time.time()))
     signature = generate_signature(
         key=secret_key,
@@ -85,6 +87,8 @@ def start_task(task_id, did, ip):
         'X-Signature': signature,
         'X-Forwarded-For': ip,
         'X-AppId': app_id,
+        'X-HfUid': user_id,
+        'X-HfUserName': user_name,
     }
     data = {
         "agentVersion": agent_version,
@@ -97,7 +101,7 @@ def start_task(task_id, did, ip):
     return data, msg
 
 
-def create_task(image_url, did, ip):
+def create_task(image_url, did, ip, user_id, user_name):
     timestamp = str(int(time.time()))
     signature = generate_signature(
         key=secret_key,
@@ -110,6 +114,8 @@ def create_task(image_url, did, ip):
         'X-Signature': signature,
         'X-Forwarded-For': ip,
         'X-AppId': app_id,
+        'X-HfUid': user_id,
+        'X-HfUserName': user_name,
     }
     data = {
         "agentVersion": agent_version,
@@ -121,7 +127,7 @@ def create_task(image_url, did, ip):
     return data, msg
 
 
-def save_task(image_url, show_image, task_id, location_id, did, ip):
+def save_task(image_url, show_image, task_id, location_id, did, ip, user_id, user_name):
     timestamp = str(int(time.time()))
     signature = generate_signature(
         key=secret_key,
@@ -134,6 +140,8 @@ def save_task(image_url, show_image, task_id, location_id, did, ip):
         'X-Signature': signature,
         'X-Forwarded-For': ip,
         'X-AppId': app_id,
+        'X-HfUid': user_id,
+        'X-HfUserName': user_name,
     }
     data = {
         "agentVersion": agent_version,
@@ -148,7 +156,7 @@ def save_task(image_url, show_image, task_id, location_id, did, ip):
     return data, msg
 
 
-def query_task(task_id, execution_id, did, ip):
+def query_task(task_id, execution_id, did, ip, user_id, user_name):
     timestamp = str(int(time.time()))
     signature = generate_signature(
         key=secret_key,
@@ -161,6 +169,8 @@ def query_task(task_id, execution_id, did, ip):
         'X-Signature': signature,
         'X-Forwarded-For': ip,
         'X-AppId': app_id,
+        'X-HfUid': user_id,
+        'X-HfUserName': user_name,
     }
     data = {
         "agentVersion": agent_version,
@@ -173,7 +183,7 @@ def query_task(task_id, execution_id, did, ip):
     return data, msg
 
 
-def upload_image(image, upload_type, did, ip):
+def upload_image(image, upload_type, did, ip, user_id, user_name):
     if image is None:
         return None
     if upload_type == 'image':
@@ -198,6 +208,8 @@ def upload_image(image, upload_type, did, ip):
             'X-Signature': signature,
             'X-Forwarded-For': ip,
             'X-AppId': app_id,
+            'X-HfUid': user_id,
+            'X-HfUserName': user_name,
         }
         response = requests.post(base_url + upload_url, files=files, headers=headers)
         data, msg = parse_response(response)
@@ -212,16 +224,16 @@ def load_description(file_path):
 
 def check_login_status(headers):
     if not headers:
-        return False
+        return None, None
 
     try:
         text = headers.get(login_status_key)
         if not text or "." not in text:
-            return False
+            return None, None
 
         infos = text.split(".")
         if len(infos) < 2:
-            return False
+            return None, None
 
         info = infos[1]
         cover = len(info) % 4
@@ -234,22 +246,22 @@ def check_login_status(headers):
 
         data = datas.get(login_info_key)
         if not data:
-            return False
+            return None, None
 
         user_id = data.get("_id")
         user_name = data.get("user")
-        return bool(user_id and user_name)
+        return user_id, user_name
 
     except Exception as e:
         print(f"An error occurred: {repr(e)}")
-        return False
+        return None, None
 
 
 def generate_image(main_image, background_image, did, request: gr.Request):
     if not did:
         did = str(uuid.uuid4())
-    login_status = check_login_status(request.request.headers)
-    if not login_status:
+    user_id, user_name = check_login_status(request.request.headers)
+    if not user_id or not user_name:
         m = "Please log in to your Hugging Face account to use the features of this application."
         return gr.Warning(m), did
     if main_image is None or background_image is None:
@@ -265,7 +277,9 @@ def generate_image(main_image, background_image, did, request: gr.Request):
         image=main_image,
         upload_type='image',
         did=did,
-        ip=client_ip
+        ip=client_ip,
+        user_id=user_id,
+        user_name=user_name
     )
     if not upload_image_data:
         return gr.Warning(upload_image_msg), did
@@ -276,7 +290,9 @@ def generate_image(main_image, background_image, did, request: gr.Request):
     create_task_data, create_task_msg = create_task(
         image_url=image_url,
         did=did,
-        ip=client_ip
+        ip=client_ip,
+        user_id=user_id,
+        user_name=user_name
     )
     if not create_task_data:
         return gr.Warning(create_task_msg), did
@@ -289,7 +305,9 @@ def generate_image(main_image, background_image, did, request: gr.Request):
         image=background_image,
         upload_type='background_image',
         did=did,
-        ip=client_ip
+        ip=client_ip,
+        user_id=user_id,
+        user_name=user_name
     )
     if not upload_image_data:
         return gr.Warning(upload_image_msg), did
@@ -300,7 +318,9 @@ def generate_image(main_image, background_image, did, request: gr.Request):
         task_id=task_id,
         location_id=upload_image_data,
         did=did,
-        ip=client_ip
+        ip=client_ip,
+        user_id=user_id,
+        user_name=user_name
     )
     if not save_task_data:
         return gr.Warning(save_task_msg), did
@@ -311,7 +331,9 @@ def generate_image(main_image, background_image, did, request: gr.Request):
     start_task_data, start_task_msg = start_task(
         task_id=save_task_id,
         did=did,
-        ip=client_ip
+        ip=client_ip,
+        user_id=user_id,
+        user_name=user_name
     )
     if not start_task_data:
         return gr.Warning(start_task_msg), did
@@ -326,7 +348,9 @@ def generate_image(main_image, background_image, did, request: gr.Request):
             task_id=save_task_id,
             execution_id=execution_id,
             did=did,
-            ip=client_ip
+            ip=client_ip,
+            user_id=user_id,
+            user_name=user_name
         )
         if not query_task_data:
             return gr.Warning(query_task_msg), did
@@ -343,7 +367,7 @@ def generate_image(main_image, background_image, did, request: gr.Request):
         elif status == "Success" or status == "Blocked":
             img = results[0].get("image")
             if img and str(img).strip() != "":
-                return url_to_image(img, ip=client_ip), did
+                return url_to_image(img, ip=client_ip, user_id=user_id, user_name=user_name), did
         end_time = int(time.time())
         if end_time - start_time > 3600:
             m = 'Query task timeout.'

assets/title.html

@@ -92,7 +92,7 @@
         <div class="content-text">
             Explore our other applications on Hugging Face Spaces:
             <span class="nav-links">
-                <a href="https://huggingface.co/spaces/WeShopAI/WeShopAI-Pose-Generator">WeShopAI Pose Generator</a>
+                <a href="https://huggingface.co/spaces/WeShopAI/WeShopAI-Fashion-Model-Pose-Change">WeShopAI Fashion Model Pose Change</a>
                 <a href="https://huggingface.co/spaces/WeShopAI/WeShopAI-Bad-Hand-Fixer">WeShopAI Bad Hand Fixer</a>
             </span>
         </div>