app.py

import gradio as gr
import requests
import random
import urllib.parse
import logging
from PIL import Image
from io import BytesIO
from deep_translator import GoogleTranslator
from langdetect import detect

# Настройка логирования
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def generate_image(prompt, style, width, height, seed):
    if seed == -1:
        seed = random.randint(0, 999999)
    
    # Добавляем теги для стилей
    style_tags = {
        "Свой": "",
        "Аниме": "anime style, detailed, vibrant colors, cartoonish",
        "Реализм": "realism style, hyperrealistic, photorealistic",
        "3D": "3D style, rendering, high detail, volumetric lighting",
        "Краски": "paintings style, impressionist, detailed brush strokes, oil painting",
        "Пиксель-арт": "pixel art style, low resolution, retro, nostalgic",
        "Машины с лицами": "machines with faces, mechanical humanoid, futuristic, detailed features",
        "Космический": "space style, celestial bodies, galaxies, stars, nebulae",
        "Монстры": "monster style, mythical creatures, detailed anatomy, fantasy",
        "Супергерои": "superhero style, comic book, vibrant colors, dynamic poses",
        "Детская книга": "children's book style, whimsical, colorful, simple shapes",
        "Готический": "gothic style, dark, intricate details, brooding atmosphere",
        "Ретро": "retro style, vintage, 8-bit, nostalgic",
        "Стимпанк": "steampunk style, steam-powered machinery, brass and copper elements, Victorian era",
        "Фэнтези": "fantasy style, magical creatures, enchanted forests, mythical landscapes",
        "Народные мотивы": "folk art style, traditional patterns, cultural motifs, vibrant colors",
        "Минимализм": "minimalist style, clean lines, simple shapes, neutral colors",
        "Гаражный хардкор": "garage punk style, industrial, rugged, DIY, metal textures",
        "Стример": "streamer style, gaming, charismatic, energetic, modern",
        "Новый хоррор": "new horror style, dark atmosphere, suspenseful, twisted, modern",
        "Детская мультфильм": "children's animation style, colorful, cartoonish, simple shapes",
        "Арт-ноу": "art nouveau style, ornate, flowing lines, detailed patterns",
        "Викторианский": "victorian style, intricate details, ornate architecture, vintage",
        "Грейский тон": "grayscale style, monochrome, detailed textures, high contrast",
        "Моцарт": "mozart style, classical music, baroque, elegant, detailed",
        "Гамбит": "chess style, board game, strategy, detailed pieces, high contrast",
        "Киберпанк": "cyberpunk style, futuristic, neon lights, high-tech, detailed environments",
        "Грот": "cave style, underground, detailed textures, stone formations, natural",
        "Цирк": "circus style, vibrant, acrobatic, colorful, dynamic",
        "Арт-деко": "art deco style, geometric patterns, ornate, luxurious, 1920s",
        "Стратегические игры": "strategy game style, hexagonal tiles, detailed maps, military units",
        "Инди-игра": "indie game style, charming, minimalist, vibrant colors, simple graphics",
        "Медиевский": "medieval style, knightly armor, castles, detailed textures, historical",
        "Макет": "model kit style, detailed parts, assembly instructions, high precision",
        "Футуристический": "futuristic style, sleek design, high-tech, minimalistic, modern",
        "Пасхалка": "easter egg style, hidden details, playful, humorous, subtle",
        "Микстап": "meme style, humorous, exaggerated, pop culture references, viral",
        "Скульптура": "sculpture style, detailed modeling, textured surfaces, realistic materials",
        "Дизайн интерьера": "interior design style, modern, functional, detailed textures, cozy",
        "Спорт": "sports style, dynamic action, detailed athletes, high contrast, vibrant colors",
        "Голливудский": "hollywood style, glamour, high contrast, detailed sets, cinematic",
        "Тематический парк": "theme park style, vibrant, detailed rides, futuristic, entertaining",
        "Космический полет": "space travel style, detailed spaceships, planets, starfields, high contrast",
        "Лабиринт": "maze style, intricate pathways, detailed textures, puzzle-like, high contrast",
        "Тропический лес": "tropical rainforest style, lush vegetation, detailed textures, vibrant colors",
        "Экспериментальный": "experimental style, abstract, unconventional, detailed patterns, vibrant colors"
    }

    language = detect(prompt)
    
    if language != 'en':
        prompt = GoogleTranslator(source=language, target='en').translate(prompt)
    
    prompt = style_tags.get(style, "") + ". " + prompt if style_tags.get(style, "") else prompt
    prompt = prompt.strip()
    
    # URL-кодирование промпта
    encoded_prompt = urllib.parse.quote(prompt)
    
    url = f"https://image.pollinations.ai/prompt/{encoded_prompt}?width={width}&height={height}&seed={seed}&nologo=true&nofeed=true"
    logger.info(f"Generated URL: {url}")
    
    try:
        response = requests.get(url, timeout=150)
        if response.status_code == 200:
            logger.info("Image generated successfully")
            # Преобразуем байты в изображение
            image = Image.open(BytesIO(response.content))
            return image
        else:
            logger.error(f"Failed to generate image. Status code: {response.status_code}, Response: {response.text}")
            raise gr.Error("Извините, изображение создавалось слишком долго или произошла ошибка. Попробуйте снова.")
    except requests.exceptions.Timeout:
        logger.error("Request timed out")
        raise gr.Error("Извините, изображение создавалось слишком долго, мы отменили генерацию.")
    except requests.exceptions.RequestException as e:
        logger.error(f"Request exception: {e}")
        raise gr.Error("Произошла ошибка при генерации изображения. Попробуйте снова.")

# Ссылка на файл CSS
css_url = "https://neurixyufi-aihub.static.hf.space/style.css"

# Получение CSS по ссылке
response = requests.get(css_url)
css = response.text + ".gradio-container{max-width: 700px !important} h1{text-align:center}"

# Создаем интерфейс Gradio
with gr.Blocks(css=css) as demo:
    gr.Markdown("")
    gr.Markdown("# Генератор Изображений (Lite)")
    with gr.Tab("Основные настройки"):
        prompt_input = gr.Textbox(label="Описание изображения", placeholder="Введите описание изображения на английском языке")
        
        with gr.Accordion("Стиль", open=False):
            style_radio = gr.Radio(label="Выбор стиля", choices=[
                "Свой",
                "Аниме",
                "Реализм",
                "3D",
                "Краски",
                "Пиксель-арт",
                "Машины с лицами",
                "Космический",
                "Монстры",
                "Супергерои",
                "Детская книга",
                "Готический",
                "Ретро",
                "Стимпанк",
                "Фэнтези",
                "Народные мотивы",
                "Минимализм",
                "Гаражный хардкор",
                "Стример",
                "Новый хоррор",
                "Детская мультфильм",
                "Арт-ноу",
                "Викторианский",
                "Грейский тон",
                "Моцарт",
                "Гамбит",
                "Киберпанк",
                "Грот",
                "Цирк",
                "Арт-деко",
                "Стратегические игры",
                "Инди-игра",
                "Медиевский",
                "Макет",
                "Футуристический",
                "Пасхалка",
                "Микстап",
                "Скульптура",
                "Дизайн интерьера",
                "Спорт",
                "Голливудский",
                "Тематический парк",
                "Космический полет",
                "Лабиринт",
                "Тропический лес",
                "Экспериментальный"
            ], value="Свой")
    
    with gr.Tab("Дополнительные настройки"):
        with gr.Row():
            width_slider = gr.Slider(label="Ширина", minimum=80, maximum=4000, value=1024)
            height_slider = gr.Slider(label="Высота", minimum=80, maximum=4000, value=1024)
        with gr.Row():
            seed_slider = gr.Slider(label="Сид", minimum=-1, maximum=999999, value=-1, step=1)
    
    create_button = gr.Button("Создать")
    output_image = gr.Image(label="Изображение", type="pil") 

    create_button.click(
        fn=generate_image,
        inputs=[prompt_input, style_radio, width_slider, height_slider, seed_slider],
        outputs=[output_image]
    )

demo.queue(max_size=150).launch()