Upload 7 files

app.py

@@ -337,6 +337,36 @@ def get_model_map(session, cookies, session_token):
         raise
 
 
+# 获取Hugging Face Space URL
+def get_space_url():
+    # 尝试从环境变量获取
+    space_url = os.environ.get("SPACE_URL")
+    if space_url:
+        return space_url
+    
+    # 如果SPACE_URL不存在，尝试从SPACE_ID构建
+    space_id = os.environ.get("SPACE_ID")
+    if space_id:
+        username, space_name = space_id.split("/")
+        # 将空间名称中的下划线替换为连字符
+        # 注意：Hugging Face生成的URL会自动将空间名称中的下划线(_)替换为连字符(-)
+        # 例如："abacus_chat_proxy" 会变成 "abacus-chat-proxy"
+        space_name = space_name.replace("_", "-")
+        return f"https://{username}-{space_name}.hf.space"
+    
+    # 如果以上都不存在，尝试从单独的用户名和空间名构建
+    username = os.environ.get("SPACE_USERNAME")
+    space_name = os.environ.get("SPACE_NAME")
+    if username and space_name:
+        # 将空间名称中的下划线替换为连字符
+        # 同上，Hugging Face会自动进行此转换
+        space_name = space_name.replace("_", "-")
+        return f"https://{username}-{space_name}.hf.space"
+    
+    # 默认返回None
+    return None
+
+
 def init_session():
     get_password()
     global USER_NUM, MODELS, USER_DATA
@@ -374,12 +404,6 @@ def update_cookie(session, cookies):
     return cookies
 
 
-# 获取空间URL
-SPACE_URL = get_space_url()
-if SPACE_URL:
-    print(f"Space URL: {SPACE_URL}")
-    print("注意：Hugging Face生成的URL会自动将空间名称中的下划线(_)替换为连字符(-)")
-
 # 初始化会话
 init_session()
 
@@ -1325,34 +1349,12 @@ def set_delete_chat():
         return jsonify({"success": False, "error": str(e)}), 500
 
 
-# 获取Hugging Face Space URL
-def get_space_url():
-    # 尝试从环境变量获取
-    space_url = os.environ.get("SPACE_URL")
-    if space_url:
-        return space_url
-    
-    # 如果SPACE_URL不存在，尝试从SPACE_ID构建
-    space_id = os.environ.get("SPACE_ID")
-    if space_id:
-        username, space_name = space_id.split("/")
-        # 将空间名称中的下划线替换为连字符
-        # 注意：Hugging Face生成的URL会自动将空间名称中的下划线(_)替换为连字符(-)
-        # 例如："abacus_chat_proxy" 会变成 "abacus-chat-proxy"
-        space_name = space_name.replace("_", "-")
-        return f"https://{username}-{space_name}.hf.space"
-    
-    # 如果以上都不存在，尝试从单独的用户名和空间名构建
-    username = os.environ.get("SPACE_USERNAME")
-    space_name = os.environ.get("SPACE_NAME")
-    if username and space_name:
-        # 将空间名称中的下划线替换为连字符
-        # 同上，Hugging Face会自动进行此转换
-        space_name = space_name.replace("_", "-")
-        return f"https://{username}-{space_name}.hf.space"
-    
-    # 默认返回None
-    return None
+# 获取空间URL
+SPACE_URL = get_space_url()
+if SPACE_URL:
+    print(f"Space URL: {SPACE_URL}")
+    print("注意：Hugging Face生成的URL会自动将空间名称中的下划线(_)替换为连字符(-)")
+
 
 # 定期保存token使用统计
 def auto_save_stats():