Update app.py

app.py

@@ -1,31 +1,40 @@
-import gradio as gr
-import cv2
-import numpy as np
-from PIL import Image
-
-def convert_to_sketch(image):
-  img = np.array(image)
-
-  #ubah gambar ke grayscale
-  gray_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-
-  #inversi warna
-  inverted_image = cv2.bitwise_not(gray_image)
-
-  #terapkan gaussian blur
-  blurred = cv2.GaussianBlur(inverted_image, (21, 21), sigmaX=0, sigmaY=0)
-
-  #inversi hasil blur
-  inverted_blurred = cv2.bitwise_not(blurred)
-
-  #buat sketsa
-  sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
-
-  return 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()
+import gradio as gr
+import cv2
+import numpy as np
+from PIL import Image
+import io
+
+def convert_to_sketch(image):
+    img = np.array(image)
+
+    #ubah gambar ke grayscale
+    gray_image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+
+    #inversi warna
+    inverted_image = cv2.bitwise_not(gray_image)
+
+    #terapkan gaussian blur
+    blurred = cv2.GaussianBlur(inverted_image, (21, 21), sigmaX=0, sigmaY=0)
+
+    #inversi hasil blur
+    inverted_blurred = cv2.bitwise_not(blurred)
+
+    #buat sketsa
+    sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
+
+    # Konversi sketsa menjadi gambar PIL dan simpan sebagai JPG
+    pil_image = Image.fromarray(sketch)
+    
+    # Menyimpan gambar dalam format jpg
+    img_byte_arr = io.BytesIO()
+    pil_image.save(img_byte_arr, format="JPEG")
+    img_byte_arr = img_byte_arr.getvalue()
+
+    return img_byte_arr
+
+#interface
+gr.Interface(fn=convert_to_sketch,
+            inputs="image",
+            outputs="image",
+            title="Konversi gambar ke sketsa",
+            description="Unggah gambar dan dapatkan versi sketsa dalam format JPG").launch()