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;}
"""
js = """
document.addEventListener('DOMContentLoaded', (event) => {
  if (document.cookie.split(';').some((item) => item.trim().startsWith('code='))) {
    const codeValue = document.cookie.split('; ').find(row => row.startsWith('code=')).split('=')[1];
    document.getElementById('code').value = codeValue;
  }
});

function saveCode() {
  const codeValue = document.getElementById('code').value;
  document.cookie = 'code=' + codeValue + ';max-age=31536000'; // Сохраняем на 1 год
}
"""

# Создание интерфейса с помощью Gradio
with gr.Blocks(css=css, theme='YTheme/Sketch') as demo:
    gr.HTML(f"<script>{js}</script>")
    with gr.Row():
        code = gr.Textbox(label="Ключ доступа", type="password", elem_id="code", change=saveCode)
    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()