Update app.py

app.py

@@ -1,79 +1,51 @@
 import gradio as gr
 import requests
 import os
-from PIL import Image
-from io import BytesIO
 
-# Функция для генерации изображения с использованием DALL-E 3 API
-def generate_image(prompt, seed=None, sampling_method=None, cfg=None, width=None, height=None):
+# Функция для генерации изображения с помощью DALL-E 3 API
+def generate_image(prompt, seed, sampling_method, cfg, width, height):
     headers = {
-        'Authorization': f'Bearer {os.getenv("API_KEY")}',
-        'Content-Type': 'application/json',
+        "Authorization": f"Bearer {os.getenv('API_KEY')}",
+        "Content-Type": "application/json",
     }
-
     data = {
-        'prompt': prompt,
-        'n': 1,
-        'size': '1024x1024',
+        "prompt": prompt,
+        "num_images": 1,
+        "seed": seed,
+        "sampling_method": sampling_method,
+        "cfg": cfg,
+        "width": width,
+        "height": height
     }
-
-    # Добавление расширенных настроек, если они предоставлены
-    if seed is not None:
-        data['seed'] = seed
-    if sampling_method is not None:
-        data['sampling'] = sampling_method
-    if cfg is not None:
-        data['cfg'] = cfg
-    if width is not None and height is not None:
-        data['size'] = f'{width}x{height}'
-
-    response = requests.post('https://api.openai.com/v1/images/generations', headers=headers, json=data)
-    response_json = response.json()
-
-    # Проверка на наличие ошибок
-    if response.status_code != 200:
-        return "Error: " + response_json.get('error', 'Unknown error')
-
-    # Получение URL сгенерированного изображения
-    image_url = response_json['data'][0]['urls'][-1]
-
-    # Загрузка изображения и конвертация в формат, подходящий для Gradio
-    image_response = requests.get(image_url)
-    image = Image.open(BytesIO(image_response.content))
-
-    return image
-
-# CSS для скрытия футера
+    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
+    else:
+        return "Ошибка при генерации изображения"
+
+# CSS для скрытия footer
 css = """
-footer {
-    display: none;
-}
+footer { display: none; }
 """
 
 with gr.Blocks(css=css) as demo:
-    with gr.Tab("Базовые настройки"):
-        with gr.Row():
-            prompt_input = gr.Textbox(label="Prompt", lines=3, placeholder="Введите описание изображения")
-
-    with gr.Tab("Расширенные настройки"):
-        with gr.Row():
-            seed_input = gr.Number(label="Seed")
-            sampling_method_input = gr.Dropdown(label="Sampling Method", choices=["ddim", "plms"])
-            cfg_input = gr.Number(label="CFG")
-        with gr.Row():
-            width_input = gr.Number(label="Width")
-            height_input = gr.Number(label="Height")
-
-    with gr.Row():
-        generate_button = gr.Button("Генерация", variant="primary")
-        output_image = gr.Image(label="Сгенерированное изображение")
-
-    # Обработчик кнопки для генерации изображения
+    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
+        outputs=output_image,
     )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()