Update app.py

app.py

@@ -1,12 +1,12 @@
-# app.py - Fixed Version
+# app.py - Streamlit Version
+import streamlit as st
 from PIL import Image
-import gradio as gr
-import numpy as np
+import io
 
-def stack_horizontal(img1_path, img2_path):
+def stack_horizontal(img1, img2):
     """Stapelt zwei Bilder horizontal nebeneinander."""
-    img1 = Image.open(img1_path).convert("RGB")
-    img2 = Image.open(img2_path).convert("RGB")
+    img1 = img1.convert("RGB")
+    img2 = img2.convert("RGB")
     
     h1, h2 = img1.size[1], img2.size[1]
     if h1 != h2:
@@ -22,53 +22,66 @@ def stack_horizontal(img1_path, img2_path):
     
     return stacked_img
 
-def blend_images_cpu(image1_path, image2_path, alpha: float):
+def blend_images_cpu(img1, img2, alpha):
     """Blendet zwei Bilder mit Alpha-Faktor."""
-    img1 = Image.open(image1_path).convert("RGBA")
-    img2 = Image.open(image2_path).convert("RGBA")
+    img1 = img1.convert("RGBA")
+    img2 = img2.convert("RGBA")
     
     if img1.size != img2.size:
         img2 = img2.resize(img1.size, Image.Resampling.LANCZOS)
     
-    # Blend gibt RGBA zurück, konvertiere zu RGB für konsistente Ausgabe
     blended = Image.blend(img1, img2, alpha)
     return blended.convert("RGB")
 
-def infer(image1, image2, method, alpha):
-    """Haupt-Inferenzfunktion."""
-    # Gradio gibt manchmal None zurück wenn kein Bild hochgeladen wurde
-    if image1 is None or image2 is None:
-        return None
-    
-    try:
-        if method == "Blend":
-            return blend_images_cpu(image1, image2, alpha)
-        elif method == "Stack Horizontal":
-            return stack_horizontal(image1, image2)
-    except Exception as e:
-        print(f"Fehler bei der Verarbeitung: {e}")
-        return None
+# Streamlit UI
+st.set_page_config(page_title="Image Transformer", layout="wide")
+
+st.title("🖼️ CPU Image Transformer")
+st.markdown("**Blend** oder **Stack** - Reine CPU-Operation")
+
+col1, col2 = st.columns(2)
 
-# Gradio Interface
-if __name__ == "__main__":
-    methods = ["Blend", "Stack Horizontal"]
+with col1:
+    st.subheader("Eingabe")
+    img1_file = st.file_uploader("Bild 1 (Basisbild)", type=["png", "jpg", "jpeg", "webp"])
+    img2_file = st.file_uploader("Bild 2 (Zusatzbild)", type=["png", "jpg", "jpeg", "webp"])
     
-    with gr.Blocks(title="CPU Image Transformer") as demo:
-        gr.Markdown("# CPU Image Transformer (Blend & Stack)")
-        gr.Markdown("Wähle zwischen Blending oder Horizontalem Stacking. Reine CPU-Operation.")
-        
-        with gr.Row():
-            with gr.Column():
-                img1 = gr.Image(type="filepath", label="Bild 1 (Basisbild)")
-                img2 = gr.Image(type="filepath", label="Bild 2 (Zusatzbild)")
-                method = gr.Dropdown(methods, label="Transformationsmethode", value="Blend")
-                alpha = gr.Slider(minimum=0.0, maximum=1.0, step=0.05, value=0.5, 
-                                 label="Blending-Faktor (nur bei 'Blend')")
-                btn = gr.Button("Verarbeiten")
-            
-            with gr.Column():
-                output = gr.Image(label="Resultat")
-        
-        btn.click(fn=infer, inputs=[img1, img2, method, alpha], outputs=output)
+    method = st.selectbox("Transformationsmethode", ["Blend", "Stack Horizontal"])
     
-    demo.queue().launch(share=False, server_name="127.0.0.1", server_port=7860)
+    alpha = st.slider("Blending-Faktor (nur bei Blend)", 0.0, 1.0, 0.5, 0.05)
+    
+    process_btn = st.button("🚀 Verarbeiten", type="primary")
+
+with col2:
+    st.subheader("Resultat")
+    result_placeholder = st.empty()
+
+# Verarbeitung
+if process_btn:
+    if img1_file and img2_file:
+        with st.spinner("Verarbeite..."):
+            try:
+                img1 = Image.open(img1_file)
+                img2 = Image.open(img2_file)
+                
+                if method == "Blend":
+                    result = blend_images_cpu(img1, img2, alpha)
+                else:
+                    result = stack_horizontal(img1, img2)
+                
+                result_placeholder.image(result, use_container_width=True)
+                
+                # Download-Button
+                buf = io.BytesIO()
+                result.save(buf, format="PNG")
+                st.download_button(
+                    label="💾 Download Resultat",
+                    data=buf.getvalue(),
+                    file_name=f"result_{method.lower().replace(' ', '_')}.png",
+                    mime="image/png"
+                )
+                
+            except Exception as e:
+                st.error(f"Fehler: {e}")
+    else:
+        st.warning("⚠️ Bitte beide Bilder hochladen!")