Fix login

app.py

@@ -28,6 +28,8 @@ agent_name = os.getenv('agent_name')
 app_id = os.getenv('app_id')
 login_status_key = os.getenv('login_status_key')
 login_info_key = os.getenv('login_info_key')
+login_status_key2 = os.getenv('login_status_key2')
+login_mark_key = os.getenv('login_mark_key')
 
 
 def parse_response(response):
@@ -223,14 +225,17 @@ def load_description(file_path):
 
 
 def check_login_status(headers):
-    print(f"headers={headers}")
     if not headers:
         return None, None
 
     try:
+        login_status = 1
         text = headers.get(login_status_key)
         if not text or "." not in text:
-            return None, None
+            text = headers.get(login_status_key2)
+            login_status = 2
+            if not text or "." not in text:
+                return None, None
 
         infos = text.split(".")
         if len(infos) < 2:
@@ -244,15 +249,20 @@ def check_login_status(headers):
         decoded_bytes = base64.b64decode(info)
         decoded_str = decoded_bytes.decode('utf-8')
         datas = json.loads(decoded_str)
-
-        data = datas.get(login_info_key)
-        if not data:
+        if login_status == 1:
+            data = datas.get(login_info_key)
+            if not data:
+                return None, None
+            user_id = data.get("_id")
+            user_name = data.get("user")
+            return user_id, user_name
+        elif login_status == 2:
+            user_id = datas.get("uuid")
+            user_name = datas.get("user")
+            return user_id, user_name
+        else:
             return None, None
 
-        user_id = data.get("_id")
-        user_name = data.get("user")
-        return user_id, user_name
-
     except Exception as e:
         print(f"An error occurred: {repr(e)}")
         return None, None
@@ -261,16 +271,13 @@ def check_login_status(headers):
 def generate_image(main_image, background_image, did, request: gr.Request):
     if not did:
         did = str(uuid.uuid4())
-    print(f"request.request.__dict__={request.request.__dict__}")
-    print(f"request.request.cookies={request.request.cookies}")
-    print(f"request.request.scope={request.request.scope }")
-    print(f"request.request._cookies ={request.request._cookies}")
-    print(f"request.username={request.username}")
-    print(f"request.session_hash={request.session_hash}")
     user_id, user_name = check_login_status(request.request.headers)
+    print(f"user_id = {user_id}")
+    print(f"user_name = {user_name}")
     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
+    user_id = f"{login_mark_key}{user_id}"
     if main_image is None or background_image is None:
         m = "Please upload both the main image and the background reference image before generating."
         return gr.Warning(m), did