Update app.py

app.py

@@ -1,236 +1,235 @@
-# app.py
-
-import streamlit as st
-import cv2
-import numpy as np
-from PIL import Image
-from io import BytesIO
-import mediapipe as mp  # NEW for background removal
-
-# ----- Custom CSS Styling -----
-def local_css():
-    st.markdown("""
-    <style>
-    /* Main Background */
-    .stApp {
-        background-image: url('https://img.freepik.com/free-photo/arrows-pastel-colors_23-2148488400.jpg?semt=ais_hybrid&w=740');
-        background-size: cover;
-        background-position: center;
-        background-attachment: fixed;
-        min-height: 100vh;
-        color: darkred;
-    }
-
-    /* Sidebar Background */
-    section[data-testid="stSidebar"] {
-        background-image: url('https://i.pinimg.com/736x/10/76/df/1076df6744238e75e79047f7c2d2bbec.jpg');
-        background-size: cover;
-        background-position: center;
-    }
-
-    /* Container Styling */
-    .css-1d391kg {
-        background: linear-gradient(135deg, #89f7fe 0%, #66a6ff 100%);
-        border-radius: 15px;
-        padding: 20px;
-    }
-
-    /* Button Styling */
-    .stButton > button {
-        color: white;
-        background: linear-gradient(45deg, #ff6a00, #ee0979);
-        border: none;
-        border-radius: 10px;
-        padding: 0.75em 1.5em;
-        font-size: 1.1em;
-        box-shadow: 0px 4px 15px rgba(0,0,0,0.2);
-    }
-    .stButton > button:hover {
-        background: linear-gradient(45deg, #43cea2, #185a9d);
-    }
-
-    /* Profile Links Styling */
-    .profile-links img {
-        vertical-align: middle;
-        margin-right: 8px;
-    }
-    .profile-links a {
-        text-decoration: none;
-        color: #333;
-        font-size: 0.9em;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
-def sidebar_profiles():
-    st.sidebar.markdown("### 🎉Author: Maria Nadeem🌟")
-    st.sidebar.markdown("### 🔗 Connect With Me")
-    st.sidebar.markdown("""
-    <hr>
-    <div class="profile-links">
-        <a href="https://github.com/marianadeem755" target="_blank">
-            <img src="https://cdn-icons-png.flaticon.com/512/25/25231.png" width="20px"> GitHub
-        </a><br><br>
-        <a href="https://www.kaggle.com/marianadeem755" target="_blank">
-            <img src="https://cdn4.iconfinder.com/data/icons/logos-and-brands/512/189_Kaggle_logo_logos-512.png" width="20px"> Kaggle
-        </a><br><br>
-        <a href="mailto:marianadeem755@gmail.com">
-            <img src="https://cdn-icons-png.flaticon.com/512/561/561127.png" width="20px"> Email
-        </a><br><br>
-        <a href="https://huggingface.co/maria355" target="_blank">
-            <img src="https://huggingface.co/front/assets/huggingface_logo-noborder.svg" width="20px"> Hugging Face
-        </a>
-    </div>
-    <hr>
-    """, unsafe_allow_html=True)
-
-# ----- Filters Functions -----
-def apply_grayscale(img):
-    return cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-
-def apply_blur(img, ksize):
-    return cv2.GaussianBlur(img, (ksize, ksize), 0)
-
-def apply_canny(img, threshold1, threshold2):
-    return cv2.Canny(img, threshold1, threshold2)
-
-def apply_sepia(img):
-    kernel = np.array([[0.272, 0.534, 0.131],
-                       [0.349, 0.686, 0.168],
-                       [0.393, 0.769, 0.189]])
-    sepia_img = cv2.transform(img, kernel)
-    sepia_img = np.clip(sepia_img, 0, 255)
-    return sepia_img
-
-def apply_pencil_sketch(img):
-    if len(img.shape) == 3:
-        gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    else:
-        gray_img = img
-    inv_img = 255 - gray_img
-    blur_img = cv2.GaussianBlur(inv_img, (21, 21), 0)
-    sketch = cv2.divide(gray_img, 255 - blur_img, scale=256)
-    return sketch
-
-def apply_invert(img):
-    return cv2.bitwise_not(img)
-
-def apply_background_removal(img):
-    mp_selfie_segmentation = mp.solutions.selfie_segmentation
-    with mp_selfie_segmentation.SelfieSegmentation(model_selection=1) as selfie_segmentation:
-        rgb_img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-        results = selfie_segmentation.process(rgb_img)
-        mask = results.segmentation_mask
-        condition = mask > 0.5
-        bg_color = np.ones(img.shape, dtype=np.uint8) * 255
-        output_img = np.where(condition[..., None], img, bg_color)
-        return output_img
-
-# Convert to PIL
-def convert_image(img):
-    if len(img.shape) == 2:
-        return Image.fromarray(img)
-    else:
-        return Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
-
-# Downloadable image
-def download_image(img):
-    buf = BytesIO()
-    img.save(buf, format="PNG")
-    byte_im = buf.getvalue()
-    return byte_im
-
-# ----- Streamlit App Starts Here -----
-def main():
-    st.set_page_config(page_title="Advanced Image Filter Studio", page_icon="🎨", layout="wide")
-    local_css()
-
-    st.title("🎨 Advanced Image Filter Studio")
-    st.write("Upload an image, apply **amazing filters**, and download your creation!")
-
-    # Sidebar Profiles
-    sidebar_profiles()
-
-    # Sidebar
-    st.sidebar.header("1. Upload Image")
-    uploaded_file = st.sidebar.file_uploader("Choose an image", type=["jpg", "jpeg", "png"])
-
-    if uploaded_file is not None:
-        file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
-        opencv_image = cv2.imdecode(file_bytes, 1)
-
-        # Filters Section
-        st.sidebar.header("2. Choose Filters")
-        grayscale = st.sidebar.checkbox("Grayscale")
-        blur = st.sidebar.checkbox("Blur")
-        canny = st.sidebar.checkbox("Edge Detection")
-        sepia = st.sidebar.checkbox("Sepia Effect")
-        sketch = st.sidebar.checkbox("Pencil Sketch")
-        invert = st.sidebar.checkbox("Invert Colors")
-        remove_bg = st.sidebar.checkbox("Remove Background (Simple)")
-
-        st.sidebar.header("3. Filter Parameters")
-        blur_strength = st.sidebar.slider("Blur Intensity (odd numbers)", 1, 49, 15, step=2)
-        threshold1 = st.sidebar.slider("Canny Threshold1", 50, 300, 100)
-        threshold2 = st.sidebar.slider("Canny Threshold2", 50, 300, 150)
-
-        # Process Image
-        final_image = opencv_image.copy()
-
-        with st.spinner("🖌️ Applying Filters..."):
-            if grayscale:
-                final_image = apply_grayscale(final_image)
-
-            if blur:
-                if len(final_image.shape) == 2:
-                    final_image = cv2.cvtColor(final_image, cv2.COLOR_GRAY2BGR)
-                final_image = apply_blur(final_image, blur_strength)
-
-            if canny:
-                if len(final_image.shape) != 2:
-                    final_image = cv2.cvtColor(final_image, cv2.COLOR_BGR2GRAY)
-                final_image = apply_canny(final_image, threshold1, threshold2)
-
-            if sepia:
-                if len(final_image.shape) == 2:
-                    final_image = cv2.cvtColor(final_image, cv2.COLOR_GRAY2BGR)
-                final_image = apply_sepia(final_image)
-
-            if sketch:
-                if len(final_image.shape) != 2:
-                    final_image = cv2.cvtColor(final_image, cv2.COLOR_BGR2GRAY)
-                final_image = apply_pencil_sketch(final_image)
-
-            if invert:
-                final_image = apply_invert(final_image)
-
-            if remove_bg:
-                if len(final_image.shape) != 3 or final_image.shape[2] != 3:
-                    final_image = cv2.cvtColor(final_image, cv2.COLOR_GRAY2BGR)
-                final_image = apply_background_removal(final_image)
-
-        # Columns to show images
-        col1, col2 = st.columns(2)
-
-        with col1:
-            st.subheader("Original Image")
-            st.image(uploaded_file, use_column_width=True)
-
-        with col2:
-            st.subheader("Processed Image")
-            final_pil = convert_image(final_image)
-            st.image(final_pil, use_column_width=True)
-
-        # Download button
-        st.markdown("---")
-        st.download_button(
-            label="📥 Download Processed Image",
-            data=download_image(final_pil),
-            file_name="processed_image.png",
-            mime="image/png"
-        )
-    else:
-        st.info("👈 Please upload an image from the sidebar to get started.")
-
-if __name__ == "__main__":
-    main()
+# app.py
+import streamlit as st
+import cv2
+import numpy as np
+from PIL import Image
+from io import BytesIO
+import mediapipe as mp  # NEW for background removal
+
+# ----- Custom CSS Styling -----
+def local_css():
+    st.markdown("""
+    <style>
+    /* Main Background */
+    .stApp {
+        background-image: url('https://img.freepik.com/free-photo/arrows-pastel-colors_23-2148488400.jpg?semt=ais_hybrid&w=740');
+        background-size: cover;
+        background-position: center;
+        background-attachment: fixed;
+        min-height: 100vh;
+        color: darkred;
+    }
+
+    /* Sidebar Background */
+    section[data-testid="stSidebar"] {
+        background-image: url('https://i.pinimg.com/736x/10/76/df/1076df6744238e75e79047f7c2d2bbec.jpg');
+        background-size: cover;
+        background-position: center;
+    }
+
+    /* Container Styling */
+    .css-1d391kg {
+        background: linear-gradient(135deg, #89f7fe 0%, #66a6ff 100%);
+        border-radius: 15px;
+        padding: 20px;
+    }
+
+    /* Button Styling */
+    .stButton > button {
+        color: white;
+        background: linear-gradient(45deg, #ff6a00, #ee0979);
+        border: none;
+        border-radius: 10px;
+        padding: 0.75em 1.5em;
+        font-size: 1.1em;
+        box-shadow: 0px 4px 15px rgba(0,0,0,0.2);
+    }
+    .stButton > button:hover {
+        background: linear-gradient(45deg, #43cea2, #185a9d);
+    }
+
+    /* Profile Links Styling */
+    .profile-links img {
+        vertical-align: middle;
+        margin-right: 8px;
+    }
+    .profile-links a {
+        text-decoration: none;
+        color: #333;
+        font-size: 0.9em;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+def sidebar_profiles():
+    st.sidebar.markdown("### 🎉Author: Maria Nadeem🌟")
+    st.sidebar.markdown("### 🔗 Connect With Me")
+    st.sidebar.markdown("""
+    <hr>
+    <div class="profile-links">
+        <a href="https://github.com/marianadeem755" target="_blank">
+            <img src="https://cdn-icons-png.flaticon.com/512/25/25231.png" width="20px"> GitHub
+        </a><br><br>
+        <a href="https://www.kaggle.com/marianadeem755" target="_blank">
+            <img src="https://cdn4.iconfinder.com/data/icons/logos-and-brands/512/189_Kaggle_logo_logos-512.png" width="20px"> Kaggle
+        </a><br><br>
+        <a href="mailto:marianadeem755@gmail.com">
+            <img src="https://cdn-icons-png.flaticon.com/512/561/561127.png" width="20px"> Email
+        </a><br><br>
+        <a href="https://huggingface.co/maria355" target="_blank">
+            <img src="https://huggingface.co/front/assets/huggingface_logo-noborder.svg" width="20px"> Hugging Face
+        </a>
+    </div>
+    <hr>
+    """, unsafe_allow_html=True)
+
+# ----- Filters Functions -----
+def apply_grayscale(img):
+    return cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+
+def apply_blur(img, ksize):
+    return cv2.GaussianBlur(img, (ksize, ksize), 0)
+
+def apply_canny(img, threshold1, threshold2):
+    return cv2.Canny(img, threshold1, threshold2)
+
+def apply_sepia(img):
+    kernel = np.array([[0.272, 0.534, 0.131],
+                       [0.349, 0.686, 0.168],
+                       [0.393, 0.769, 0.189]])
+    sepia_img = cv2.transform(img, kernel)
+    sepia_img = np.clip(sepia_img, 0, 255)
+    return sepia_img
+
+def apply_pencil_sketch(img):
+    if len(img.shape) == 3:
+        gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    else:
+        gray_img = img
+    inv_img = 255 - gray_img
+    blur_img = cv2.GaussianBlur(inv_img, (21, 21), 0)
+    sketch = cv2.divide(gray_img, 255 - blur_img, scale=256)
+    return sketch
+
+def apply_invert(img):
+    return cv2.bitwise_not(img)
+
+def apply_background_removal(img):
+    mp_selfie_segmentation = mp.solutions.selfie_segmentation
+    with mp_selfie_segmentation.SelfieSegmentation(model_selection=1) as selfie_segmentation:
+        rgb_img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+        results = selfie_segmentation.process(rgb_img)
+        mask = results.segmentation_mask
+        condition = mask > 0.5
+        bg_color = np.ones(img.shape, dtype=np.uint8) * 255
+        output_img = np.where(condition[..., None], img, bg_color)
+        return output_img
+
+# Convert to PIL
+def convert_image(img):
+    if len(img.shape) == 2:
+        return Image.fromarray(img)
+    else:
+        return Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
+
+# Downloadable image
+def download_image(img):
+    buf = BytesIO()
+    img.save(buf, format="PNG")
+    byte_im = buf.getvalue()
+    return byte_im
+
+# ----- Streamlit App Starts Here -----
+def main():
+    st.set_page_config(page_title="Advanced Image Filter Studio", page_icon="🎨", layout="wide")
+    local_css()
+
+    st.title("🎨 Advanced Image Filter Studio")
+    st.write("Upload an image, apply **amazing filters**, and download your creation!")
+
+    # Sidebar Profiles
+    sidebar_profiles()
+
+    # Sidebar
+    st.sidebar.header("1. Upload Image")
+    uploaded_file = st.sidebar.file_uploader("Choose an image", type=["jpg", "jpeg", "png"])
+
+    if uploaded_file is not None:
+        file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
+        opencv_image = cv2.imdecode(file_bytes, 1)
+
+        # Filters Section
+        st.sidebar.header("2. Choose Filters")
+        grayscale = st.sidebar.checkbox("Grayscale")
+        blur = st.sidebar.checkbox("Blur")
+        canny = st.sidebar.checkbox("Edge Detection")
+        sepia = st.sidebar.checkbox("Sepia Effect")
+        sketch = st.sidebar.checkbox("Pencil Sketch")
+        invert = st.sidebar.checkbox("Invert Colors")
+        remove_bg = st.sidebar.checkbox("Remove Background (Simple)")
+
+        st.sidebar.header("3. Filter Parameters")
+        blur_strength = st.sidebar.slider("Blur Intensity (odd numbers)", 1, 49, 15, step=2)
+        threshold1 = st.sidebar.slider("Canny Threshold1", 50, 300, 100)
+        threshold2 = st.sidebar.slider("Canny Threshold2", 50, 300, 150)
+
+        # Process Image
+        final_image = opencv_image.copy()
+
+        with st.spinner("🖌️ Applying Filters..."):
+            if grayscale:
+                final_image = apply_grayscale(final_image)
+
+            if blur:
+                if len(final_image.shape) == 2:
+                    final_image = cv2.cvtColor(final_image, cv2.COLOR_GRAY2BGR)
+                final_image = apply_blur(final_image, blur_strength)
+
+            if canny:
+                if len(final_image.shape) != 2:
+                    final_image = cv2.cvtColor(final_image, cv2.COLOR_BGR2GRAY)
+                final_image = apply_canny(final_image, threshold1, threshold2)
+
+            if sepia:
+                if len(final_image.shape) == 2:
+                    final_image = cv2.cvtColor(final_image, cv2.COLOR_GRAY2BGR)
+                final_image = apply_sepia(final_image)
+
+            if sketch:
+                if len(final_image.shape) != 2:
+                    final_image = cv2.cvtColor(final_image, cv2.COLOR_BGR2GRAY)
+                final_image = apply_pencil_sketch(final_image)
+
+            if invert:
+                final_image = apply_invert(final_image)
+
+            if remove_bg:
+                if len(final_image.shape) != 3 or final_image.shape[2] != 3:
+                    final_image = cv2.cvtColor(final_image, cv2.COLOR_GRAY2BGR)
+                final_image = apply_background_removal(final_image)
+
+        # Columns to show images
+        col1, col2 = st.columns(2)
+
+        with col1:
+            st.subheader("Original Image")
+            st.image(uploaded_file, use_column_width=True)
+
+        with col2:
+            st.subheader("Processed Image")
+            final_pil = convert_image(final_image)
+            st.image(final_pil, use_column_width=True)
+
+        # Download button
+        st.markdown("---")
+        st.download_button(
+            label="📥 Download Processed Image",
+            data=download_image(final_pil),
+            file_name="processed_image.png",
+            mime="image/png"
+        )
+    else:
+        st.info("👈 Please upload an image from the sidebar to get started.")
+
+if __name__ == "__main__":
+    main()