Update app.py

app.py

@@ -1,10 +1,9 @@
 import streamlit as st
-from PIL import Image, ImageEnhance
+from PIL import Image, ImageEnhance, ImageFilter
 import numpy as np
-from deoldify import DeOldify
 
 # Title of the app
-st.title("Image Enhancer and Colorizer")
+st.title("Image Enhancer and Filter Application")
 
 # File uploader to upload the image
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
@@ -16,24 +15,32 @@ if uploaded_file is not None:
     # Display the original image
     st.image(image, caption='Original Image', use_container_width=True)
 
-    # Brightness adjustment
-    st.sidebar.header("Adjust Brightness")
-    brightness = st.sidebar.slider("Brightness", 0.0, 2.0, 1.0)
-    enhancer_brightness = ImageEnhance.Brightness(image)
-    enhanced_image_brightness = enhancer_brightness.enhance(brightness)
-    
-    # Contrast adjustment
-    st.sidebar.header("Adjust Contrast")
-    contrast = st.sidebar.slider("Contrast", 0.0, 2.0, 1.0)
-    enhancer_contrast = ImageEnhance.Contrast(enhanced_image_brightness)
-    enhanced_image_contrast = enhancer_contrast.enhance(contrast)
-
-    # Show the enhanced image
-    st.image(enhanced_image_contrast, caption='Enhanced Image', use_container_width=True)
-
-    # Colorization
-    if st.sidebar.button("Colorize Image"):
-        # Ensure the uploaded image is in grayscale
+    # Sidebar options for filters
+    st.sidebar.header("Select Filter")
+    filter_option = st.sidebar.selectbox("Choose a filter:", 
+                                          ["None", "Enhance", "Grayscale", "Colorize", "Blur", "Sharpen", "Edge Enhance"])
+
+    # Enhancement options
+    if filter_option == "Enhance":
+        st.sidebar.header("Adjust Brightness and Contrast")
+        brightness = st.sidebar.slider("Brightness", 0.0, 2.0, 1.0)
+        contrast = st.sidebar.slider("Contrast", 0.0, 2.0, 1.0)
+        
+        # Enhance brightness and contrast
+        enhancer_brightness = ImageEnhance.Brightness(image)
+        enhanced_image_brightness = enhancer_brightness.enhance(brightness)
+        enhancer_contrast = ImageEnhance.Contrast(enhanced_image_brightness)
+        enhanced_image = enhancer_contrast.enhance(contrast)
+        
+        st.image(enhanced_image, caption='Enhanced Image', use_container_width=True)
+
+    elif filter_option == "Grayscale":
+        # Convert to grayscale
+        gray_image = image.convert("L")
+        st.image(gray_image, caption='Grayscale Image', use_container_width=True)
+
+    elif filter_option == "Colorize":
+        # Ensure the uploaded image is in grayscale for colorization
         if image.convert('L') == image:
             colorizer = DeOldify()
             colorized_image = colorizer.colorize(image)
@@ -41,8 +48,26 @@ if uploaded_file is not None:
         else:
             st.sidebar.warning("Please upload a black and white image for colorization.")
 
-    # Option to download the enhanced image
-    st.sidebar.download_button("Download Enhanced Image", 
-                                 data=np.array(enhanced_image_contrast).tobytes(), 
-                                 file_name="enhanced_image.png", 
-                                 mime="image/png")
+    elif filter_option == "Blur":
+        # Apply a blur filter
+        blurred_image = image.filter(ImageFilter.BLUR)
+        st.image(blurred_image, caption='Blurred Image', use_container_width=True)
+
+    elif filter_option == "Sharpen":
+        # Apply a sharpen filter
+        sharpened_image = image.filter(ImageFilter.SHARPEN)
+        st.image(sharpened_image, caption='Sharpened Image', use_container_width=True)
+
+    elif filter_option == "Edge Enhance":
+        # Apply an edge enhancement filter
+        edge_enhanced_image = image.filter(ImageFilter.EDGE_ENHANCE)
+        st.image(edge_enhanced_image, caption='Edge Enhanced Image', use_container_width=True)
+
+    # Option to download the processed image
+    processed_image = enhanced_image if filter_option == "Enhance" else gray_image if filter_option == "Grayscale" else colorized_image if filter_option == "Colorize" else blurred_image if filter_option == "Blur" else sharpened_image if filter_option == "Sharpen" else edge_enhanced_image if filter_option == "Edge Enhance" else None
+
+    if processed_image:
+        st.sidebar.download_button("Download Processed Image", 
+                                     data=np.array(processed_image).tobytes(), 
+                                     file_name="processed_image.png", 
+                                     mime="image/png")