Update app.py

app.py

@@ -6,7 +6,7 @@ except ImportError:
     subprocess.check_call(["pip", "install", "opencv-python-headless"])
     import cv2
 
-from PIL import Image, ImageTk
+from PIL import Image
 import numpy as np
 import gradio as gr
 import os
@@ -17,7 +17,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)
     
     # Invert the grayscale image
     img_inverted = cv2.bitwise_not(img_gray)
@@ -51,7 +51,7 @@ 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 numpy array to PIL Image before returning
     output_image = Image.fromarray(sketch)
     
     # Return the processed sketch image for display
@@ -61,14 +61,14 @@ def sketch_interface(image, blur_strength, brightness, contrast):
 interface = gr.Interface(
     fn=sketch_interface,
     inputs=[
-        gr.Image(type="pil", label="Upload Image"),
+        gr.Image(type="pil", label="Upload Image", image_mode="RGB", shape=(256, 256)),  # Support for various formats
         gr.Slider(1, 51, step=2, label="Blur Strength", value=21),
         gr.Slider(-100, 100, label="Brightness", value=0),
         gr.Slider(1, 3, step=0.1, label="Contrast", value=1)
     ],
     outputs=[gr.Image(type="pil", label="Sketch Output")],  # Output is now just the image
     title="Cartoon to Sketch Converter",
-    description="Upload an image to convert it into a sketch, adjust the blur strength, brightness, and contrast for different effects."
+    description="Upload an image (PNG, JPEG, or others) to convert it into a sketch. Adjust the blur strength, brightness, and contrast for different effects."
 )
 
 # Launch the Gradio app