| import gradio as gr | |
| import requests | |
| import random | |
| import sys | |
| import os | |
| def download(url): | |
| filename = str(random.randint(111111111, 999999999)) + ".png" | |
| file_path = os.path.join(os.path.dirname(__file__), filename) | |
| response = requests.get(url) | |
| if response.status_code == 200: | |
| with open(file_path, "wb") as f: | |
| f.write(response.content) | |
| return(file_path) | |
| def process(prompt): | |
| download_file("https://image.pollinations.ai/prompt/" + text + ('.' * random.randint(0, 100))) | |
| iface = gr.Interface(fn=process, inputs="text", outputs="image") | |
| iface.launch() |