app.py

import os
import gradio as gr
from openai import OpenAI
client = OpenAI(api_key=os.getenv("API_KEY"))

def generate_image(prompt, code, size_512):
    if code != os.getenv("code"):
        raise gr.Error("❗ Не верный ключ!")
        return None
    size = "512x512" if size_512 else "1024x1024"
    response = client.images.generate(
        model="dall-e-3",
        prompt=prompt,
        size=size,
        n=1,
    )

    image_url = response.data[0].url
    print(image_url)
    return image_url

css = """
footer {visibility: hidden !important;}
"""
# Создание интерфейса с помощью Gradio
with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
    with gr.Row():
        with gr.Column():
            with gr.Row():
                code = gr.Textbox(label="Ключ доступа", type="password")
            with gr.Row():
                prompt_input = gr.Textbox(label="Описание изображения", lines=3)
            with gr.Row():
                size_512_checkbox = gr.Checkbox(label="512x512", value=False)
            submit_btn = gr.Button("Генерация", variant='primary')
        with gr.Column():
            image_output = gr.Image(label="Изображение")
    
    submit_btn.click(fn=generate_image, inputs=[prompt_input, code, size_512_checkbox], outputs=image_output)

demo.launch()