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)