Update app.py

app.py

@@ -2,6 +2,8 @@ import barcode
 from barcode.writer import ImageWriter
 import base64
 import gradio as gr
+from io import BytesIO
+from PIL import Image
 
 def encode_file_to_base64(file_path):
     """Reads any file and encodes it to base64."""
@@ -10,17 +12,25 @@ def encode_file_to_base64(file_path):
     return encoded_data
 
 def generate_barcode(data: str, barcode_type: str = "code128"):
-    """Generates a barcode from given data."""
-    barcode_class = barcode.get_barcode_class(barcode_type)
-    barcode_instance = barcode_class(data, writer=ImageWriter())
-    output_path = "barcode.png"
-    barcode_instance.save(output_path)
-    return output_path
+    """Generates a barcode from given data and returns an image in memory."""
+    try:
+        barcode_class = barcode.get_barcode_class(barcode_type)
+        barcode_instance = barcode_class(data, writer=ImageWriter())
+
+        # Simpan barcode ke dalam BytesIO
+        output = BytesIO()
+        barcode_instance.write(output)
+
+        # Buka gambar dengan PIL untuk memastikan kompatibilitas dengan Gradio
+        output.seek(0)
+        img = Image.open(output)
+        return img
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 def barcode_interface(input_type, text, file, barcode_type):
     if input_type == "Text" and text:
-        encoded_text = base64.b64encode(text.encode('utf-8')).decode('utf-8')
-        return generate_barcode(encoded_text, barcode_type)
+        return generate_barcode(text, barcode_type)
     elif input_type == "File" and file:
         encoded_file = encode_file_to_base64(file.name)
         return generate_barcode(encoded_file, barcode_type)
@@ -35,7 +45,7 @@ interface = gr.Interface(
         gr.File(label="Upload File"),
         gr.Dropdown(["code128", "ean13", "ean8"], label="Barcode Type", value="code128"),
     ],
-    outputs=gr.Image(type="filepath", label="Generated Barcode"),
+    outputs=gr.Image(type="numpy", label="Generated Barcode"),
     description="Generate a Barcode from either text or any file.",
     css="footer {visibility: hidden}"
 )