cora / tests /test_pollinations.py
tokgae's picture
Upload folder using huggingface_hub
38ab39c verified
import requests
from io import BytesIO
from PIL import Image
import random
def test_pollinations(prompt, model='flux'):
print(f"🎨 Testing Pollinations with model='{model}'...")
seed = random.randint(0, 100000)
# Pollinations URL format: https://image.pollinations.ai/prompt/{prompt}
# We must URL encode the prompt
import urllib.parse
encoded_prompt = urllib.parse.quote(prompt)
url = f"https://image.pollinations.ai/prompt/{encoded_prompt}?nologo=true"
print(f"Requesting: {url}")
try:
response = requests.get(url, timeout=30)
if response.status_code == 200:
img = Image.open(BytesIO(response.content))
filename = f"pollinations_simple.png"
img.save(filename)
print(f"✅ SUCCESS! Saved {filename}")
else:
print(f"❌ FAILED: {response.status_code} - {response.text}")
except Exception as e:
print(f"❌ ERROR: {e}")
if __name__ == "__main__":
test_pollinations("roman mercenaries, digital art")