Update app.py

app.py

@@ -1,73 +1,76 @@
-# app.py (Korrigiert und erweitert)
+# app.py (Mit Blend- und Stack-Funktion)
 
 from PIL import Image
 import gradio as gr
 import numpy as np
-import os # Wird hinzugefügt, falls Beispielbilder im Ordner 'examples' liegen
+# Wir brauchen NumPy für das horizontale Stapeln (wird von PIL nicht direkt unterstützt)
 
-# Der Kern der Anwendung: Blending-Funktion
-def blend_images_cpu(image1_path, image2_path, alpha: float):
+# 1. Stacking-Funktion
+def stack_horizontal(img1_path, img2_path):
     """
-    Verschmilzt zwei Bilder mit einem linearen Blending-Faktor (alpha).
-    Beide Bilder werden auf die Größe des ersten Bildes skaliert.
+    Stapelt zwei Bilder horizontal nebeneinander. 
+    Beide Bilder werden auf die Höhe des ersten Bildes skaliert.
     """
+    img1 = Image.open(img1_path).convert("RGB")
+    img2 = Image.open(img2_path).convert("RGB")
+    
+    # 1. Höhe angleichen (Voraussetzung für horizontales Stapeln)
+    h1, h2 = img1.size[1], img2.size[1]
+    if h1 != h2:
+        # Skaliere img2 auf die Höhe von img1, behalte Seitenverhältnis
+        w2_new = int(img2.size[0] * h1 / h2)
+        img2 = img2.resize((w2_new, h1))
+    
+    # 2. Neues Bild erstellen und Bilder hineinkopieren
+    total_width = img1.size[0] + img2.size[0]
+    total_height = img1.size[1]
     
-    try:
-        # 1. Bilder laden und in den RGBA-Modus konvertieren (für sauberes Blending)
-        img1 = Image.open(image1_path).convert("RGBA")
-        img2 = Image.open(image2_path).convert("RGBA")
+    stacked_img = Image.new('RGB', (total_width, total_height))
+    stacked_img.paste(img1, (0, 0))
+    stacked_img.paste(img2, (img1.size[0], 0))
+    
+    return stacked_img
 
-        # 2. Größe anpassen: img2 wird auf die Größe von img1 skaliert
-        if img1.size != img2.size:
-            # print("Größe der Bilder wird angepasst.") # Im Gradio-Space nicht sichtbar
-            img2 = img2.resize(img1.size)
+# 2. Blending-Funktion (wie zuvor)
+def blend_images_cpu(image1_path, image2_path, alpha: float):
+    img1 = Image.open(image1_path).convert("RGBA")
+    img2 = Image.open(image2_path).convert("RGBA")
+    
+    if img1.size != img2.size:
+        img2 = img2.resize(img1.size)
         
-        # 3. Lineares Blending durchführen
-        # Image.blend(Bild1, Bild2, alpha)
-        # alpha=0.0 -> 100% Bild 1, alpha=1.0 -> 100% Bild 2
-        blended_img = Image.blend(img1, img2, alpha)
+    return Image.blend(img1, img2, alpha)
 
-        # Gradio kann direkt ein PIL Image-Objekt zurückgeben
-        return blended_img
 
-    except Exception as e:
-        print(f"Ein Fehler ist aufgetreten: {e}")
-        # Bei Fehler ein leeres Bild zurückgeben
-        return np.zeros((200, 200, 3), dtype=np.uint8) 
-
-# Liste der Beispiel-Eingaben
-# WICHTIG: Die Pfade müssen zu existierenden Dateien in deinem 'examples'-Ordner führen!
-# Annahme: Im Ordner 'examples' liegen 'vogel.jpg', 'wasser.jpg', 'muster.png'
-example_inputs = [
-    # 1. Bild A dominant (Alpha = 0.2)
-    ["examples/vogel.jpg", "examples/wasser.jpg", 0.2], 
-    
-    # 2. 50/50 Mischung (Alpha = 0.5)
-    ["examples/vogel.jpg", "examples/wasser.jpg", 0.5], 
-    
-    # 3. Bild B dominant (Alpha = 0.8)
-    ["examples/vogel.jpg", "examples/wasser.jpg", 0.8], 
-    
-    # 4. Beispiel mit einem anderen Paar (Alpha = 0.4)
-    ["examples/vogel.jpg", "examples/muster.png", 0.4] 
-]
+# 3. Haupt-Inferenzfunktion
+def infer(image1_path, image2_path, method, alpha: float):
+    if method == "Blend":
+        # Blending-Faktor (Alpha) wird nur bei Blend genutzt
+        return blend_images_cpu(image1_path, image2_path, alpha)
+    elif method == "Stack Horizontal":
+        # Alpha-Faktor wird ignoriert
+        return stack_horizontal(image1_path, image2_path)
 
 
 # Gradio Interface erstellen
 if __name__ == "__main__":
+    
+    # Liste der verfügbaren Methoden
+    methods = ["Blend", "Stack Horizontal"]
+    
+    # HINWEIS: gr.Examples ist hier NICHT enthalten, um Startprobleme zu vermeiden!
     gr.Interface(
-        title="CPU-Friendly Image Blending (Pillow)",
-        description="Lade zwei Bilder hoch und verschmelze sie mit einem einstellbaren Alpha-Faktor. **Alpha** steuert den Anteil von Bild 2: *0.0 = Nur Bild 1*, *1.0 = Nur Bild 2*",
-        fn=blend_images_cpu,
+        title="CPU Image Transformer (Blend & Stack)",
+        description="Wähle zwischen Blending (Überblenden mit Alpha) oder Horizontalem Stacking (Aneinanderreihen). Reine CPU-Operation.",
+        fn=infer,
         inputs=[
             gr.Image(type="filepath", label="Bild 1 (Basisbild)"),
-            gr.Image(type="filepath", label="Bild 2 (Überlagerung)"),
-            gr.Slider(minimum=0.0, maximum=1.0, step=0.05, value=0.5, label="Blending-Faktor (Alpha)")
+            gr.Image(type="filepath", label="Bild 2 (Zusatzbild)"),
+            gr.Dropdown(methods, label="Transformationsmethode", value="Blend"),
+            gr.Slider(minimum=0.0, maximum=1.0, step=0.05, value=0.5, label="Blending-Faktor (Alpha, nur bei 'Blend' aktiv)")
         ],
         outputs=[
-            gr.Image(label="Verschmolzenes Bild")
+            gr.Image(label="Resultat")
         ],
-        # HIER IST DIE KORREKTUR: gr.Examples anstatt gr.Example
-        examples=example_inputs, 
         cache_examples=False,
-    ).launch(share=False)
+    ).queue().launch(share=False)