Update app.py

app.py

@@ -1,24 +1,27 @@
-from flask import Flask, request, jsonify
-import openai
+import gradio as gr
+import os
+import requests
 
-# Инициализация Flask приложения
-app = Flask(__name__)
+def generate_image(prompt, auth_key):
+    headers = {
+        "Authorization": f"Bearer {auth_key}",
+        "Content-Type": "application/json"
+    }
+    data = {
+        "description": prompt
+    }
+    response = requests.post("http://localhost:5000/generate-image", json=data, headers=headers)
+    if response.status_code != 200:
+        raise gr.Error(f"Ошибка при генерации изображения: {response.json().get('error')}")
+    return response.json()['image_url']
 
-@app.route('/generate-image', methods=['POST'])
-def generate_image():
-    # Проверка наличия авторизационного ключа в заголовках запроса
-    auth_key = request.headers.get('Authorization')
-    openai.api_key = auth_key
+with gr.Blocks() as demo:
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Описание изображения", lines=2)
+        auth_input = gr.Textbox(label="Ключ авторизации", type="password")
+        submit_button = gr.Button("Сгенерировать изображение")
+        image_output = gr.Image(label="Сгенерированное изображение")
     
-    # Получение описания изображения из тела запроса
-    description = request.json.get('description')
-    if not description:
-        return jsonify({"error": "No description provided"}), 400
+    submit_button.click(fn=generate_image, inputs=[prompt_input, auth_input], outputs=image_output)
 
-    response = openai.Image.create(prompt=description, n=1, model="dall-e-3", size="1024x1024", quality="hd")
-    image_url = response['data'][0]['url']  # Пример получения URL изображения
-
-    return jsonify({"image_url": image_url})
-
-if __name__ == '__main__':
-    app.run(debug=True)
+demo.launch()