app.py

import gradio as gr
import os
import torch
from diffusers import StableDiffusionXLPipeline
from PIL import Image
import time

def generate_nudes(prompt, negative_prompt, num_images=10):
    images = []
    folder = "/content/ki_stehende_nackte"
    os.makedirs(folder, exist_ok=True)

    progress = gr.Progress()

    for i in range(num_images):
        progress((i + 1) / num_images, desc=f"Bild {i+1}/{num_images} wird generiert...")

        generator = torch.Generator("cuda").manual_seed(i + int(time.time()))

        result = pipe(
            prompt=prompt,
            negative_prompt=negative_prompt,
            num_inference_steps=35,
            guidance_scale=4.5,
            height=1216,
            width=832,
            generator=generator
        )

        img = result.images[0]
        filename = f"stehend_nackt_{i+1}.png"
        img.save(f"{folder}/{filename}")
        images.append(img)

    progress(1.0, desc="Fertig! Alle Bilder in /content/ki_stehende_nackte")
    return images, f"{num_images} stehende Nackte generiert!\nDownload-Ordner: {folder}\nRechtsklick → Download als ZIP"

with gr.Blocks(title="KI Standing Nude Generator - Colab Fix") as demo:
    gr.Markdown("# Juggernaut XL Ragnarok – Unlimited Stehende Nackte")
    gr.Markdown("**Tipp:** Englisch-Prompt + detailliert = geilste Ergebnisse")

    prompt = gr.Textbox(
        label="Prompt",
        value="masterpiece, best quality, ultra realistic, 8k, full body portrait of beautiful young woman standing completely nude, naked, perfect anatomy, detailed skin texture, natural breasts, shaved pussy, long hair, seductive pose, looking at viewer, soft natural lighting, photorealistic, sharp focus",
        lines=5
    )

    negative = gr.Textbox(
        label="Negative Prompt",
        value="deformed, bad anatomy, extra limbs, blurry, lowres, watermark, ugly, mutated hands, poorly drawn face, bad proportions, clothed, text",
        lines=3
    )

    num_images = gr.Slider(1, 30, value=8, step=1, label="Anzahl Bilder (mehr = länger, Colab-Limit beachten)")

    generate_btn = gr.Button("GENERIEREN", variant="primary")

    gallery = gr.Gallery(label="Deine stehenden Nackten")
    status = gr.Textbox(label="Status")

    generate_btn.click(
        generate_nudes,
        inputs=[prompt, negative, num_images],
        outputs=[gallery, status]
    )

demo.launch(share=True, debug=True)