Update app.py

app.py

@@ -1,135 +1,197 @@
 import os
-import pandas as pd
+import zipfile
 import gradio as gr
+import pandas as pd
 from reportlab.pdfgen import canvas
-from reportlab.lib.pagesizes import letter
-from reportlab.lib import colors
+from reportlab.lib.pagesizes import inch
 from reportlab.lib.utils import ImageReader
-from reportlab.lib.styles import getSampleStyleSheet
 import barcode
 from barcode.writer import ImageWriter
 
-# Ensure output directories exist
+# Create folders to store generated files
 LABELS_DIR = "labels"
 BARCODES_DIR = "barcodes"
 os.makedirs(LABELS_DIR, exist_ok=True)
 os.makedirs(BARCODES_DIR, exist_ok=True)
 
-# Function to process the uploaded file
-def process_file(file_path):
+############################################
+# 1) Draw the static collage / background
+############################################
+def draw_label_collage(c, width, height):
+    """
+    Draws the static 'collage' or background layout 
+    (like the big "F", USPS text, lines, placeholders).
+    """
+    # Big "F" at top-left
+    c.setFont("Helvetica-Bold", 72)
+    c.drawString(15, height - 90, "F")
+
+    # USPS postage info in top-left area
+    c.setFont("Helvetica", 8)
+    c.drawString(70, height - 25, "US POSTAGE AND FEES PAID")
+    c.drawString(70, height - 35, "Apr 12 2018")
+    c.drawString(70, height - 45, "Mailed from ZIP 77024")
+    c.drawString(70, height - 55, "First-Class Pkg Svc")
+    c.drawString(70, height - 65, "CommercialBasePrice")
+
+    # Placeholder for QR code (top-right)
+    c.setFont("Helvetica", 6)
+    c.drawString(width - 60, height - 20, "QR CODE")
+
+    # "USPS FIRST-CLASS PKG" bar
+    c.setFont("Helvetica-Bold", 14)
+    c.drawString(15, height - 110, "USPS FIRST-CLASS PKG")
+
+    # Horizontal line
+    c.setLineWidth(1)
+    c.line(15, height - 115, width - 15, height - 115)
+
+############################################
+# 2) Draw the dynamic text & overlay
+############################################
+def draw_label_text(c, row, width, height):
+    """
+    Overlays dynamic text from each row:
+    Warehouse info, recipient address, tracking number, etc.
+    """
+    # Example columns from your data (adjust as needed)
+    warehouse_line1 = row.get("Warehouse Line1", "WAREHOUSE 2")
+    warehouse_line2 = row.get("Warehouse Line2", "11919 WINK RD")
+    warehouse_line3 = row.get("Warehouse Line3", "HOUSTON TX 77024-7134")
+    order_number    = row.get("Order Number", "286")
+
+    recipient_name  = row.get("Recipient Name", "John Doe")
+    recipient_addr1 = row.get("Recipient Addr1", "321 Street Over There")
+    recipient_addr2 = row.get("Recipient Addr2", "Salt Lake City, UT 11212")
+    recipient_addr3 = row.get("Recipient Addr3", "United States")
+
+    tracking_number = row.get("Tracking Number", "9400 1102 0079 3961 8936 91")
+
+    # Warehouse info (top-left under the bar)
+    c.setFont("Helvetica-Bold", 9)
+    c.drawString(15, height - 130, warehouse_line1)
+    c.setFont("Helvetica", 9)
+    c.drawString(15, height - 145, warehouse_line2)
+    c.drawString(15, height - 160, warehouse_line3)
+
+    # Order number (top-right)
+    c.drawString(width - 80, height - 130, f"Order: {order_number}")
+
+    # Recipient info (bold name)
+    c.setFont("Helvetica-Bold", 12)
+    c.drawString(15, height - 200, recipient_name)
+    c.setFont("Helvetica", 10)
+    c.drawString(15, height - 215, recipient_addr1)
+    c.drawString(15, height - 230, recipient_addr2)
+    c.drawString(15, height - 245, recipient_addr3)
+
+    # "USPS TRACKING #"
+    c.setFont("Helvetica-Bold", 12)
+    c.drawString(15, height - 280, "USPS TRACKING #")
+
+    # Tracking number
+    c.setFont("Helvetica-Bold", 10)
+    c.drawString(15, height - 295, tracking_number)
+
+############################################
+# 3) Create the 4x6 USPS-style label
+############################################
+def create_usps_label(row, index=1):
+    """
+    Creates a 4x6 inch PDF label using the collage background,
+    then overlays dynamic text and a barcode at the bottom.
+    """
+    # 3A) Generate Code128 barcode
+    tracking_number = row.get("Tracking Number", "9400110200793961893691")
+    barcode_filename = os.path.join(BARCODES_DIR, f"barcode_{index}.png")
+    code128 = barcode.get_barcode_class("code128")
+    code128_obj = code128(tracking_number, writer=ImageWriter())
+    code128_obj.save(barcode_filename)
+
+    # 3B) Prepare a 4x6 inch PDF
+    width, height = 4 * inch, 6 * inch
+    label_filename = os.path.join(LABELS_DIR, f"label_{index}.pdf")
+    c = canvas.Canvas(label_filename, pagesize=(width, height))
+
+    # (1) Draw static collage
+    draw_label_collage(c, width, height)
+
+    # (2) Overlay dynamic text
+    draw_label_text(c, row, width, height)
+
+    # (3) Place the barcode at the bottom
+    if os.path.exists(barcode_filename):
+        c.drawImage(barcode_filename, 15, 0.8 * inch, width=3 * inch, height=0.8 * inch)
+    else:
+        c.setFont("Helvetica", 10)
+        c.drawString(15, 0.8 * inch, "Barcode generation failed.")
+
+    c.showPage()
+    c.save()
+    return label_filename
+
+############################################
+# 4) Gradio Function to Process File Upload
+############################################
+def process_file(file_obj):
     try:
-        # Load data from file
-        if file_path.endswith('.csv'):
-            df = pd.read_csv(file_path)
-        elif file_path.endswith(('.xls', '.xlsx')):
-            df = pd.read_excel(file_path)
+        # Clean old label/barcode files
+        for f in os.listdir(LABELS_DIR):
+            os.remove(os.path.join(LABELS_DIR, f))
+        for f in os.listdir(BARCODES_DIR):
+            os.remove(os.path.join(BARCODES_DIR, f))
+
+        # Read the file
+        if file_obj.name.endswith(".csv"):
+            df = pd.read_csv(file_obj)
+        elif file_obj.name.endswith((".xls", ".xlsx")):
+            df = pd.read_excel(file_obj)
         else:
             return "Invalid file format. Please upload a .csv, .xls, or .xlsx file."
 
-        # Ensure required columns exist
-        required_cols = [
-            "Order ID", "Order Date", "Shipper Name", "Shipper Address", "Shipper Phone",
-            "Receiver Name", "Receiver Address", "Receiver Phone", "Package Weight (kg)",
-            "Shipping Method", "Tracking Number", "Estimated Delivery Date"
+        # Basic check for columns
+        needed_cols = [
+            "Warehouse Line1", "Warehouse Line2", "Warehouse Line3", "Order Number",
+            "Recipient Name", "Recipient Addr1", "Recipient Addr2", "Recipient Addr3",
+            "Tracking Number"
         ]
-        
-        missing_cols = [col for col in required_cols if col not in df.columns]
-        if missing_cols:
-            return f"Missing columns: {', '.join(missing_cols)}"
+        missing = [c for c in needed_cols if c not in df.columns]
+        if missing:
+            return f"Missing columns: {', '.join(missing)}"
 
-        # Process each row and generate labels
-        label_files = []
+        # Generate a PDF label for each row
         for i, row in df.iterrows():
-            label_path = create_shipping_label(row, i + 1)
-            if os.path.exists(label_path):
-                label_files.append(label_path)
-
-        return label_files if label_files else "No labels generated."
-
-    except Exception as e:
-        return f"Error processing file: {str(e)}"
+            create_usps_label(row.to_dict(), index=i+1)
 
-# Function to create a USPS-style shipping label
-def create_shipping_label(row, index):
-    try:
-        tracking_number = str(row["Tracking Number"])
-        barcode_path = os.path.join(BARCODES_DIR, f"barcode_{index}.png")
-
-        # Generate barcode
-        ean = barcode.get_barcode_class('code128')
-        ean = ean(tracking_number, writer=ImageWriter())
-        ean.save(barcode_path)
-
-        # Create PDF label
-        label_path = os.path.join(LABELS_DIR, f"label_{index}.pdf")
-        c = canvas.Canvas(label_path, pagesize=letter)
-
-        # Header: USPS branding and tracking info
-        c.setFont("Helvetica-Bold", 20)
-        c.drawString(50, 750, "USPS FIRST-CLASS PKG")
-        c.setFont("Helvetica", 12)
-        c.drawString(50, 730, f"Order ID: {row['Order ID']}  |  Date: {row['Order Date']}")
-
-        # Sender information
-        c.setFont("Helvetica-Bold", 12)
-        c.drawString(50, 700, "From:")
-        c.setFont("Helvetica", 12)
-        c.drawString(50, 685, f"{row['Shipper Name']}")
-        c.drawString(50, 670, f"{row['Shipper Address']}")
-        c.drawString(50, 655, f"Phone: {row['Shipper Phone']}")
-
-        # Receiver information
-        c.setFont("Helvetica-Bold", 12)
-        c.drawString(50, 620, "To:")
-        c.setFont("Helvetica-Bold", 14)
-        c.drawString(50, 605, row["Receiver Name"])  # Bold receiver name
-        c.setFont("Helvetica", 12)
-        c.drawString(50, 590, row["Receiver Address"])
-        c.drawString(50, 575, f"Phone: {row['Receiver Phone']}")
-
-        # Shipping details
-        c.setFont("Helvetica-Bold", 12)
-        c.drawString(50, 540, "Shipping Method:")
-        c.setFont("Helvetica", 12)
-        c.drawString(50, 525, row["Shipping Method"])
-
-        c.setFont("Helvetica-Bold", 12)
-        c.drawString(250, 540, "Package Weight:")
-        c.setFont("Helvetica", 12)
-        c.drawString(250, 525, f"{row['Package Weight (kg)']} kg")
-
-        c.setFont("Helvetica-Bold", 12)
-        c.drawString(50, 500, "Estimated Delivery:")
-        c.setFont("Helvetica", 12)
-        c.drawString(50, 485, row["Estimated Delivery Date"])
-
-        # Add barcode
-        c.setFont("Helvetica-Bold", 14)
-        c.drawString(50, 450, "USPS TRACKING #")
-        if os.path.exists(barcode_path):
-            c.drawImage(barcode_path, 50, 400, width=300, height=70)
-        else:
-            c.drawString(50, 400, "Barcode generation failed.")
+        # Zip all PDFs
+        zip_filename = "usps_labels.zip"
+        with zipfile.ZipFile(zip_filename, "w") as zf:
+            for f in os.listdir(LABELS_DIR):
+                if f.endswith(".pdf"):
+                    zf.write(os.path.join(LABELS_DIR, f), f)
 
-        # Tracking number below barcode
-        c.setFont("Helvetica-Bold", 12)
-        c.drawString(50, 370, tracking_number)
+        return zip_filename
 
-        # Finalize PDF
-        c.showPage()
-        c.save()
-
-        return label_path
     except Exception as e:
-        return f"Error creating label: {str(e)}"
+        return f"Error processing file: {str(e)}"
 
-# Gradio interface
+############################################
+# 5) Gradio Interface
+############################################
 iface = gr.Interface(
     fn=process_file,
     inputs=gr.File(type="filepath"),
-    outputs=gr.File(type="filepath", label="Download Labels"),
-    title="Shipping Label Generator",
-    description="Upload a .csv, .xls, or .xlsx file to generate USPS-style shipping labels."
+    outputs=gr.File(type="filepath", label="Download Zipped Labels"),
+    title="USPS-Style Label Generator (4x6)",
+    description=(
+        "Upload a CSV/XLS/XLSX with columns:\n"
+        "Warehouse Line1, Warehouse Line2, Warehouse Line3, Order Number,\n"
+        "Recipient Name, Recipient Addr1, Recipient Addr2, Recipient Addr3,\n"
+        "Tracking Number\n\n"
+        "Each row creates a 4x6 USPS-style PDF label. Download the ZIP!"
+    ),
 )
 
-iface.launch()
+# Launch the app
+if __name__ == "__main__":
+    iface.launch()