import gradio as gr
import openai
import os

openai.api_key = os.environ.get("OPENAI_API_KEY")

# Generate an image from text using DALL-E
def open_ai_txt2img(prompt):
    try:
        response = openai.Image.create(
            prompt=prompt,
            n=3,
            size="1024x1024",
        )
        return [img["url"] for img in response["data"]]
    except openai.error.OpenAIError as e:
        print(e.http_status)
        print(e.error)

# Gradio Interface
def generator(prompt):
    return open_ai_txt2img(prompt)

with gr.Blocks() as demo:
    prompt = gr.Textbox(label="Prompt")
    submit = gr.Button(label="Generate")
    image1, image2, image3 = gr.Image(), gr.Image(), gr.Image()
    submit.click(generator, inputs=[prompt], outputs=[image1, image2, image3], api_name="mmd")

demo.launch()