Update app.py

app.py

@@ -1,88 +1,65 @@
 import gradio as gr
-import openai
 import os
 from PIL import Image
 import requests
 from io import BytesIO
 
-# Загрузим API ключ из переменных окружения
-openai.api_key = os.getenv("API_KEY")
+# Получаем API ключ из переменных окружения
+API_KEY = os.getenv("API_KEY")
 
-# Функция для генерации изображения
-def generate_image(prompt, seed, sampling, cfg, width, height):
-    response = openai.Image.create(
-        prompt=prompt, 
-        n=1, 
-        size=f"{width}x{height}",
-        seed=seed,
-        sampling=sampling,
-        cfg_scale=cfg
-    )
-    image_url = response['data'][0]['url']
-    image = Image.open(requests.get(image_url, stream=True).raw)
-    return image
+# Функция для генерации изображения с помощью DALL-E 3 API
+def generate_image(prompt, seed, sampling_method, cfg, width, height):
+    headers = {
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    data = {
+        "prompt": prompt,
+        "seed": seed,
+        "sampling_method": sampling_method,
+        "cfg": cfg,
+        "width": width,
+        "height": height
+    }
+    
+    response = requests.post("https://api.openai.com/v1/images/generate", headers=headers, json=data)
+    
+    if response.status_code == 200:
+        image_data = response.json()
+        # Предполагаем, что API возвращает URL изображения
+        image_url = image_data["data"]["image_url"]
+        image_response = requests.get(image_url)
+        image = Image.open(BytesIO(image_response.content))
+        return image
+    else:
+        # В случае ошибки возвращаем сообщение об ошибке
+        return f"Error: {response.text}"
 
 # Функция для улучшения качества изображения
 def upscale_image(image, upscale_factor):
-    # Преобразуем PIL Image в Bytes, чтобы отправить его обратно в Gradio UI
-    buffered = BytesIO()
-    image.save(buffered, format="JPEG")
-    img_str = buffered.getvalue()
+    headers = {
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    files = {
+        "image": image
+    }
+    data = {
+        "upscale_factor": upscale_factor
+    }
     
-    response = openai.Image.modify(
-        image=img_str,
-        n=1,
-        size=f"{image.width * upscale_factor}x{image.height * upscale_factor}"
-    )
-    upscaled_image_url = response['data'][0]['url']
-    upscaled_image = Image.open(requests.get(upscaled_image_url, stream=True).raw)
-    return upscaled_image
-
-# Создаем интерфейс с вкладками
-with gr.Blocks() as demo:
-    gr.Markdown("<style>footer { display: none; }</style>")  # Скрываем footer
-    with gr.Tabs():
-        with gr.TabItem("Базовые настройки"):
-            with gr.Row():
-                prompt_input = gr.Textbox(label="Prompt", lines=3, placeholder="Введите описание изображения")
-        with gr.TabItem("Расширенные настройки"):
-            with gr.Row():
-                seed_input = gr.Number(label="Seed", value=0)
-                sampling_input = gr.Dropdown(label="Sampling Method", choices=["ancestral", "ddim"])
-                cfg_input = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7)
-                width_input = gr.Number(label="Width", value=256)
-                height_input = gr.Number(label="Height", value=256)
-        with gr.TabItem("Настройки апскейлинга"):
-            with gr.Row():
-                upscale_factor_input = gr.Slider(label="Upscale Factor", minimum=2, maximum=6, value=2)
-
-    with gr.Row():
-        generate_button = gr.Button("Генерация")
-        upscale_button = gr.Button("Улучшить качество", visible=False)
-
-    original_image_output = gr.Image(label="Сгенерированное изображение", visible=False)
-    upscaled_image_output = gr.Image(label="Улучшенное изображение", visible=False)
-
-    # Обработчики событий кнопок
-    def generate_button_click():
-        image = generate_image(
-            prompt_input.value, 
-            seed_input.value, 
-            sampling_input.value, 
-            cfg_input.value, 
-            width_input.value, 
-            height_input.value
-        )
-        original_image_output.update(value=image, visible=True)
-        upscale_button.update(visible=True)
-        return image
-
-    def upscale_button_click(image):
-        upscaled_image = upscale_image(image, upscale_factor_input.value)
-        upscaled_image_output.update(value=upscaled_image, visible=True)
+    response = requests.post("https://api.openai.com/v1/images/upscale", headers=headers, files=files, data=data)
+    
+    if response.status_code == 200:
+        image_data = response.json()
+        # Предполагаем, что API возвращает URL улучшенного изображения
+        image_url = image_data["data"]["image_url"]
+        image_response = requests.get(image_url)
+        upscaled_image = Image.open(BytesIO(image_response.content))
         return upscaled_image
+    else:
+        # В случае ошибки возвращаем сообщение об ошибке
+        return f"Error: {response.text}"
 
-    generate_button.click(fn=generate_button_click, inputs=[], outputs=[original_image_output])
-    upscale_button.click(fn=upscale_button_click, inputs=[original_image_output], outputs=[upscaled_image_output])
+# Определение интерфейса с использованием блоков Gradio (код интерфейса остается тем же)
 
+# Запуск интерфейса
 demo.launch()