Update app.py

app.py

@@ -2,30 +2,36 @@ import gradio as gr
 import cv2
 import numpy as np
 from PIL import Image
+import os
 
 def convert_to_sketch(image):
-  img = np.array(image)
+    img = np.array(image)
 
-  #ubah gambar ke grayscale
-  gray_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    gray_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
 
-  #inversi warna
-  inverted_image = cv2.bitwise_not(gray_image)
+    inverted_image = cv2.bitwise_not(gray_image)
 
-  #terapkan gaussian blur
-  blurred = cv2.GaussianBlur(inverted_image, (21, 21), sigmaX=0, sigmaY=0)
+    blurred = cv2.GaussianBlur(inverted_image, (21, 21), sigmaX=0, sigmaY=0)
 
-  #inversi hasil blur
-  inverted_blurred = cv2.bitwise_not(blurred)
+    inverted_blurred = cv2.bitwise_not(blurred)
 
-  #buat sketsa
-  sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
+    sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
 
-  return sketch
+    sketch_image = Image.fromarray(sketch)
 
-#interface
-gr.Interface(fn=convert_to_sketch,
-            inputs="image",
-            outputs="image",
-            title="Konversi gambar ke sketsa",
-            description="Unggah gambar dan dapatkan versi sketsa").launch()
+    output_path = "sketch_result.png"
+    sketch_image.save(output_path)
+
+    return sketch_image, output_path
+
+# Interface
+gr.Interface(
+    fn=convert_to_sketch,
+    inputs="image",
+    outputs=[
+        gr.Image(label="Pratinjau Sketsa"),
+        gr.File(label="Unduh Gambar Sketsa")
+    ],
+    title="Konversi gambar ke sketsa",
+    description="Unggah gambar dan dapatkan versi sketsa sebagai pratinjau dan file unduhan"
+).launch()