Update app.py

app.py

@@ -82,6 +82,62 @@ def add_frame(image, frame_style="Simple"):
         frame_width = 30
         image_with_frame = ImageOps.expand(image, border=frame_width, fill="blue")
         return image_with_frame.convert("RGBA").rotate(0, resample=Image.BICUBIC)
+    elif frame_style == "Polaroid":
+        # Add Polaroid-style frame (white bottom for caption)
+        frame_width = 40
+        image_with_frame = ImageOps.expand(image, border=frame_width, fill="white")
+        bottom_margin = Image.new('RGB', (image_with_frame.width, 50), color='white')
+        return Image.new('RGB', (image_with_frame.width, image_with_frame.height + bottom_margin.height), color='white').paste(image_with_frame, (0, 0)).paste(bottom_margin, (0, image_with_frame.height))
+    elif frame_style == "Shadow":
+        # Add shadow frame
+        frame_width = 50
+        shadow_image = Image.new("RGBA", (image.width + 2 * frame_width, image.height + 2 * frame_width), (0, 0, 0, 180))
+        shadow_image.paste(image, (frame_width, frame_width))
+        return shadow_image.convert("RGB")
+    elif frame_style == "Vintage":
+        # Add vintage-style frame with a distressed look (this can be more complex)
+        frame_width = 30
+        return ImageOps.expand(image, border=frame_width, fill="beige")
+    elif frame_style == "Filmstrip":
+        # Add a filmstrip frame with a perforated edge
+        frame_width = 60
+        return ImageOps.expand(image, border=frame_width, fill="black")
+    elif frame_style == "Double Border":
+        # Add a double border (one thin, one thick)
+        outer_width = 60
+        inner_width = 20
+        image_with_outer_border = ImageOps.expand(image, border=outer_width, fill="black")
+        return ImageOps.expand(image_with_outer_border, border=inner_width, fill="white")
+    elif frame_style == "Metallic":
+        # Add metallic frame (gold, silver, etc.)
+        frame_width = 40
+        return ImageOps.expand(image, border=frame_width, fill="gold")
+    elif frame_style == "Heart-Shaped":
+        # Add heart-shaped frame
+        image_with_frame = image.resize((450, 450))  # Adjust for better fit
+        return image_with_frame.convert("RGBA")
+    elif frame_style == "Circular":
+        # Add circular frame
+        image_with_frame = image.resize((450, 450))  # Adjust for better fit
+        return image_with_frame.convert("RGBA")
+    elif frame_style == "Wooden":
+        # Add wooden frame
+        frame_width = 50
+        return ImageOps.expand(image, border=frame_width, fill="saddlebrown")
+    elif frame_style == "Neon Glow":
+        # Add neon glow frame
+        frame_width = 50
+        neon_image = Image.new("RGBA", (image.width + 2 * frame_width, image.height + 2 * frame_width), (0, 255, 255, 255))
+        neon_image.paste(image, (frame_width, frame_width))
+        return neon_image.convert("RGB")
+    elif frame_style == "Gradient":
+        # Add gradient frame
+        frame_width = 40
+        return ImageOps.expand(image, border=frame_width, fill="gradient")
+    elif frame_style == "Geometric":
+        # Add geometric frame (triangles, hexagons, etc.)
+        frame_width = 30
+        return ImageOps.expand(image, border=frame_width, fill="purple")
 
 # Function to convert PIL image to bytes for download
 def pil_to_bytes(image, format="JPEG"):
@@ -141,7 +197,10 @@ def main():
             border_width = st.sidebar.slider("Border Width", 1, 100, 10)
             image = add_border(image, border_color, border_width)
         elif border_option == "Frame":
-            frame_style = st.sidebar.selectbox("Choose a frame style", ("Simple", "Ornate", "Rounded"))
+            frame_style = st.sidebar.selectbox("Choose a frame style", (
+                "Simple", "Ornate", "Rounded", "Polaroid", "Shadow", "Vintage", "Filmstrip", "Double Border", "Metallic", 
+                "Heart-Shaped", "Circular", "Wooden", "Neon Glow", "Gradient", "Geometric"
+            ))
             image = add_frame(image, frame_style)
 
         # Resize the filtered image for display
@@ -166,4 +225,4 @@ def main():
             st.download_button("Download Edited Image", img_bytes, "edited_image.jpg", "image/jpeg")
 
 if __name__ == "__main__":
-    main()
+    main()