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__)
 
@@ -20,6 +21,17 @@ def handle_request():
         if not model or not messages or len(messages) == 0:
             return jsonify({"error": "Bad Request: Missing required fields"}), 400
 
+        authorization_header = request.headers.get('Authorization')
+        if not authorization_header:
+            return jsonify({"error": "Unauthorized: Missing Authorization header"}), 401
+
+        # Extract tokens from Authorization header
+        tokens = authorization_header.split(' ')[1].split(',')
+        if len(tokens) == 1:
+            selected_token = tokens[0]
+        else:
+            selected_token = random.choice(tokens)
+
         prompt = messages[-1]['content']
         new_url = f'https://api.siliconflow.cn/v1/{model}/text-to-image'
 
@@ -34,7 +46,7 @@ def handle_request():
         headers = {
             'accept': 'application/json',
             'content-type': 'application/json',
-            'Authorization': request.headers.get('Authorization')
+            'Authorization': f'Bearer {selected_token}'
         }
 
         response = requests.post(new_url, headers=headers, json=new_request_body)