Update app.py

app.py

@@ -3,36 +3,30 @@ import cv2
 import numpy as np
 from PIL import Image
 
-def pencil_effect(img, high_blur_value, block_size, c_value):
+def color_pencil_effect(img, median_blur_value, block_size, c_value):
     # Convert to grayscale
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
 
-    # Apply high Gaussian blur
-    blurred = cv2.GaussianBlur(gray, (high_blur_value, high_blur_value), 0)
+    # Apply median blur to smooth the grayscale image
+    gray = cv2.medianBlur(gray, median_blur_value)
 
     # Adaptive thresholding for edges
-    edges = cv2.adaptiveThreshold(blurred, 255,
+    edges = cv2.adaptiveThreshold(gray, 255,
                                   cv2.ADAPTIVE_THRESH_MEAN_C,
                                   cv2.THRESH_BINARY,
                                   blockSize=block_size,
                                   C=c_value)
 
-    return edges
+    # Reduce colors using bilateral filter to achieve a pencil effect
+    color = cv2.bilateralFilter(img, d=9, sigmaColor=150, sigmaSpace=150)
 
-def cartoonize_image(img, high_blur_value, block_size, c_value):
-    # Apply pencil effect to get edges
-    edges = pencil_effect(img, high_blur_value, block_size, c_value)
+    # Combine edges with the color image to create the pencil effect
+    pencil_effect = cv2.bitwise_and(color, color, mask=edges)
 
-    # Reduce colors using bilateral filter
-    color = cv2.bilateralFilter(img, d=9, sigmaColor=100, sigmaSpace=100)
-
-    # Combine edges with color image to create a more artistic effect
-    cartoon = cv2.bitwise_and(color, color, mask=edges)
-
-    return cartoon
+    return pencil_effect
 
 # Streamlit app layout
-st.title("Pencil Effect Cartoonizer")
+st.title("Color Pencil Effect Creator")
 st.subheader("Upload an image and adjust parameters")
 
 # Upload image
@@ -46,34 +40,34 @@ if uploaded_file is not None:
     # Create a container for sliders on the left side
     with st.sidebar:
         st.header("Adjust Parameters")
-        high_blur_value = st.slider("High Blur Value", min_value=3, max_value=31, value=21, step=2)  # Increased range for more blur
+        median_blur_value = st.slider("Median Blur Value", min_value=3, max_value=11, value=5, step=2)
         block_size = st.slider("Block Size", min_value=3, max_value=21, value=9, step=2)
         c_value = st.slider("C Value", min_value=0, max_value=20, value=5)  # Increased default value for more effect
 
-    # Show a spinner while processing the cartoonization
+    # Show a spinner while processing the pencil effect
     with st.spinner('Processing...'):
-        # Cartoonize the image with selected parameters in real-time
-        cartoon_image = cartoonize_image(img, high_blur_value, block_size, c_value)
+        # Apply the color pencil effect with selected parameters in real-time
+        pencil_image = color_pencil_effect(img, median_blur_value, block_size, c_value)
 
-    # Display original and cartoonized images side by side on the right side
+    # Display original and pencil effect images side by side on the right side
     col1, col2 = st.columns(2)
     
     with col1:
         st.image(img.astype(np.uint8), caption='Original Image', use_column_width=True)
 
     with col2:
-        st.image(cartoon_image.astype(np.uint8), caption='Pencil Effect Cartoonized Image', use_column_width=True)
+        st.image(pencil_image.astype(np.uint8), caption='Color Pencil Effect', use_column_width=True)
 
-    # Download button for cartoonized image
-    if st.button("Download Cartoonized Image"):
-        # Convert the cartoonized image to PIL format for download
-        cartoon_image_pil = Image.fromarray(cartoon_image)
-        cartoon_image_pil.save("pencil_cartoonized_image.png")
+    # Download button for the pencil effect image
+    if st.button("Download Pencil Effect Image"):
+        # Convert the pencil effect image to PIL format for download
+        pencil_image_pil = Image.fromarray(pencil_image)
+        pencil_image_pil.save("color_pencil_effect_image.png")
         
-        with open("pencil_cartoonized_image.png", "rb") as f:
+        with open("color_pencil_effect_image.png", "rb") as f:
             st.download_button(
                 label="Download Image",
                 data=f,
-                file_name="pencil_cartoonized_image.png",
+                file_name="color_pencil_effect_image.png",
                 mime="image/png"
             )