from PIL import Image
import requests 
from openai import OpenAI
import gradio as gr
client = OpenAI(api_key= "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")


def get_image(prompt):
    response = client.images.generate(
        model = "dall-e-3",
        prompt = prompt,
        size = "1024x1024",
        n = 1
    )
    image_url = response.data[0].url
    data = requests.get(image_url).content
    f = open('img.jpg','wb') 
    f.write(data)
    f.close()
    img = Image.open('img.jpg')
    return img



gr.Interface(fn=get_image,inputs = "textbox", outputs= gr.Image(type='pil')).launch()