Update app.py

app.py

@@ -1,18 +1,18 @@
 import streamlit as st
 from PIL import Image
 import io
+import traceback
 
-def compress_image(image, quality=85):
-    img_byte_arr = io.BytesIO()
-    image.save(img_byte_arr, format='JPEG', quality=quality)
-    img_byte_arr.seek(0)
-    return Image.open(img_byte_arr)
-
-def convert_image_format(image, target_format='webp', quality=85):
-    img_byte_arr = io.BytesIO()
-    image.save(img_byte_arr, format=target_format.upper(), quality=quality)
-    img_byte_arr.seek(0)
-    return Image.open(img_byte_arr)
+def compress_image(image, format='webp', quality=85):
+    try:
+        img_byte_arr = io.BytesIO()
+        image.save(img_byte_arr, format=format.upper(), quality=quality)
+        img_byte_arr.seek(0)
+        return Image.open(img_byte_arr)
+    except Exception as e:
+        st.error(f"Fehler bei Bildkompression: {e}")
+        st.error(traceback.format_exc())
+        return None
 
 def main():
     st.title("Bild Optimierungs-Tool")
@@ -22,69 +22,76 @@ def main():
     uploaded_file = st.file_uploader("Wählen Sie ein Bild", type=['jpg', 'png', 'jpeg', 'webp'])
     
     if uploaded_file is not None:
-        original_image = Image.open(uploaded_file)
-        
-        st.subheader("Originalbild")
-        st.image(original_image, caption="Ursprüngliches Bild")
-        
-        compression_quality = st.sidebar.slider(
-            "Kompressionsqualität", 
-            min_value=1, 
-            max_value=100, 
-            value=85
-        )
-        
-        target_formats = st.sidebar.multiselect(
-            "Zielformate", 
-            ['jpg', 'png', 'webp'],
-            default=['webp']
-        )
-        
-        if st.button("Bild optimieren"):
-            results = {}
+        try:
+            original_image = Image.open(uploaded_file)
             
-            for format in target_formats:
-                optimized_image = convert_image_format(
-                    original_image, 
-                    target_format=format, 
-                    quality=compression_quality
-                )
-                
-                img_byte_arr = io.BytesIO()
-                optimized_image.save(img_byte_arr, format=format.upper(), quality=compression_quality)
-                img_byte_arr = img_byte_arr.getvalue()
-                
-                original_size = len(uploaded_file.getvalue())
-                optimized_size = len(img_byte_arr)
-                compression_ratio = (1 - optimized_size / original_size) * 100
-                
-                results[format] = {
-                    'image': optimized_image,
-                    'original_size': original_size,
-                    'optimized_size': optimized_size,
-                    'compression_ratio': compression_ratio
-                }
+            st.subheader("Originalbild")
+            st.image(original_image, caption="Ursprüngliches Bild")
             
-            st.subheader("Optimierungsergebnisse")
-            for format, result in results.items():
-                col1, col2 = st.columns(2)
-                
-                with col1:
-                    st.image(result['image'], caption=f"Optimiert ({format.upper()})")
-                
-                with col2:
-                    st.write(f"Ursprüngliche Größe: {result['original_size']} Bytes")
-                    st.write(f"Optimierte Größe: {result['optimized_size']} Bytes")
-                    st.write(f"Kompressionsrate: {result['compression_ratio']:.2f}%")
+            # Ursprungsgröße anzeigen
+            original_size_bytes = len(uploaded_file.getvalue())
+            st.write(f"Ursprüngliche Bildgröße: {original_size_bytes} Bytes")
+            
+            compression_quality = st.sidebar.slider(
+                "Kompressionsqualität", 
+                min_value=1, 
+                max_value=100, 
+                value=85
+            )
+            
+            # Nur ein Format wählbar
+            target_format = st.sidebar.selectbox(
+                "Zielformat", 
+                ['webp', 'jpg', 'png']
+            )
+            
+            if st.button("Bild optimieren"):
+                try:
+                    # Bild optimieren
+                    optimized_image = compress_image(
+                        original_image, 
+                        format=target_format, 
+                        quality=compression_quality
+                    )
+                    
+                    if optimized_image:
+                        # Speichern im Speicher
+                        img_byte_arr = io.BytesIO()
+                        optimized_image.save(img_byte_arr, format=target_format.upper(), quality=compression_quality)
+                        img_byte_arr.seek(0)
+                        
+                        # Größenvergleich
+                        optimized_size_bytes = img_byte_arr.getbuffer().nbytes
+                        compression_ratio = (1 - optimized_size_bytes / original_size_bytes) * 100
+                        
+                        st.subheader("Optimierungsergebnisse")
+                        col1, col2 = st.columns(2)
+                        
+                        with col1:
+                            st.image(optimized_image, caption=f"Optimiert ({target_format.upper()})")
+                        
+                        with col2:
+                            st.write(f"Ursprüngliche Größe: {original_size_bytes} Bytes")
+                            st.write(f"Optimierte Größe: {optimized_size_bytes} Bytes")
+                            st.write(f"Kompressionsrate: {compression_ratio:.2f}%")
+                        
+                        # Download-Button
+                        st.download_button(
+                            label=f"Download {target_format.upper()}",
+                            data=img_byte_arr,
+                            file_name=f"optimized_image.{target_format}",
+                            mime=f"image/{target_format}"
+                        )
+                    else:
+                        st.error("Bildoptimierung fehlgeschlagen")
                 
-                img_byte_arr = io.BytesIO()
-                result['image'].save(img_byte_arr, format=format.upper(), quality=compression_quality)
-                st.download_button(
-                    label=f"Download {format.upper()}",
-                    data=img_byte_arr.getvalue(),
-                    file_name=f"optimized_image.{format}",
-                    mime=f"image/{format}"
-                )
+                except Exception as e:
+                    st.error(f"Fehler bei Bildoptimierung: {e}")
+                    st.error(traceback.format_exc())
+        
+        except Exception as e:
+            st.error(f"Fehler beim Laden des Bildes: {e}")
+            st.error(traceback.format_exc())
 
 if __name__ == "__main__":
     main()