Update app.py

app.py

@@ -2,6 +2,7 @@ from flask import Flask, request, jsonify, Response
 import requests
 import json
 import time
+import random
 
 app = Flask(__name__)
 
@@ -31,10 +32,24 @@ def handle_request():
             "guidance_scale": 1
         }
 
+        # 从传入的 Authorization 头中随机选择一个 token
+        authorization_header = request.headers.get('Authorization')
+        if authorization_header:
+            # 去掉 "Bearer " 前缀并分割 token
+            tokens = authorization_header.replace("Bearer ", "").split(',')
+            if len(tokens) > 1:
+                selected_token = random.choice(tokens).strip()
+            else:
+                selected_token = tokens[0].strip()
+            # 重新格式化为 "Bearer 随机选择的token"
+            selected_token = f"Bearer {selected_token}"
+        else:
+            return jsonify({"error": "Unauthorized: Missing Authorization header"}), 401
+
         headers = {
             'accept': 'application/json',
             'content-type': 'application/json',
-            'Authorization': request.headers.get('Authorization')
+            'Authorization': selected_token
         }
 
         response = requests.post(new_url, headers=headers, json=new_request_body)