Create app.py

app.py

@@ -0,0 +1,95 @@
+import gradio as gr
+import torch
+from diffusers import StableDiffusionInpaintPipeline
+
+# ------------------------------------------------------------------------------
+# 1) PIPELINE LADEN
+# ------------------------------------------------------------------------------
+# Modell für Inpainting: runwayml/stable-diffusion-inpainting
+# Läuft standardmäßig auf CPU (langsam, aber kostenlos).
+# Wenn eine GPU verfügbar ist, kannst du pipe.to("cuda") aktivieren.
+MODEL_ID = "runwayml/stable-diffusion-inpainting"
+
+print("Lade Inpainting-Pipeline... (kann einen Moment dauern)")
+pipe = StableDiffusionInpaintPipeline.from_pretrained(
+    MODEL_ID,
+    torch_dtype=torch.float32
+)
+
+# CPU-Modus (Garantie für kostenlos), kann aber 30+ Sek. pro Bild dauern.
+pipe.to("cpu")
+
+# ------------------------------------------------------------------------------
+# 2) INPAINT-FUNKTION
+# ------------------------------------------------------------------------------
+def inpaint_room(original_image, prompt):
+    """
+    Führt ein Inpainting auf dem gesamten Bild durch:
+    - 'mask_image' = original_image bedeutet, dass quasi alles überschrieben werden darf.
+    - Für echtes Home Staging (Möbel entfernen + neue einfügen) bräuchte man eine echte Maske
+      oder ein Segmentierungsmodell. Hier nur das Minimal-Demo.
+    """
+    if original_image is None:
+        return None
+
+    result = pipe(
+        prompt=prompt,
+        image=original_image,
+        mask_image=original_image,  # Ganzes Bild als "Maske"
+    ).images[0]
+
+    return result
+
+
+# ------------------------------------------------------------------------------
+# 3) GRADIO INTERFACE (Blocks Layout)
+# ------------------------------------------------------------------------------
+def build_app():
+    with gr.Blocks() as demo:
+        gr.Markdown("## 🏠 KI-gestütztes Virtual Home Staging (Demo)")
+        gr.Markdown(
+            "Lade ein Foto hoch (am besten ein leeres/unmöbliertes Zimmer) "
+            "und gib ein Stichwort / Stil an (z.B. 'modern furniture, bright decor').\n\n"
+            "**Achtung:** Auf CPU dauert die Generierung pro Bild ggf. über 30 Sekunden."
+        )
+
+        with gr.Row():
+            with gr.Column():
+                input_image = gr.Image(
+                    label="Schritt 1: Dein Zimmerfoto",
+                    type="pil"
+                )
+                prompt_text = gr.Textbox(
+                    label="Schritt 2: Was soll eingefügt werden?",
+                    value="modern furniture, bright lighting, minimal decor",
+                    lines=2
+                )
+                run_button = gr.Button("Starte Staging")
+
+            with gr.Column():
+                result_image = gr.Image(label="Ergebnis")
+
+        # Button-Click -> Inpainting ausführen
+        run_button.click(
+            fn=inpaint_room,
+            inputs=[input_image, prompt_text],
+            outputs=[result_image]
+        )
+
+        gr.Markdown(
+            "**Hinweis:**\n"
+            "- Dies ist ein reines Demo. Ohne echte Masken, also wird das gesamte Bild neu generiert.\n"
+            "- Für professionelles Homestaging brauchst du meist einen Segmentierungsschritt.\n"
+            "- Free CPU kann sehr langsam sein. Wenn eine Community-GPU verfügbar ist, geht's schneller."
+        )
+
+    return demo
+
+
+# ------------------------------------------------------------------------------
+# 4) START
+# ------------------------------------------------------------------------------
+demo = build_app()
+
+if __name__ == "__main__":
+    demo.launch()