Update main.py

main.py

@@ -1,8 +1,40 @@
 import os
 import gradio as gr
 from openai import OpenAI
+from flask import Flask, request, jsonify
 client = OpenAI(api_key=os.getenv("API_KEY"))
 
+
+# Инициализация Flask приложения
+app = Flask(__name__)
+
+@app.route('/generate-image', methods=['POST'])
+def generate_image():
+    # Проверка наличия авторизационного ключа в заголовках запроса
+    auth_key = request.headers.get('Authorization')
+    client = OpenAI(api_key=auth_key)
+    
+    # Получение описания изображения из тела запроса
+    description = request.json.get('description')
+    if not description:
+        return jsonify({"error": "No description provided"}), 400
+
+    response = client.images.generate(
+    model="dall-e-3",
+    prompt=prompt,
+    size="1024x1024",
+    quality="hd",
+    n=1,
+    )
+    image_url = response.data[0].url  # Пример получения URL изображения
+
+    return jsonify({"image_url": image_url})
+
+if __name__ == '__main__':
+    app.run(debug=True)
+
+
+
 def generate_image(prompt, code):
     if code != os.getenv("code"):
         raise gr.Error("❗ Не верный ключ!")