Update app.py

app.py

@@ -3,6 +3,9 @@ import pandas as pd
 import gradio as gr
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
+from reportlab.lib import colors
+from reportlab.lib.utils import ImageReader
+from reportlab.lib.styles import getSampleStyleSheet
 import barcode
 from barcode.writer import ImageWriter
 
@@ -12,13 +15,6 @@ BARCODES_DIR = "barcodes"
 os.makedirs(LABELS_DIR, exist_ok=True)
 os.makedirs(BARCODES_DIR, exist_ok=True)
 
-# Required columns for the input file
-REQUIRED_COLUMNS = [
-    "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"
-]
-
 # Function to process the uploaded file
 def process_file(file_path):
     try:
@@ -30,8 +26,14 @@ def process_file(file_path):
         else:
             return "Invalid file format. Please upload a .csv, .xls, or .xlsx file."
 
-        # Ensure all required columns exist
-        missing_cols = [col for col in REQUIRED_COLUMNS if col not in df.columns]
+        # 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"
+        ]
+        
+        missing_cols = [col for col in required_cols if col not in df.columns]
         if missing_cols:
             return f"Missing columns: {', '.join(missing_cols)}"
 
@@ -47,7 +49,7 @@ def process_file(file_path):
     except Exception as e:
         return f"Error processing file: {str(e)}"
 
-# Function to create a shipping label
+# Function to create a USPS-style shipping label
 def create_shipping_label(row, index):
     try:
         tracking_number = str(row["Tracking Number"])
@@ -62,27 +64,58 @@ def create_shipping_label(row, index):
         label_path = os.path.join(LABELS_DIR, f"label_{index}.pdf")
         c = canvas.Canvas(label_path, pagesize=letter)
 
-        # Add shipping details
+        # 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, 750, f"Order ID: {row['Order ID']}")
-        c.drawString(50, 730, f"Date: {row['Order Date']}")
-        c.drawString(50, 710, f"From: {row['Shipper Name']}")
-        c.drawString(50, 690, f"Address: {row['Shipper Address']}")
-        c.drawString(50, 670, f"Phone: {row['Shipper Phone']}")
-        c.drawString(50, 650, f"To: {row['Receiver Name']}")
-        c.drawString(50, 630, f"Address: {row['Receiver Address']}")
-        c.drawString(50, 610, f"Phone: {row['Receiver Phone']}")
-        c.drawString(50, 590, f"Weight: {row['Package Weight (kg)']} kg")
-        c.drawString(50, 570, f"Method: {row['Shipping Method']}")
-        c.drawString(50, 550, f"Tracking #: {row['Tracking Number']}")
-        c.drawString(50, 530, f"ETA: {row['Estimated Delivery Date']}")
+        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=200, height=50)
+            c.drawImage(barcode_path, 50, 400, width=300, height=70)
         else:
             c.drawString(50, 400, "Barcode generation failed.")
 
+        # Tracking number below barcode
+        c.setFont("Helvetica-Bold", 12)
+        c.drawString(50, 370, tracking_number)
+
+        # Finalize PDF
         c.showPage()
         c.save()
 
@@ -93,10 +126,10 @@ def create_shipping_label(row, index):
 # Gradio interface
 iface = gr.Interface(
     fn=process_file,
-    inputs=gr.File(type="filepath"),  # FIXED: Use 'filepath' instead of '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 shipping labels."
+    description="Upload a .csv, .xls, or .xlsx file to generate USPS-style shipping labels."
 )
 
 iface.launch()