Update app.py

app.py

@@ -1,14 +1,8 @@
-# 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
-
+import cv2
 from PIL import Image, ImageTk
 import numpy as np
 import gradio as gr
+import os
 
 # Function to convert image to sketch
 def convert_to_sketch(img, blur_strength, brightness, contrast):
@@ -51,7 +45,13 @@ def sketch_interface(image, blur_strength, brightness, contrast):
     sketch = convert_to_sketch(image, blur_strength, brightness, contrast)
     
     # Ensure the output is a valid image (PIL Image)
-    return Image.fromarray(sketch)
+    output_image = Image.fromarray(sketch)
+    
+    # Save the processed image temporarily
+    output_image.save("sketch_output.png")
+    
+    # Return the processed sketch image and the file path for download
+    return output_image, "sketch_output.png"
 
 # Create Gradio interface
 interface = gr.Interface(
@@ -62,7 +62,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"), 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."
 )