Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import requests
 import os
+from PIL import Image
+import io
 
 # Функция для генерации изображения с помощью DALL-E 3 API
 def generate_image(prompt, seed, sampling_method, cfg, width, height):
@@ -19,33 +21,11 @@ def generate_image(prompt, seed, sampling_method, cfg, width, height):
     }
     response = requests.post("https://api.openai.com/v1/engines/dalle-3/images", headers=headers, json=data)
     if response.status_code == 200:
-        image_url = response.json()["data"][0]["url"]
-        return image_url
+        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 "Ошибка при генерации изображения"
+        return None  # Возвращаем None в случае ошибки
 
-# CSS для скрытия footer
-css = """
-footer { display: none; }
-"""
-
-with gr.Blocks(css=css) as demo:
-    with gr.Tabs():
-        with gr.TabItem("Базовые настройки"):
-            prompt_input = gr.Textbox(label="Введите текст для генерации изображения", lines=3, placeholder="Введите текст...")
-        with gr.TabItem("Расширенные настройки"):
-            seed_input = gr.Number(label="Сид", value=0)
-            sampling_method_input = gr.Dropdown(label="Метод семплирования", choices=["Default", "DDIM", "PLMS"], value="Default")
-            cfg_input = gr.Slider(label="CFG", minimum=1, maximum=20, value=7)
-            width_input = gr.Number(label="Ширина", value=512)
-            height_input = gr.Number(label="Высота", value=512)
-    generate_button = gr.Button("Генерация", variant="primary")
-    output_image = gr.Image()
-    
-    generate_button.click(
-        generate_image,
-        inputs=[prompt_input, seed_input, sampling_method_input, cfg_input, width_input, height_input],
-        outputs=output_image,
-    )
-
-demo.launch()
+# Остальная часть кода без изменений...