Update app.py

app.py

@@ -13,8 +13,6 @@ import requests
 from flask import Flask, request, Response, jsonify, stream_with_context, render_template, redirect, session
 from curl_cffi import requests as curl_requests
 from werkzeug.middleware.proxy_fix import ProxyFix
-import get_xid
-import threading
 
 
 current_dir = os.path.dirname(os.path.abspath(__file__))
@@ -992,16 +990,6 @@ def handle_stream_response(response, model):
 
     return generate()
 
-def schedule_task():
-    while True:
-        # 创建并启动一个新线程来执行任务
-        logger.info("开始获取XID")
-        xid = get_xid.main()
-        if xid is not None:
-            logger.info(f"成功获取XID: {xid}", "Server")
-            CONFIG["XIDS"] = xid
-
-        time.sleep(3 * 60 * 60)
 
 def initialization():
     sso_array = os.getenv("SSO", "").split(',')
@@ -1016,8 +1004,6 @@ def initialization():
     if CONFIG["API"]["PROXY"]:
         logger.info(f"代理已设置: {CONFIG['API']['PROXY']}", "Server")
 
-    thread = threading.Thread(target=schedule_task)
-    thread.start()
 
     logger.info("初始化完成", "Server")
 
@@ -1057,7 +1043,16 @@ def manager():
 def get_manager_tokens():
     if not check_auth():
         return jsonify({"error": "Unauthorized"}), 401
-    return jsonify(token_manager.get_token_status_map())
+    return (jsonify(token_manager.get_token_status_map())
+
+@app.route('/manager/api/xids'))
+def set_xids():
+    if request.headers.get('Authorization') != CONFIG["ADMIN"]["PASSWORD"]:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    CONFIG["XIDS"] = request.json
+    logger.info(f"get XIDS: {CONFIG['XIDS']}", "Server")
+    return jsonify({"success": True})
 
 
 @app.route('/manager/api/add', methods=['POST'])