app.py

import os
import requests
import gradio as gr

api_key = os.getenv("API_KEY")

def generate_image(prompt, code):
    if code != os.getenv("code"):
        raise gr.Error("❗ Не верный ключ!")
        return None
    headers = {
        "Authorization": f"Bearer {api_key}",
        "Content-Type": "application/json",
    }
    data = {
        "prompt": prompt,
        "num_images": 1,
    }
    response = requests.post("https://api.openai.com/v1/images/generations", headers=headers, json=data)
    response_json = response.json()
    
    # Получаем URL первого сгенерированного изображения
    image_url = response_json["data"][0]["url"]
    
    return image_url
css = """
footer {visibility: hidden !important;}
"""
# Создание интерфейса с помощью Gradio
with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
    with gr.Row():
        code = gr.Textbox(label="Ключ доступа", type="password")
    with gr.Row():
        prompt_input = gr.Textbox(label="Описание изображения")
    submit_btn = gr.Button("Генерация", variant='primary')
    image_output = gr.Image(label="Изображение")
    
    submit_btn.click(fn=generate_image, inputs=[prompt_input, code], outputs=image_output)

demo.launch()