Update app.py

app.py

@@ -1,9 +1,16 @@
-import cv2
-from PIL import Image
+# Attempt to import cv2 and install opencv-python-headless if not available
+try:
+    import cv2
+except ImportError:
+    import subprocess
+    subprocess.check_call(["pip", "install", "opencv-python-headless"])
+    import cv2
+
+from PIL import Image, ImageTk
 import numpy as np
 import gradio as gr
 
-# Convert image to sketch function
+# Function to convert image to sketch
 def convert_to_sketch(img, blur_strength):
     img = np.array(img)  # Convert PIL Image to numpy array
     img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
@@ -22,8 +29,11 @@ def sketch_interface(image, blur_strength):
 # Create Gradio interface
 interface = gr.Interface(
     fn=sketch_interface,
-    inputs=[gr.Image(type="pil"), gr.Slider(1, 51, step=2, label="Blur Strength", value=21)],
-    outputs=gr.Image(type="pil"),
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Slider(1, 51, step=2, label="Blur Strength", value=21)
+    ],
+    outputs=gr.Image(type="pil", label="Sketch Output"),
     title="Cartoon to Sketch Converter",
     description="Upload an image to convert it into a sketch, and adjust the blur strength for different effects."
 )