Update app.py

app.py

@@ -1,5 +1,5 @@
-import qrcode
-from PIL import Image
+import barcode
+from barcode.writer import ImageWriter
 import base64
 import gradio as gr
 
@@ -9,49 +9,35 @@ def encode_file_to_base64(file_path):
         encoded_data = base64.b64encode(f.read()).decode("utf-8")
     return encoded_data
 
-def generate_qr(data: str, error_correction: int = qrcode.constants.ERROR_CORRECT_L, box_size: int = 10, border: int = 4, fill_color: str = "black", back_color: str = "white"):
-    """Generates a QR code from given data."""
-    final_url = f"https://flowly-ai.vercel.app/tools/qr/qr-code/read/{data}"
-    
-    qr = qrcode.QRCode(
-        version=1,
-        error_correction=error_correction,
-        box_size=box_size,
-        border=border,
-    )
-    qr.add_data(final_url)
-    qr.make(fit=True)
-    img_qr = qr.make_image(fill_color=fill_color, back_color=back_color)
-    
-    output_path = "qr_code.png"
-    img_qr.save(output_path)
+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
 
-def qr_code_interface(input_type, text, file, error_correction, box_size, border, fill_color, back_color):
+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_qr(encoded_text, error_correction, box_size, border, fill_color, back_color)
+        return generate_barcode(encoded_text, barcode_type)
     elif input_type == "File" and file:
         encoded_file = encode_file_to_base64(file.name)
-        return generate_qr(encoded_file, error_correction, box_size, border, fill_color, back_color)
+        return generate_barcode(encoded_file, barcode_type)
     else:
         return None
 
 interface = gr.Interface(
-    fn=qr_code_interface,
+    fn=barcode_interface,
     inputs=[
         gr.Radio(["Text", "File"], label="Input Type", value="Text"),
         gr.Textbox(label="Enter Text"),
         gr.File(label="Upload File"),
-        gr.Slider(0, 3, 1, label="Error Correction", value=0),
-        gr.Slider(1, 20, 1, label="Box Size", value=10),
-        gr.Slider(1, 10, 1, label="Border Size", value=4),
-        gr.ColorPicker(label="QR Fill Color", value="#000000"),
-        gr.ColorPicker(label="QR Background Color", value="#FFFFFF"),
+        gr.Dropdown(["code128", "ean13", "ean8"], label="Barcode Type", value="code128"),
     ],
-    outputs=gr.Image(type="filepath", label="Generated QR Code"),
-    description="Generate a QR Code from either text or any file with custom styles and colors.",
+    outputs=gr.Image(type="filepath", label="Generated Barcode"),
+    description="Generate a Barcode from either text or any file.",
     css="footer {visibility: hidden}"
 )
 
-interface.launch(share=True)
+interface.launch(share=True)