commit

app.py

@@ -2,41 +2,73 @@ import gradio as gr
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Публичная модель без gated (работает без token)
+# Публичная модель (работает без token)
 pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, safety_checker=None)
+
+# Оптимизации для бесплатного tier (скорость и память)
 pipe.enable_model_cpu_offload()
 pipe.enable_vae_slicing()
 pipe.enable_attention_slicing()
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-def generate(prompt, negative_prompt="", steps=30, seed=42):
+# Стили
+styles = {
+    "Без стиля": "",
+    "Аниме": "anime style, highly detailed",
+    "Реализм": "photorealistic, ultra detailed",
+    "Киберпанк": "cyberpunk, neon lights",
+    "Дисней": "disney pixar style, cute",
+    "Фэнтези": "fantasy art, magical",
+    "Супергерой": "superhero style, dramatic",
+    "Гарри Поттер": "harry potter style, wizard hat, hogwarts, magic wand"
+}
+
+def generate(prompt, style, negative_prompt, steps, seed):
     try:
-        negative = negative_prompt or "blurry, low quality, ugly"
+        full_prompt = f"{prompt}, {styles[style]}"
+        negative = negative_prompt or "blurry, low quality, ugly, deformed"
+
         generator = torch.Generator("cuda").manual_seed(seed) if torch.cuda.is_available() else None
-        image = pipe(prompt, negative_prompt=negative, num_inference_steps=steps, generator=generator).images[0]
-        return image
+
+        image = pipe(
+            full_prompt,
+            negative_prompt=negative,
+            num_inference_steps=steps,
+            generator=generator
+        ).images[0]
+
+        return image, None
     except Exception as e:
-        return f"Ошибка: {str(e)}. Попробуй проще prompt."
+        return None, f"Ошибка: {str(e)}. Попробуй проще prompt или меньше шагов."
 
-with gr.Blocks(theme=gr.themes.Dark()) as demo:
+with gr.Blocks() as demo:
     gr.Markdown("# Epic AI Art Generator 🔥")
-    gr.Markdown("Сделано для моей девушки ❤️ Генерирует эпичные арты! Попробуй на русском или английском.")
+    gr.Markdown("Сделано с ❤️ для моей самой красивой девушки на свете")
+    gr.Markdown("Генерирует эпичные арты! Попробуй на русском или английском.")
 
     with gr.Row():
-        prompt = gr.Textbox(label="Prompt (описание)", placeholder="рыжий кот в стиле Гарри Поттера, волшебная шляпа")
-        negative = gr.Textbox(label="Negative prompt", placeholder="blurry, low quality")
-        steps = gr.Slider(20, 50, value=30, step=5, label="Шаги (steps)")
-        seed = gr.Slider(0, 999999, value=42, label="Seed")
+        with gr.Column():
+            prompt = gr.Textbox(label="Prompt (описание)", placeholder="рыжий кот в стиле Гарри Поттера")
+            style = gr.Dropdown(choices=list(styles.keys()), value="Без стиля", label="Стиль")
+            negative = gr.Textbox(label="Negative prompt", placeholder="blurry, low quality")
+            steps = gr.Slider(20, 50, value=30, step=5, label="Шаги (steps)")
+            seed = gr.Slider(0, 999999, value=42, label="Seed")
+
+            btn = gr.Button("Generate Magic 🔥")
 
-        btn = gr.Button("Generate 🔥")
+        with gr.Column():
+            output = gr.Image(label="Арт")
+            error = gr.Textbox(label="Статус", interactive=False)
 
-    output = gr.Image(label="Арт")
+    gallery = gr.Gallery(label="История", columns=3)
 
-    btn.click(generate, inputs=[prompt, negative, steps, seed], outputs=output)
+    btn.click(generate, inputs=[prompt, style, negative, steps, seed], outputs=[output, error]) \
+       .then(lambda g, img: g + [img] if img else g, inputs=[gallery, output], outputs=gallery)
 
     gr.Markdown("### Примеры:")
     gr.Markdown("- 'рыжий кот космонавт'")
     gr.Markdown("- 'романтическая пара на закате'")
     gr.Markdown("- 'кот в мантии волшебника, стиль Гарри Поттера'")
+    gr.Markdown("- 'мы как супергерои'")
 
 demo.launch()