Displaying a generated image from openai

Beginners
2

Hi @unifiedGround ,

here is the complete example, you missed the steps of fetching the image and rendering a PIL Image


import openai
from PIL import Image
import requests
import gradio as gr

OPEN_AI_KEY = "......"
openai = openai.OpenAI(api_key=OPEN_AI_KEY)


def generateImage(prompt):
    response = openai.images.generate(
        model="dall-e-3",
        prompt=prompt,
        size="1024x1024",
        quality="standard",
        n=1,
    )
    return response.data[0].url


def handlePrompt(prompt):
    urlString = generateImage(prompt)
    print(urlString)
    image = Image.open(requests.get(urlString, stream=True).raw)
    return image


with gr.Blocks() as demo:
    gr.HTML("<H1> Unified Ground Image generator</H1>")
    with gr.Row():
        input = gr.Textbox(label="Enter Prompt")
        output = gr.Image()
    generate_btn = gr.Button("Generate Image")
    generate_btn.click(fn=handlePrompt, inputs=input, outputs=output)

demo.launch()