Update app.py

app.py

@@ -1,5 +1,5 @@
 import cv2
-from PIL import Image, ImageTk
+from PIL import Image
 import numpy as np
 import gradio as gr
 import os
@@ -35,7 +35,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
+    # Apply contrast adjustment (alpha for contrast, beta for brightness)
     img = cv2.convertScaleAbs(img, alpha=contrast, beta=brightness)
     return img
 
@@ -48,10 +48,11 @@ def sketch_interface(image, blur_strength, brightness, contrast):
     output_image = Image.fromarray(sketch)
     
     # Save the processed image temporarily
-    output_image.save("sketch_output.png")
+    output_path = "sketch_output.png"
+    output_image.save(output_path)
     
-    # Return the processed sketch image and the file path for download
-    return output_image, "sketch_output.png"
+    # Return the processed sketch image and a downloadable file path
+    return output_image, output_path
 
 # Create Gradio interface
 interface = gr.Interface(
@@ -62,9 +63,12 @@ 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