Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 from io import BytesIO
 from PIL import Image
 import datetime
+import os
 
 def encode_file_to_base64(file_path):
     """Reads any file and encodes it to base64."""
@@ -31,10 +32,12 @@ def generate_barcode(data: str, barcode_type: str = "code128"):
         file_name = f"flowly_ai_premium_barcode_generator_{timestamp}.png"
         
         # Simpan file gambar sebagai PNG
-        img.save(file_name)
+        file_path = os.path.join("generated_barcodes", file_name)
+        os.makedirs(os.path.dirname(file_path), exist_ok=True)  # Ensure the directory exists
+        img.save(file_path)
         
-        # Mengembalikan gambar untuk ditampilkan di Gradio dan nama file
-        return img, file_name
+        # Mengembalikan gambar untuk ditampilkan di Gradio dan path file untuk diunduh
+        return img, file_path
     except Exception as e:
         return f"Error: {str(e)}", None
 
@@ -55,7 +58,7 @@ interface = gr.Interface(
         gr.File(label="Upload File"),
         gr.Dropdown(["code128", "ean13", "ean8"], label="Barcode Type", value="code128"),
     ],
-    outputs=[gr.Image(type="numpy", label="Generated Barcode"), gr.Download(label="Download Barcode")],
+    outputs=[gr.Image(type="numpy", label="Generated Barcode"), gr.File(label="Download Barcode")],
     description="Generate a Barcode from either text or any file.",
     css="footer {visibility: hidden}"
 )