Create app.py

app.py

@@ -0,0 +1,91 @@
+import asyncio
+asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
+from quart import Quart, request, jsonify
+import random
+from g4f import ChatCompletion
+
+import json
+import ast
+
+app = Quart(__name__)
+
+def decode_unicode_escape(input_text):
+    try:
+        decoded_json = json.loads('{"content": ' + input_text + '}')
+        content = decoded_json["content"].encode("utf-8").decode("unicode-escape")
+        return content.encode().decode('unicode_escape')
+    except Exception as e:
+        return str(e)
+
+async def get_completion(model, prompt):
+    response = await ChatCompletion.create_async(
+        model=model,
+        messages = ast.literal_eval(f"[{prompt}]")
+        #messages=[{"role": "user", "content": prompt}],
+    )
+    return response
+
+@app.route('/chat/completions', methods=['POST'])
+async def chat_completions():
+    try:
+        app.config['QUART_ROUTE_TIMEOUT'] = 180  # Устанавливаем таймаут в 180 секунд
+        app.config["JSON_AS_ASCII"] = False
+        app.config["JSONIFY_MIMETYPE"] = "application/json; charset=utf-8"
+
+        # Получаем данные из POST-запроса
+        request_data = await request.get_data()
+        request_data_str = request_data.decode('utf-8')
+
+        # Извлекаем значение из JSON данных
+        request_json = json.loads(request_data_str)
+        user_message = request_json.get("message", "")
+        model = request_json.get("model", "")
+        password = request_json.get("password", "")
+
+        # Проверяем код пароля
+        if password != "P@ssw0rd":
+            return jsonify({"error": "Unauthorized", "message": "Неверный код пароля"}), 401
+
+        response = await get_completion(model, user_message)
+
+        completion_id = random.randint(1, 100)
+        completion_timestamp = '2022-01-01 12:00:00'
+        data = {
+            "id": f"chatcmpl-{completion_id}",
+            "object": "chat.completion",
+            "created": completion_timestamp,
+            "model": model,
+            "choices": [
+                {
+                    "index": 0,
+                    "message": {
+                        "role": "assistant",
+                        "content": response,
+                    },
+                    "finish_reason": "stop",
+                }
+            ],
+            "usage": {
+                "prompt_tokens": None,
+                "completion_tokens": None,
+                "total_tokens": None,
+            }
+        }
+
+        headers = {'Content-Type': 'application/json; charset=utf-8'} 
+
+        # Преобразуйте данные в JSON и укажите кодировку UTF-8
+        response = json.dumps(data, ensure_ascii=False).encode('utf-8')
+
+        # Создайте объект Response с указанием заголовков и кодировки
+        return app.response_class(response, content_type='application/json; charset=utf-8')
+
+    except Exception as e:
+        # Вывести ошибку в консоль для отладки
+        # print(f"Error: {str(e)}")
+        # Вернуть ошибку 500 с сообщением об ошибке
+        return jsonify({"error": "Internal Server Error", "message": str(e)}), 500
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000, debug=True)