Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import openai
 import os
+import requests
+from PIL import Image
+from io import BytesIO
 
 # Получаем API ключ из переменных среды
 api_key = os.getenv("API_KEY")
@@ -9,16 +12,27 @@ api_key = os.getenv("API_KEY")
 if api_key is None:
     raise ValueError("API_KEY не найден. Установите переменную среды API_KEY с вашим ключом.")
 
+openai.api_key = api_key
+
 # Функция для генерации изображения с помощью DALL-E 3 API
 def generate_image(prompt):
-    openai.api_key = api_key
-    
-    # Выполнение запроса к API для генерации изображения
+    # Создаем задачу для генерации изображения
     response = openai.Image.create(prompt=prompt, n=1)
-    image_url = response['data'][0]['url']
-    
-    # Возврат URL изображения для отображения
-    return image_url
+
+    # Получаем файловый ID из ответа
+    file_id = response['data'][0]['id']
+
+    # Получаем список изображений для данного файла
+    images = openai.Image.list(file_id=file_id)
+
+    # Получаем первое изображение из списка
+    image_url = images['data'][0]['url']
+
+    # Загружаем изображение по URL и конвертируем в формат, пригодный для Gradio
+    response = requests.get(image_url)
+    image = Image.open(BytesIO(response.content))
+
+    return image
 
 with gr.Blocks() as demo:
     gr.Markdown("# Генератор изображений DALL-E 3")
@@ -27,13 +41,12 @@ with gr.Blocks() as demo:
         with gr.TabItem("Базовые настройки"):
             with gr.Column():
                 prompt_input = gr.Textbox(label="Введите ваш запрос (prompt)", placeholder="Например: 'собака в космическом костюме'")
+                generate_button = gr.Button("Сгенерировать изображение")
+                output_image = gr.Image(label="Сгенерированное изображение")
+                generate_button.click(fn=generate_image, inputs=prompt_input, outputs=output_image)
         
         with gr.TabItem("Расширенные настройки"):
             gr.Markdown("## Скоро...")
             # Здесь могут быть добавлены дополнительные настройки
-        
-        generate_button = gr.Button("Сгенерировать изображение")
-        output_image = gr.Image(label="Сгенерированное изображение")
-        generate_button.click(fn=generate_image, inputs=prompt_input, outputs=output_image)
-        
+
 demo.launch()