Create app.py

app.py

@@ -0,0 +1,91 @@
+import gradio as gr
+import pandas as pd
+from reportlab.lib.pagesizes import inch
+from reportlab.pdfgen import canvas
+from reportlab.lib.utils import ImageReader
+import barcode
+from barcode.writer import ImageWriter
+import os
+import zipfile
+
+# Ensure 'labels' directory exists
+os.makedirs("labels", exist_ok=True)
+
+def generate_label(data, index):
+    """
+    Generate a 4x6 inch shipping label as a PDF.
+    """
+    pdf_filename = f"labels/shipping_label_{index+1}.pdf"
+    c = canvas.Canvas(pdf_filename, pagesize=(4 * inch, 6 * inch))
+
+    # Draw sender details (top-left)
+    c.setFont("Helvetica-Bold", 12)
+    c.drawString(30, 500, "From:")
+    c.setFont("Helvetica", 10)
+    c.drawString(30, 485, data['Sender Name'])
+    c.drawString(30, 470, data['Sender Address'])
+    c.drawString(30, 455, data['Sender City, State ZIP'])
+
+    # Draw recipient details (middle)
+    c.setFont("Helvetica-Bold", 12)
+    c.drawString(30, 400, "To:")
+    c.setFont("Helvetica", 10)
+    c.drawString(30, 385, data['Recipient Name'])
+    c.drawString(30, 370, data['Recipient Address'])
+    c.drawString(30, 355, data['Recipient City, State ZIP'])
+
+    # Draw shipping service
+    c.setFont("Helvetica-Bold", 12)
+    c.drawString(30, 320, f"Service: {data['Shipping Service']}")
+
+    # Draw tracking number
+    c.setFont("Helvetica-Bold", 12)
+    c.drawString(30, 280, "Tracking #:")
+    c.setFont("Helvetica", 12)
+    c.drawString(30, 260, data['Tracking Number'])
+
+    # Generate barcode
+    barcode_filename = f"labels/barcode_{index+1}.png"
+    ean = barcode.get_barcode_class('code128')
+    ean_obj = ean(data['Tracking Number'], writer=ImageWriter())
+    ean_obj.save(barcode_filename)
+
+    # Add barcode to label
+    barcode_img = ImageReader(barcode_filename)
+    c.drawImage(barcode_img, 30, 120, width=300, height=100)
+
+    # Save PDF
+    c.showPage()
+    c.save()
+
+def process_csv(file):
+    """
+    Read CSV and generate labels.
+    """
+    df = pd.read_csv(file.name)
+
+    # Generate labels for each row
+    for i, row in df.iterrows():
+        generate_label(row, i)
+
+    # Zip all PDFs into a single file for download
+    zip_filename = "labels/shipping_labels.zip"
+    with zipfile.ZipFile(zip_filename, "w") as zipf:
+        for label_file in os.listdir("labels"):
+            if label_file.endswith(".pdf"):
+                zipf.write(f"labels/{label_file}", label_file)
+
+    return zip_filename
+
+# Gradio Interface
+interface = gr.Interface(
+    fn=process_csv,
+    inputs=gr.File(label="Upload CSV File"),
+    outputs=gr.File(label="Download Shipping Labels (ZIP)"),
+    title="4x6 Shipping Label Generator",
+    description="Upload a CSV file with shipping details, and this app will generate professional 4x6 shipping labels with barcodes.",
+)
+
+# Run the app
+if __name__ == "__main__":
+    interface.launch()