Update app.py

app.py

@@ -3,14 +3,26 @@ import requests
 import json
 import time
 import random
+import logging
 
 app = Flask(__name__)
 
+# 配置日志
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+# 添加 StreamHandler 以实时输出日志到控制台
+console_handler = logging.StreamHandler()
+console_handler.setLevel(logging.INFO)
+formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+console_handler.setFormatter(formatter)
+logger.addHandler(console_handler)
+
 @app.route('/')
 def index():
-    return "text-to-image with siliconflow", 200
+    return "flux2api with siliconflow", 200
 
-@app.route('/ai/v1/chat/completions', methods=['POST'])
+@app.route('/v1/chat/completions', methods=['POST'])
 def handle_request():
     try:
         body = request.json
@@ -43,6 +55,8 @@ def handle_request():
                 selected_token = tokens[0].strip()
             # 重新格式化为 "Bearer 随机选择的token"
             selected_token = f"Bearer {selected_token}"
+            # 记录命中的 token
+            logger.info(f"Request hit token: {selected_token}")
         else:
             return jsonify({"error": "Unauthorized: Missing Authorization header"}), 401