Create app.py

app.py

@@ -0,0 +1,57 @@
+import gradio as gr
+import requests
+import os
+
+API_KEY = os.getenv("API_KEY")
+DALLE_API_URL = "https://api.openai.com/v1/dalle"
+
+def call_dalle_api(prompt, size, num_images, style, aspect_ratio, temperature):
+    headers = {
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    data = {
+        "prompt": prompt,
+        "size": size,
+        "num_images": num_images,
+        "style": style,
+        "aspect_ratio": aspect_ratio,
+        "temperature": temperature
+        # Добавьте здесь другие параметры, если они необходимы
+    }
+    response = requests.post(DALLE_API_URL, headers=headers, json=data)
+    if response.status_code == 200:
+        images = response.json()["images"]
+        return [gr.Image(data=image) for image in images]
+    else:
+        raise Exception("API request failed with status code: " + str(response.status_code))
+
+def generate_image(prompt, size, num_images, style, aspect_ratio, temperature):
+    try:
+        return call_dalle_api(prompt, size, num_images, style, aspect_ratio, temperature)
+    except Exception as e:
+        return str(e)
+
+with gr.Blocks() as demo:
+    with gr.Tab("Базовые настройки"):
+        prompt_input = gr.Textbox(label="Введите ваш запрос")
+    
+    with gr.Tab("Расширенные настройки"):
+        with gr.Row():
+            with gr.Column():
+                size_input = gr.Slider(minimum=256, maximum=1024, step=256, label="Размер изображения", value=512)
+                num_images_input = gr.Slider(minimum=1, maximum=10, step=1, label="Количество изображений", value=1)
+            with gr.Column():
+                style_input = gr.Dropdown(choices=['Фотореалистичный', 'Художественный', 'Мультфильм'], label="Стиль")
+                aspect_ratio_input = gr.Dropdown(choices=['1:1', '16:9', '4:3'], label="Аспектное соотношение")
+                temperature_input = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, label="Температура", value=0.5)
+    
+    generate_button = gr.Button("Сгенерировать изображение")
+    images_output = gr.Gallery(label="Сгенерированные изображения")
+
+    generate_button.click(
+        generate_image,
+        inputs=[prompt_input, size_input, num_images_input, style_input, aspect_ratio_input, temperature_input],
+        outputs=images_output
+    )
+
+demo.launch()