import gradio as gr
import requests
import random
import sys
import os

def process(Prompt):
    filename = str(random.randint(111111111, 999999999)) + ".png"
    
    file_path = os.path.join(os.path.dirname(__file__), filename)

    response = requests.get("https://image.pollinations.ai/prompt/" + Prompt + "&width=1536&height=1536&nologo=true&seed=" + str(random.randint(0, 999999999)))

    if response.status_code == 200:

        with open(file_path, "wb") as f:

            f.write(response.content)

        return(file_path)

title = "Pollinations Image Generator" 
description = "Pollinations API + Randomizer"

iface = gr.Interface(fn=process, inputs="text", outputs="image", title=title, description=description)
iface.launch()