Fix login

app.py

@@ -26,6 +26,8 @@ agent_name = os.getenv('agent_name')
 
 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):
@@ -222,9 +224,13 @@ def check_login_status(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:
@@ -238,15 +244,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
@@ -259,6 +270,7 @@ def generate_image(main_image, text_description, did, request: gr.Request):
     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:
         m = "Please upload both the main image and the background reference image before generating."
         return gr.Warning(m), did