Update app.py

app.py

@@ -2,11 +2,65 @@ import gradio as gr
 import requests
 import random
 
-def generate_image(prompt, width, height, seed):
+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"
+    }
+    
+    prompt = style_tags.get(style, "") + " " + prompt if style_tags.get(style, "") else prompt
+    prompt = prompt.strip()
+    
     url = f"https://image.pollinations.ai/prompt/{prompt}?width={width}&height={height}&seed={seed}&nologo=true&nofeed=true"
-    response = requests.get(url)
+    response = requests.get(url, timeout=150)
     if response.status_code == 200:
         return response.content
     else:
@@ -15,16 +69,66 @@ def generate_image(prompt, width, height, seed):
 # Создаем интерфейс Gradio
 with gr.Blocks() as demo:
     with gr.Tab("Основные настройки"):
-        prompt_input = gr.Textbox(label="Промпт", placeholder="Введите промпт на английском языке")
+        prompt_input = gr.Textbox(label="Описание изображения", placeholder="Введите описание изображения на английском языке")
+        
+        with gr.Accordion("Стиль", open=False):
+            style_radio = gr.Radio(label="Выбор стиля", choices=[
+                "Свой",
+                "Аниме",
+                "Реализм",
+                "3D",
+                "Краски",
+                "Пиксель-арт",
+                "Машины с лицами",
+                "Космический",
+                "Монстры",
+                "Супергерои",
+                "Детская книга",
+                "Готический",
+                "Ретро",
+                "Стимпанк",
+                "Фэнтези",
+                "Народные мотивы",
+                "Минимализм",
+                "Гаражный хардкор",
+                "Стример",
+                "Новый хоррор",
+                "Детская мультфильм",
+                "Арт-ноу",
+                "Викторианский",
+                "Грейский тон",
+                "Моцарт",
+                "Гамбит",
+                "Киберпанк",
+                "Грот",
+                "Цирк",
+                "Арт-деко",
+                "Стратегические игры",
+                "Инди-игра",
+                "Медиевский",
+                "Макет",
+                "Футуристический",
+                "Пасхалка",
+                "Микстап",
+                "Скульптура",
+                "Дизайн интерьера",
+                "Спорт",
+                "Голливудский",
+                "Тематический парк",
+                "Космический полет",
+                "Лабиринт",
+                "Тропический лес",
+                "Экспериментальный"
+            ], value="Свой")
     
     with gr.Tab("Дополнительные настройки"):
         width_slider = gr.Slider(label="Ширина", minimum=1, maximum=4000, value=1024)
-        height_slider = gr.Slider(label="Высота", minimum=1, maximum=4000, value=512)
+        height_slider = gr.Slider(label="Высота", minimum=1, maximum=4000, value=1024)
         seed_slider = gr.Slider(label="Сид", minimum=-1, maximum=999999, value=-1, step=1)
     
     create_button = gr.Button("Создать")
     output_image = gr.Image(label="Изображение", type="pil")  # Указываем тип вывода как PIL изображение
 
-    create_button.click(fn=generate_image, inputs=[prompt_input, width_slider, height_slider, seed_slider], outputs=output_image)
+    create_button.click(fn=generate_image, inputs=[prompt_input, style_radio, width_slider, height_slider, seed_slider], outputs=output_image)
 
 demo.launch()