| 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) |
|
|
