Update app.py

app.py

@@ -1,31 +1,34 @@
-import gradio as gr
-import requests
 import os
-from PIL import Image
-import io
+import requests
+import gradio as gr
 
-# Функция для генерации изображения с помощью DALL-E 3 API
-def generate_image(prompt, seed, sampling_method, cfg, width, height):
+# Функция для генерации изображения по промпту с использованием DALL-E 3 API
+def generate_image(prompt):
+    api_key = os.getenv("API_KEY")
     headers = {
-        "Authorization": f"Bearer {os.getenv('API_KEY')}",
+        "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json",
     }
     data = {
         "prompt": prompt,
         "num_images": 1,
-        "seed": seed,
-        "sampling_method": sampling_method,
-        "cfg": cfg,
-        "width": width,
-        "height": height
     }
-    response = requests.post("https://api.openai.com/v1/engines/dalle-3/images", headers=headers, json=data)
-    if response.status_code == 200:
-        image_data = response.json()["data"][0]["url"]
-        image_response = requests.get(image_data)
-        image = Image.open(io.BytesIO(image_response.content))
-        return image
-    else:
-        return None  # Возвращаем None в случае ошибки
+    response = requests.post("https://api.openai.com/v1/images/generations", headers=headers, json=data)
+    response_json = response.json()
+    
+    # Получаем URL первого сгенерированного изображения
+    image_url = response_json["data"][0]["url"]
+    
+    return image_url
+
+# Создание интерфейса с помощью Gradio
+with gr.Blocks() as demo:
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Введите промпт для генерации изображения")
+        submit_btn = gr.Button("Сгенерировать")
+    with gr.Row():
+        image_output = gr.Image(label="Сгенерированное изображение")
+    
+    submit_btn.click(fn=generate_image, inputs=prompt_input, outputs=image_output)
 
-# Остальная часть кода без изменений...
+demo.launch()