Update app.py

app.py

@@ -1,12 +1,49 @@
-import os
 import gradio as gr
-from gradio.external import load_interface
+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):
+    headers = {
+        'Authorization': f'Bearer {os.getenv("API_KEY")}',
+        'Content-Type': 'application/json',
+    }
+
+    data = {
+        'prompt': prompt,
+        'n': 1,
+        'size': '1024x1024',
+    }
+
+    # Добавление расширенных настроек, если они предоставлены
+    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')
 
-# Загрузка интерфейса DALL-E с использованием API ключа
-api_key = os.getenv("API_KEY")
-dalle = load_interface("openai/dalle", api_key=api_key)
+    # Получение URL сгенерированного изображения
+    image_url = response_json['data'][0]['urls'][-1]
 
-# CSS для скрытия footer
+    # Загрузка изображения и конвертация в формат, подходящий для Gradio
+    image_response = requests.get(image_url)
+    image = Image.open(BytesIO(image_response.content))
+
+    return image
+
+# CSS для скрытия футера
 css = """
 footer {
     display: none;
@@ -16,31 +53,27 @@ footer {
 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("Расширенные настройки"):
+            prompt_input = gr.Textbox(label="Prompt", lines=3, placeholder="Введите описание изображения")
+
+    with gr.Tab("Расширенные нас��ройки"):
+        with gr.Row():
+            seed_input = gr.Number(label="Seed", placeholder="Введите seed (целое число)")
+            sampling_method_input = gr.Dropdown(label="Sampling Method", choices=["ddim", "plms"], placeholder="Выберите метод семплирования")
+            cfg_input = gr.Number(label="CFG", placeholder="Введите CFG (число)")
         with gr.Row():
-            seed_input = gr.Number(label="Seed", value=0)
-            sampling_method_input = gr.Dropdown(label="Sampling Method", choices=["random", "DDIM", "DPM++", "DPM++_ancestral", "Heuristic"], value="random")
-            cfg_input = gr.Slider(label="CFG", minimum=1, maximum=20, value=7)
-            width_input = gr.Number(label="Width", value=256)
-            height_input = gr.Number(label="Height", value=256)
-    
+            width_input = gr.Number(label="Width", placeholder="Введите ширину изображения")
+            height_input = gr.Number(label="Height", placeholder="Введите высоту изображения")
+
     with gr.Row():
         generate_button = gr.Button("Генерация", variant="primary")
-    
-    output_image = gr.Image(label="Сгенерированное изображение")
-
-    def generate_image(prompt, seed, sampling_method, cfg, width, height):
-        # Генерация изображения с помощью DALL-E API
-        response = dalle([prompt], num_images=1, seed=seed, sampling=sampling_method, cfg_scale=cfg, width=width, height=height)
-        image_url = response[0][0]["url"]
-        return image_url
+        output_image = gr.Image(label="Сгенерированное изображение")
 
+    # Обработчик кнопки для генерации изображения
     generate_button.click(
         generate_image,
         inputs=[prompt_input, seed_input, sampling_method_input, cfg_input, width_input, height_input],
         outputs=output_image
     )
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()