Update app.py

app.py

@@ -1,5 +1,12 @@
-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
 import os
@@ -35,7 +42,7 @@ def convert_to_sketch(img, blur_strength, brightness, contrast):
 
 # Function to adjust brightness and contrast
 def adjust_brightness_contrast(img, brightness, contrast):
-    # Apply contrast adjustment (alpha for contrast, beta for brightness)
+    # Apply contrast adjustment
     img = cv2.convertScaleAbs(img, alpha=contrast, beta=brightness)
     return img
 
@@ -48,11 +55,10 @@ def sketch_interface(image, blur_strength, brightness, contrast):
     output_image = Image.fromarray(sketch)
     
     # Save the processed image temporarily
-    output_path = "sketch_output.png"
-    output_image.save(output_path)
+    output_image.save("sketch_output.png")
     
-    # Return the processed sketch image and a downloadable file path
-    return output_image, output_path
+    # Return the processed sketch image
+    return output_image, "sketch_output.png"
 
 # Create Gradio interface
 interface = gr.Interface(
@@ -63,12 +69,9 @@ 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"),
-        gr.File(label="Download Sketch")
-    ],
+    outputs=[gr.Image(type="pil", label="Sketch Output"), gr.File(label="Download Sketch")],
     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 to convert it into a sketch, adjust the blur strength, brightness, and contrast for different effects."
 )
 
 # Launch the Gradio app