Update app.py

app.py

@@ -9,7 +9,6 @@ from flask import Flask, request, Response, stream_with_context, jsonify
 from flask_cors import CORS
 from functools import lru_cache
 from concurrent.futures import ThreadPoolExecutor
-import base64
 
 app = Flask(__name__)
 CORS(app, resources={r"/*": {"origins": "*"}})
@@ -42,6 +41,18 @@ NOTDIAMOND_URLS = [
 def get_notdiamond_url():
     return random.choice(NOTDIAMOND_URLS)
 
+# 使用LRU缓存，缓存头信息，最大容量为1
+@lru_cache(maxsize=1)
+def get_notdiamond_headers():
+    return {
+        'accept': 'text/event-stream',
+        'accept-language': 'zh-CN,zh;q=0.9',
+        'content-type': 'application/json',
+        'next-action': get_env_or_file('NEXT_ACTION', 'next_action.txt'),
+        'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
+        'cookie': get_env_or_file('COOKIES', 'cookies.txt')
+    }
+
 # 模型信息字典，包含各模型的提供商和映射
 MODEL_INFO = {
     "gpt-4-turbo-2024-04-09": {
@@ -214,54 +225,6 @@ def proxy_models():
         "data": models
     })
 
-# 缓存cookie和过期时间
-cached_cookie = None
-expires_at = 0
-
-# 获取cookie
-def get_cookie(email, password):
-    global cached_cookie, expires_at
-
-    current_time = int(time.time())
-    if cached_cookie and expires_at > current_time:
-        return cached_cookie
-
-    auth_url = "https://spuckhogycrxcbomznwo.supabase.co/auth/v1/token?grant_type=password"
-    auth_headers = {
-        'accept': '*/*',
-        'accept-language': 'zh-CN,zh;q=0.9',
-        'apikey': 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InNwdWNraG9neWNyeGNib216bndvIiwicm9sZSI6ImFub24iLCJpYXQiOjE3MDcyNDYwMzksImV4cCI6MjAyMjgyMjAzOX0.tvlGT7NZY8bijMjNIu1WhAtPnSKuDeYhtveo4DRt6xg',
-        'authorization': 'Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6InNwdWNraG9neWNyeGNib216bndvIiwicm9sZSI6ImFub24iLCJpYXQiOjE3MDcyNDYwMzksImV4cCI6MjAyMjgyMjAzOX0.tvlGT7NZY8bijMjNIu1WhAtPnSKuDeYhtveo4DRt6xg',
-        'content-type': 'application/json;charset=UTF-8',
-        'origin': 'https://chat.notdiamond.ai',
-        'priority': 'u=1, i',
-        'referer': 'https://chat.notdiamond.ai/',
-        'sec-ch-ua': '"Chromium";v="128", "Not;A=Brand";v="24", "Google Chrome";v="128"',
-        'sec-ch-ua-mobile': '?0',
-        'sec-ch-ua-platform': '"Windows"',
-        'sec-fetch-dest': 'empty',
-        'sec-fetch-mode': 'cors',
-        'sec-fetch-site': 'cross-site',
-        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
-        'x-client-info': 'supabase-ssr/0.4.0',
-        'x-supabase-api-version': '2024-01-01'
-    }
-    auth_data = {
-        "email": email,
-        "password": password,
-        "gotrue_meta_security": {}
-    }
-
-    response = requests.post(auth_url, headers=auth_headers, json=auth_data)
-    response.raise_for_status()
-    auth_response = response.json()
-
-    expires_at = auth_response['expires_at']
-    encoded_token = base64.b64encode(json.dumps(auth_response).encode('utf-8')).decode('utf-8')
-    cached_cookie = f"sb-spuckhogycrxcbomznwo-auth-token=base64-{encoded_token}"
-
-    return cached_cookie
-
 # 处理请求的API
 @app.route('/ai/v1/chat/completions', methods=['POST'])
 def handle_request():
@@ -272,17 +235,6 @@ def handle_request():
         model = MODEL_INFO.get(model_id, {}).get('mapping', model_id)
         stream = request_data.get('stream', False)
 
-        # 从请求头中获取Authorization信息
-        auth_header = request.headers.get('Authorization')
-        if not auth_header or not auth_header.startswith('Bearer '):
-            return jsonify({'error': 'Missing or invalid Authorization header'}), 401
-
-        auth_info = auth_header.split(' ')[1]
-        email, password = auth_info.split('|')
-
-        # 获取cookie
-        cookie = get_cookie(email, password)
-
         payload = {
             "messages": messages,
             "model": model,
@@ -293,15 +245,7 @@ def handle_request():
             "top_p": request_data.get('top_p', 1)
         }
 
-        headers = {
-            'accept': 'text/event-stream',
-            'accept-language': 'zh-CN,zh;q=0.9',
-            'content-type': 'application/json',
-            'next-action': get_env_or_file('NEXT_ACTION', 'next_action.txt'),
-            'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36',
-            'cookie': cookie
-        }
-
+        headers = get_notdiamond_headers()
         url = get_notdiamond_url()
         
         future = executor.submit(requests.post, url, headers=headers, json=[payload], stream=True)