import requests
from io import BytesIO
from PIL import Image
import time
import urllib.parse
import random

# Вставь свой API ключ с pollinations.ai
API_KEY = "СЮДА_ВСТАВЬ_СВОЙ_API_КЛЮЧ"

# Стили генерации
styles = [
"The Simpsons cartoon style, yellow skin, Matt Groening style",
"Studio Ghibli anime style, Hayao Miyazaki, dreamy, magical",
"Disney Pixar 3D animation style, cute, detailed",
"Rick and Morty cartoon style, sci-fi, colorful",
"Futurama cartoon style, retro future, comedy",
"1960s America, vintage, retro advertising, mad men style",
"1980s synthwave, neon grid, sunset, retrowave",
"Victorian steampunk, gears, brass, steam machines",
"Art Deco 1920s, geometric, gold, elegant",
"Soviet propaganda poster style, bold colors, heroic",
"Cyberpunk, neon lights, rain, blade runner style",
"Vaporwave aesthetic, pink, blue, retro computer",
"Sci-fi concept art, space, futuristic, epic",
"Watercolor painting, soft, dreamy, artistic",
"Oil painting, classical, moody, renaissance",
"Pixel art, 8bit, colorful, retro game",
"Pencil sketch, hand drawn, detailed, artistic",
"Japanese ukiyo-e, woodblock print, traditional",
"Pop art, Andy Warhol style, bold colors, comic",
"Impressionist painting, Monet style, soft light",
"Low poly 3D render, geometric, minimal, colorful",
"Isometric art, cute, detailed, miniature world",
"Comic book style, Marvel DC, dynamic, action",
"Stained glass window, colorful, medieval, beautiful",
]

def generate_image(style_index=0, save_path="generated.jpg"):
style = styles[style_index]
prompt = f"beautiful creative scene, masterpiece, {style}"

print(f"Генерирую: {prompt}")

encoded = urllib.parse.quote(prompt)
url = f"https://gen.pollinations.ai/image/{encoded}?width=512&height=384&seed={random.randint(1,99999)}&model=flux&nologo=true"

headers = {"Authorization": f"Bearer {API_KEY}"}

try:
response = requests.get(url, timeout=90, headers=headers)
if response.status_code != 200:
print(f"Ошибка: HTTP {response.status_code}")
return None

# Сохраняем картинку
with open(save_path, 'wb') as f:
f.write(response.content)

print(f"Готово! Сохранено: {save_path}")
return save_path

except Exception as e:
print(f"Ошибка: {e}")
return None

# Пример использования
if name == "main":
# Генерируем картинку в стиле Симпсонов (индекс 0)
generate_image(style_index=0, save_path="simpsons.jpg")

# Генерируем в стиле киберпанка (индекс 10)
generate_image(style_index=10, save_path="cyberpunk.jpg")

# Случайный стиль
generate_image(style_index=random.randint(0, len(styles)-1), save_path="random.jpg")