create app.py

app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+import torch
+from transformers import pipeline
+
+# Загрузка модели Stable Diffusion (оптимизированная)
+pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
+pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+
+# Простая NSFW-проверка (из твоего курса по security)
+nsfw_detector = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")  # Замени на NSFW модель если нужно
+
+def generate_image(prompt, image=None):
+    if nsfw_detector(prompt)[0]['label'] == 'NEGATIVE':  # Простая проверка на плохой промпт
+        return "Prompt not safe - try again!"
+
+    # Генерация
+    result = pipe(prompt).images[0]
+
+    # Если загружено фото — базовый style transfer (можно расширить)
+    if image:
+        # Здесь можно добавить face swap или style transfer (расширь позже)
+        result = result.resize(image.size)  # Простой resize для комбо
+
+    return result
+
+# Интерфейс Gradio
+iface = gr.Interface(
+    fn=generate_image,
+    inputs=[gr.Textbox(label="Description (e.g., 'Romantic couple on sunset beach')"), gr.Image(label="Upload photo for style (optional)")],
+    outputs=gr.Image(label="Generated Image"),
+    title="Epic AI Art Generator",
+    description="Generate awesome images! Made for my girl ❤️"
+)
+
+iface.launch()