import requests
import gradio as gr
from PIL import Image

def t2i(text):
    prompt_url = f"https://aadarsh-text-to-image-mymvubi2mq-el.a.run.app/t2i/{text}"
    img_url = requests.get(prompt_url).json()["url"]
    response = requests.get(img_url)
    if response.status_code:
        fp = open('image.png', 'wb')
        fp.write(response.content)
        fp.close()

    image = Image.open('image.png')
    return image
    
iface = gr.Interface(fn=t2i, inputs="text", outputs=[gr.Image(label="Generated Image")]).launch()
iface.launch(share=True)