Update app.py

app.py

@@ -6,7 +6,7 @@ from PIL import Image
 def cartoonize_image(img, median_blur_value, block_size, c_value, d_value, sigma_color, sigma_space):
     # Convert to grayscale
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    
+
     # Apply median blur
     gray = cv2.medianBlur(gray, median_blur_value)
 
@@ -20,19 +20,15 @@ def cartoonize_image(img, median_blur_value, block_size, c_value, d_value, sigma
     # Bilateral filter for color smoothing
     color = cv2.bilateralFilter(img, d=d_value, sigmaColor=sigma_color, sigmaSpace=sigma_space)
 
-    # Convert color image to grayscale and then back to BGR to reduce colors
-    color_gray = cv2.cvtColor(color, cv2.COLOR_BGR2GRAY)
-    color_gray = cv2.medianBlur(color_gray, 5)
-    
     # Create a sketch effect by blending edges with the smoothed color image
-    sketch = cv2.bitwise_and(color_gray, color_gray, mask=edges)
+    sketch = cv2.bitwise_and(color, color, mask=edges)
 
     # Ensure both images are of the same size before blending
-    if color.shape != sketch.shape:
+    if color.shape[:2] != sketch.shape[:2]:
         sketch = cv2.resize(sketch, (color.shape[1], color.shape[0]))
 
     # Combine the sketch with the original color image to create a more artistic effect
-    cartoon = cv2.addWeighted(color, 0.5, sketch[:, :, np.newaxis], 0.5, 0)  # Ensure sketch has 3 channels
+    cartoon = cv2.addWeighted(color, 0.5, sketch, 0.5, 0)
 
     return cartoon
 
@@ -67,10 +63,10 @@ if uploaded_file is not None:
     col1, col2 = st.columns(2)
     
     with col1:
-        st.image(img, caption='Original Image', use_column_width=True)
+        st.image(img.astype(np.uint8), caption='Original Image', use_column_width=True)
 
     with col2:
-        st.image(cartoon_image, caption='Cartoonized Image', use_column_width=True)
+        st.image(cartoon_image.astype(np.uint8), caption='Cartoonized Image', use_column_width=True)
 
     # Download button for cartoonized image
     if st.button("Download Cartoonized Image"):