yunogpt / image.py
AGofficial's picture
Upload 5 files
6507cd4 verified
import requests
from urllib.parse import quote
def generate_and_save_image(prompt: str, path: str, width: int = 800, height: int = 800):
base_url = "https://image.pollinations.ai/prompt/"
encoded_prompt = quote(prompt)
url = f"{base_url}{encoded_prompt}?width={width}&height={height}&nologo=true"
response = requests.get(url)
if response.status_code == 200:
with open(path, "wb") as f:
f.write(response.content)
print(f"Image saved to {path}")
else:
print(f"Failed to generate image. Status code: {response.status_code}")
print(response.text)
if __name__ == "__main__":
generate_and_save_image("Anime girl wearing school uniform and pink hair.", "output.png", 512, 512)