Update app.py

app.py

@@ -16,7 +16,7 @@ def convert_to_sketch(img, blur_strength, brightness, contrast):
     img = np.array(img)
 
     # Convert the image to grayscale
-    img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    img_gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)  # Use RGB2GRAY for PIL images
 
     # Invert the grayscale image
     img_inverted = cv2.bitwise_not(img_gray)
@@ -49,15 +49,9 @@ def sketch_interface(image, blur_strength, brightness, contrast):
     # Convert the input image to a sketch with adjustments
     sketch = convert_to_sketch(image, blur_strength, brightness, contrast)
 
-    # Ensure the output is a valid image (PIL Image)
+    # Convert the processed numpy array back to a PIL Image
     output_image = Image.fromarray(sketch)
 
-    # Check if the image was properly created
-    if output_image:
-        print("Image processed successfully!")
-    else:
-        print("Failed to process image!")
-
     # Return the processed sketch image
     return output_image