Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from PIL import Image, ImageEnhance, ImageOps, ImageFilter
+from PIL import Image, ImageEnhance, ImageOps, ImageFilter, ImageDraw, ImageFont
 import numpy as np
 from io import BytesIO
 
@@ -41,6 +41,28 @@ class ContrastFilter(ImageEditor):
         return enhancer.enhance(self.factor)
 
 
+# Sharpness Adjustment Class
+class SharpnessFilter(ImageEditor):
+    def __init__(self, image, factor):
+        super().__init__(image)
+        self.factor = factor
+
+    def apply(self):
+        enhancer = ImageEnhance.Sharpness(self.image)
+        return enhancer.enhance(self.factor)
+
+
+# Saturation Adjustment Class
+class SaturationFilter(ImageEditor):
+    def __init__(self, image, factor):
+        super().__init__(image)
+        self.factor = factor
+
+    def apply(self):
+        enhancer = ImageEnhance.Color(self.image)
+        return enhancer.enhance(self.factor)
+
+
 # Blur Effect Class
 class BlurFilter(ImageEditor):
     def __init__(self, image, radius):
@@ -51,6 +73,44 @@ class BlurFilter(ImageEditor):
         return self.image.filter(ImageFilter.GaussianBlur(self.radius))
 
 
+# Sepia Filter Class
+class SepiaFilter(ImageEditor):
+    def apply(self):
+        width, height = self.image.size
+        pixels = self.image.load()
+        for py in range(height):
+            for px in range(width):
+                r, g, b = self.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)
+
+                # Apply limits to avoid overflow
+                if tr > 255:
+                    tr = 255
+                if tg > 255:
+                    tg = 255
+                if tb > 255:
+                    tb = 255
+
+                pixels[px, py] = (tr, tg, tb)
+
+        return self.image
+
+
+# Emboss Filter Class
+class EmbossFilter(ImageEditor):
+    def apply(self):
+        return self.image.filter(ImageFilter.EMBOSS)
+
+
+# Invert Filter Class
+class InvertFilter(ImageEditor):
+    def apply(self):
+        return ImageOps.invert(self.image.convert('RGB'))
+
+
 # Border Class
 class Border:
     def __init__(self, image, border_type, color, thickness):
@@ -68,18 +128,34 @@ class Border:
 
         if self.border_type == "Solid Color":
             return ImageOps.expand(self.image, border=self.thickness, fill=border_color)
-        # Add other border types if needed
-
+        elif self.border_type == "Shadow":
+            return self.add_shadow_border()
+        elif self.border_type == "Rounded":
+            return self.add_rounded_border()
         return self.image
 
-    @staticmethod
-    def hex_to_rgb(hex_color):
+    def hex_to_rgb(self, hex_color):
         hex_color = hex_color.lstrip('#')
         if len(hex_color) == 6:
             return tuple(int(hex_color[i:i + 2], 16) for i in (0, 2, 4))
         else:
             raise ValueError("Invalid hex color format")
 
+    def add_shadow_border(self):
+        shadow = Image.new("RGBA", (self.image.width + 20, self.image.height + 20), (0, 0, 0, 0))
+        shadow.paste(self.image, (10, 10))
+        return shadow.convert("RGB")
+
+    def add_rounded_border(self):
+        radius = 20
+        img_with_border = self.image.copy()
+        img_with_border = img_with_border.convert("RGBA")
+        mask = Image.new("L", img_with_border.size, 255)
+        draw = ImageDraw.Draw(mask)
+        draw.rounded_rectangle([0, 0, img_with_border.width, img_with_border.height], radius, fill=0)
+        img_with_border.putalpha(mask)
+        return img_with_border.convert("RGB")
+
 
 # Main Application Class
 class ImageApp:
@@ -96,7 +172,8 @@ class ImageApp:
     def apply_filters(self):
         if self.image:
             # Sidebar Filters
-            filter_option = st.sidebar.selectbox("Choose a filter", ["None", "Grayscale", "Brightness", "Contrast", "Blur"])
+            filter_option = st.sidebar.selectbox("Choose a filter", ["None", "Grayscale", "Brightness", "Contrast", 
+                                                                   "Sharpness", "Saturation", "Blur", "Sepia", "Emboss", "Invert"])
             if filter_option == "Grayscale":
                 self.image = GrayscaleFilter(self.image).apply()
             elif filter_option == "Brightness":
@@ -105,13 +182,25 @@ class ImageApp:
             elif filter_option == "Contrast":
                 contrast_factor = st.sidebar.slider("Adjust Contrast", 0.1, 2.0, 1.0)
                 self.image = ContrastFilter(self.image, contrast_factor).apply()
+            elif filter_option == "Sharpness":
+                sharpness_factor = st.sidebar.slider("Adjust Sharpness", 0.1, 2.0, 1.0)
+                self.image = SharpnessFilter(self.image, sharpness_factor).apply()
+            elif filter_option == "Saturation":
+                saturation_factor = st.sidebar.slider("Adjust Saturation", 0.1, 2.0, 1.0)
+                self.image = SaturationFilter(self.image, saturation_factor).apply()
             elif filter_option == "Blur":
                 blur_radius = st.sidebar.slider("Blur Radius", 1, 10, 2)
                 self.image = BlurFilter(self.image, blur_radius).apply()
+            elif filter_option == "Sepia":
+                self.image = SepiaFilter(self.image).apply()
+            elif filter_option == "Emboss":
+                self.image = EmbossFilter(self.image).apply()
+            elif filter_option == "Invert":
+                self.image = InvertFilter(self.image).apply()
 
     def add_border(self):
         if self.image:
-            border_type = st.sidebar.selectbox("Choose Border Type", ["None", "Solid Color"])
+            border_type = st.sidebar.selectbox("Choose Border Type", ["None", "Solid Color", "Shadow", "Rounded"])
             if border_type != "None":
                 border_color = st.sidebar.color_picker("Select Border Color", "#b33c3c")
                 border_thickness = st.sidebar.slider("Border Thickness", 1, 20, 5)
@@ -136,7 +225,7 @@ class ImageApp:
 
 # Main function to run the app
 def main():
-    st.title("Modern Image Editor")
+    st.title("Modernized Image Editor")
     app = ImageApp()
 
     # Load the image