Update app.py

app.py

@@ -17,7 +17,7 @@ with col1:
 
 image = None
 filtered_image = None
-image_copy = None
+filter_option = "None"
 
 if uploaded_file:
     # Size check
@@ -30,87 +30,94 @@ if uploaded_file:
             st.error("❌ File too large! Must be under 5MB.")
     else:
         image = Image.open(uploaded_file)
-        image_copy = image.copy()  # Keep a copy for modification
 
         with col2:
-            st.header("🎛️ Select Filters")
-            # Use checkboxes for multiple filters
-            grayscale = st.checkbox("Grayscale")
-            invert = st.checkbox("Invert Colors")
-            sepia = st.checkbox("Sepia")
-            blur = st.checkbox("Blur")
-            edges = st.checkbox("Edge Detection")
-            posterize = st.checkbox("Posterize")
-            emboss = st.checkbox("Emboss")
-            solarize = st.checkbox("Solarize")
-            brightness = st.checkbox("Brightness Boost")
-            contrast = st.checkbox("Contrast Boost")
-            flip_h = st.checkbox("Flip Horizontal")
-            flip_v = st.checkbox("Flip Vertical")
-            rotate = st.checkbox("Rotate 90°")
+            st.header("🎛️ Select Filter")
+            filter_option = st.selectbox("Choose a filter:", [
+                "None", "Grayscale", "Invert Colors", "Sepia", "Blur", "Edge Detection", 
+                "Posterize", "Emboss", "Solarize", "Brightness Boost", "Contrast Boost", 
+                "Flip Horizontal", "Flip Vertical", "Rotate 90°"
+            ])
 
         with col3:
             st.header("🔍 Preview Output")
 
-            # Start with the original copy of the image
-            filtered_image = image_copy.copy()
+            if filter_option == "Grayscale":
+                filtered_image = ImageOps.grayscale(image)
+                st.image(filtered_image, caption="Grayscale", use_container_width=True)
 
-            # Apply the selected filters one by one
-            if grayscale:
-                filtered_image = ImageOps.grayscale(filtered_image)
+            elif filter_option == "Invert Colors":
+                if image.mode != "RGB":
+                    image = image.convert("RGB")
+                filtered_image = ImageOps.invert(image)
+                st.image(filtered_image, caption="Inverted", use_container_width=True)
 
-            if invert:
-                if filtered_image.mode != "RGB":
-                    filtered_image = filtered_image.convert("RGB")
-                filtered_image = ImageOps.invert(filtered_image)
-
-            if sepia:
-                width, height = filtered_image.size
-                pixels = filtered_image.load()
+            elif filter_option == "Sepia":
+                # Sepia filter
+                width, height = image.size
+                pixels = image.load()  # create the pixel map
                 for py in range(height):
                     for px in range(width):
-                        r, g, b = filtered_image.getpixel((px, py))
+                        r, g, b = image.getpixel((px, py))
 
                         tr = int(0.393 * r + 0.769 * g + 0.189 * b)
                         tg = int(0.349 * r + 0.686 * g + 0.168 * b)
                         tb = int(0.272 * r + 0.534 * g + 0.131 * b)
 
-                        if tr > 255: tr = 255
-                        if tg > 255: tg = 255
-                        if tb > 255: tb = 255
+                        # Apply filter
+                        if tr > 255:
+                            tr = 255
+                        if tg > 255:
+                            tg = 255
+                        if tb > 255:
+                            tb = 255
 
                         pixels[px, py] = (tr, tg, tb)
+                filtered_image = image
+                st.image(filtered_image, caption="Sepia", use_container_width=True)
+
+            elif filter_option == "Blur":
+                filtered_image = image.filter(ImageFilter.GaussianBlur(radius=5))
+                st.image(filtered_image, caption="Blurred", use_container_width=True)
 
-            if blur:
-                filtered_image = filtered_image.filter(ImageFilter.GaussianBlur(radius=5))
+            elif filter_option == "Edge Detection":
+                filtered_image = image.filter(ImageFilter.FIND_EDGES)
+                st.image(filtered_image, caption="Edge Detection", use_container_width=True)
 
-            if edges:
-                filtered_image = filtered_image.filter(ImageFilter.FIND_EDGES)
+            elif filter_option == "Posterize":
+                filtered_image = ImageOps.posterize(image, 4)
+                st.image(filtered_image, caption="Posterized", use_container_width=True)
 
-            if posterize:
-                filtered_image = ImageOps.posterize(filtered_image, 4)
+            elif filter_option == "Emboss":
+                filtered_image = image.filter(ImageFilter.EMBOSS)
+                st.image(filtered_image, caption="Emboss", use_container_width=True)
 
-            if emboss:
-                filtered_image = filtered_image.filter(ImageFilter.EMBOSS)
+            elif filter_option == "Solarize":
+                filtered_image = ImageOps.solarize(image, threshold=128)
+                st.image(filtered_image, caption="Solarized", use_container_width=True)
 
-            if solarize:
-                filtered_image = ImageOps.solarize(filtered_image, threshold=128)
+            elif filter_option == "Brightness Boost":
+                enhancer = ImageEnhance.Brightness(image)
+                filtered_image = enhancer.enhance(2)  # increase brightness
+                st.image(filtered_image, caption="Brightness Boosted", use_container_width=True)
 
-            if brightness:
-                enhancer = ImageEnhance.Brightness(filtered_image)
-                filtered_image = enhancer.enhance(2)
+            elif filter_option == "Contrast Boost":
+                enhancer = ImageEnhance.Contrast(image)
+                filtered_image = enhancer.enhance(2)  # increase contrast
+                st.image(filtered_image, caption="Contrast Boosted", use_container_width=True)
 
-            if contrast:
-                enhancer = ImageEnhance.Contrast(filtered_image)
-                filtered_image = enhancer.enhance(2)
+            elif filter_option == "Flip Horizontal":
+                filtered_image = image.transpose(Image.FLIP_LEFT_RIGHT)
+                st.image(filtered_image, caption="Flipped Horizontal", use_container_width=True)
 
-            if flip_h:
-                filtered_image = filtered_image.transpose(Image.FLIP_LEFT_RIGHT)
+            elif filter_option == "Flip Vertical":
+                filtered_image = image.transpose(Image.FLIP_TOP_BOTTOM)
+                st.image(filtered_image, caption="Flipped Vertical", use_container_width=True)
 
-            if flip_v:
-                filtered_image = filtered_image.transpose(Image.FLIP_TOP_BOTTOM)
+            elif filter_option == "Rotate 90°":
+                filtered_image = image.rotate(90)
+                st.image(filtered_image, caption="Rotated 90°", use_container_width=True)
 
-            if rotate:
-                filtered_image = filtered_image.rotate(90)
+            else:
+                st.info("Select a filter to see output.")
 
-            st.image(filtered_image, caption="Filtered Image", use_container_width=True)