Create app.py

app.py

@@ -0,0 +1,170 @@
+import streamlit as st
+import cv2
+import numpy as np
+from PIL import Image
+import io
+
+def apply_grabcut(image):
+    # Create a mask
+    mask = np.zeros(image.shape[:2], np.uint8)
+    
+    # Model for background and foreground
+    bgd_model = np.zeros((1, 65), np.float64)
+    fgd_model = np.zeros((1, 65), np.float64)
+    
+    # Define a rectangle around the foreground object
+    rect = (10, 10, image.shape[1]-10, image.shape[0]-10)
+    
+    # Apply GrabCut algorithm
+    cv2.grabCut(image, mask, rect, bgd_model, fgd_model, 5, cv2.GC_INIT_WITH_RECT)
+    
+    # Create a mask where sure and likely foreground are 1, and background are 0
+    mask2 = np.where((mask == 2) | (mask == 0), 0, 1).astype('uint8')
+    
+    # Separate the foreground and background
+    foreground = image * mask2[:, :, np.newaxis]
+    background = cv2.GaussianBlur(image, (31, 31), 0) * (1 - mask2[:, :, np.newaxis])
+    
+    return foreground + background
+
+# Function to convert image to sketch with adjustable outline thickness
+def image_to_sketch(image, kernel_size=3):
+    gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    inverted_image = 255 - gray_image
+    blurred_image = cv2.GaussianBlur(inverted_image, (31, 31), 0)
+    inverted_blurred = 255 - blurred_image
+    sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
+    
+    # Apply adaptive thresholding to enhance edges
+    adaptive_thresh = cv2.adaptiveThreshold(sketch, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY, 9, 2)
+    
+    # Apply morphological operation to thicken the outlines
+    kernel = np.ones((kernel_size, kernel_size), np.uint8)
+    sketch = cv2.dilate(adaptive_thresh, kernel, iterations=1)
+    
+    return sketch
+
+# Streamlit app layout
+st.set_page_config(page_title="Image to Sketch Converter", page_icon="🎨", layout="centered")
+
+# Custom CSS for heading color and footer positioning
+st.markdown("""
+    <style>
+    .title {
+        color: blue;
+        font-size: 2.5em;
+        font-weight: bold;
+        text-align: center;
+    }
+    .footer {
+        position: relative;
+        bottom: 0;
+        width: 100%;
+        background-color: orange;
+        text-align: center;
+        text-color: black;
+        padding: 10px;
+        font-weight: bold;
+        margin-top: 50px;
+    }
+    .content {
+        margin-bottom: 70px;
+    }
+    .spacing {
+        margin: 10px 10px;
+    }
+    .centered-button {
+        display: flex;
+        justify-content: center;
+        align-items: center;
+        gap: 10px;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+# Title and description
+st.markdown('<p class="title">🎨 Image to Sketch Converter</p>', unsafe_allow_html=True)
+st.markdown("""
+    Convert your images into beautiful sketches with this simple app.
+    Upload an image, and get the sketch version instantly! You can even download the sketch.
+    """)
+
+# Example conversions
+st.subheader("Example Conversions")
+
+# Load and display example image
+example_image_path = 'Dog.jpg'
+example_image = cv2.imread(example_image_path)
+
+if example_image is not None:
+    # Convert BGR to RGB for correct color display
+    example_image_rgb = cv2.cvtColor(example_image, cv2.COLOR_BGR2RGB)
+    example_blurred_background = apply_grabcut(example_image)
+    example_sketch = image_to_sketch(example_blurred_background)
+
+    col1, col2 = st.columns(2)
+
+    with col1:
+        st.image(example_image_rgb, caption='Original Image', use_column_width=True)
+    with col2:
+        st.image(example_sketch, caption='Sketch Image', use_column_width=True)
+else:
+    st.error(f"Failed to load example image from path: {example_image_path}")
+
+# User upload section
+st.subheader("Upload Your Image")
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
+
+if uploaded_file is not None:
+    # Load the image
+    image = Image.open(uploaded_file)
+    image_np = np.array(image)
+    
+    # Determine the format of the uploaded image
+    image_format = image.format.lower()
+    
+    st.write("Converting...")
+
+    # Kernel size slider for adjusting outline thickness
+    kernel_size = st.slider('Adjust Outline Thickness', 1, 10, 3)
+    
+    # Apply GrabCut to blur background
+    blurred_background_image = apply_grabcut(image_np)
+    
+    # Convert the image with blurred background to a sketch with the selected kernel size
+    sketch = image_to_sketch(blurred_background_image, kernel_size)
+    
+    col3, col4 = st.columns(2)
+
+    with col3:
+        st.image(image, caption='Uploaded Image', use_column_width=True)
+    with col4:
+        st.image(sketch, caption='Sketch', use_column_width=True)
+
+    # Add some space before the button
+    st.markdown('<div class="spacing"></div>', unsafe_allow_html=True)
+    
+    # Convert the sketch to an image and save to an in-memory file object
+    sketch_image = Image.fromarray(sketch)
+    buf = io.BytesIO()
+    sketch_image.save(buf, format=image_format.upper())
+    byte_im = buf.getvalue()
+    
+    # Provide a download link for the sketch image in the center
+    st.markdown('<div class="centered-button">', unsafe_allow_html=True)
+    btn = st.download_button(
+        label="Download Sketch",
+        data=byte_im,
+        file_name=f"sketch.{image_format}",
+        mime=f"image/{image_format}"
+    )
+    st.markdown('</div>', unsafe_allow_html=True)
+else:
+    st.info("Please upload an image to convert.")
+
+# Footer
+st.markdown("""
+    <div class="footer">
+        Made by Mallela Preethi
+    </div>
+    """, unsafe_allow_html=True)