Update app.py

app.py

@@ -1,4 +1,4 @@
-# Attempt to import opencv-python-headless and install it if not available
+# Attempt to import cv2 and install opencv-python-headless if not available
 try:
     import cv2
 except ImportError:
@@ -6,37 +6,36 @@ 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
 
-# Function to convert the image to a sketch
+# Function to convert image to sketch
 def convert_to_sketch(img, blur_strength, brightness, contrast):
     # Convert PIL Image to numpy array
     img = np.array(img)
-    
+
     # Convert the image to grayscale
     img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    
+
     # Invert the grayscale image
     img_inverted = cv2.bitwise_not(img_gray)
-    
+
     # Apply Gaussian blur to the inverted image
     img_blur = cv2.GaussianBlur(img_inverted, (blur_strength, blur_strength), sigmaX=0, sigmaY=0)
-    
+
     # Blend the grayscale and blurred inverted images to create the sketch effect
     img_blend = cv2.divide(img_gray, 255 - img_blur, scale=256)
-    
+
     # Create a white background
     white_background = 255 * np.ones_like(img_blend)
-    
+
     # Add the blended image to the white background
     sketch_with_bg = cv2.addWeighted(img_blend, 1, white_background, 1, 0)
-    
+
     # Adjust brightness and contrast
     sketch_with_bg = adjust_brightness_contrast(sketch_with_bg, brightness, contrast)
-    
+
     return sketch_with_bg
 
 # Function to adjust brightness and contrast
@@ -49,10 +48,16 @@ def adjust_brightness_contrast(img, brightness, contrast):
 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)
     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
 
@@ -65,7 +70,7 @@ interface = gr.Interface(
         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"),
+    outputs=[gr.Image(type="pil", label="Sketch Output")],
     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."
 )