Update app.py

app.py

@@ -4,12 +4,22 @@ import uuid
 import json
 import time
 import os
+import re
 
 _COOKIES = os.environ.get("COOKIES", "")
-print(_COOKIES)
+#_COOKIES = "cf_clearance=toCFbLwd4RZKgHIv_nY.7B1yXUkIfvh0QwWBzxpLa5w-1740730551-1.2.1.1-yasz4kMVvPBtgGz06uRNFlDE1VXXZ2McAq65AKeidqAagCe4ZQ5desj88RDdSZAZ3CG4tJZrZ7pVxTCUDHnvGy96C6_ocRUQc3.qbbsWyeLwHNCD2OZjtVqkVI5MHiNFarJv8qGd9uOriKAytgttX.vMYbEOstOnMGvY5DnjODPDUWDlOwwNrshAsghJbf8B32X9VokGduYk1FvqqPhs85Eq4zdHv83tGKFSpTidYD7FmQoxwiv77T2kyeBXX.6.4ExZiHE9DszGcWZ9_FO8Ub5U2zod0S9U0oqncQwQVXM; session_token=82673f97181bb5da073f2d8771f5b4cd9208f15e0df4c29c74fd849f729249e0"
+
+API_KEY = os.getenv("API_KEY", "linux.do")
 
 app = Flask(__name__)
 
+@app.before_request
+def check_api_key():
+    key = request.headers.get("Authorization")
+    if key != "Bearer "+API_KEY:
+        return jsonify({"success": False, "message": "Unauthorized: Invalid API key"}), 403
+
+
 @app.route('/v1/chat/completions', methods=['POST'])
 def chat_completions():
     print("start")
@@ -34,8 +44,6 @@ def chat_completions():
             "Content-Type": "application/json",
             "Cookie": _COOKIES
         }
-        print(headers)
-        print(akash_data)
         _stream=data.get('stream', True)
         # 发送请求到Akash API
         response = requests.post(
@@ -111,9 +119,25 @@ def chat_completions():
                 }
             )
         else:
-            result_json = response.json()
+            text_matches = re.findall(r'0:"(.*?)"', response.text)
+            parsed_text = "".join(text_matches)
+
             return Response(
-                response=json.dumps(result_json),
+                json.dumps({
+                    "object": "chat.completion",
+                    "created": int(time.time() * 1000),
+                    "model": data.get('model', "DeepSeek-R1"),
+                    "choices": [
+                        {
+                            "index": 0,
+                            "message": {
+                                "role": "user",
+                                "content": parsed_text
+                            },
+                            "finish_reason": "stop"
+                        }
+                    ]
+                },ensure_ascii=False),
                 status=response.status_code,
                 headers={
                     'Cache-Control': 'no-cache',
@@ -123,6 +147,7 @@ def chat_completions():
             )
     
     except Exception as e:
+        print(e)
         return jsonify({"error": str(e)}), 500
 
 if __name__ == '__main__':