Update app.py

app.py

@@ -4,7 +4,6 @@ import gradio as gr
 import pandas as pd
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import inch
-from reportlab.lib.utils import ImageReader
 import barcode
 from barcode.writer import ImageWriter
 
@@ -54,7 +53,7 @@ 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)
+    # Extract values from row (using .get() with defaults)
     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")
@@ -65,7 +64,7 @@ def draw_label_text(c, row, width, height):
     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")
+    tracking_number = row.get("Tracking Number", "9400110200793961893691")
 
     # Warehouse info (top-left under the bar)
     c.setFont("Helvetica-Bold", 9)
@@ -75,6 +74,7 @@ def draw_label_text(c, row, width, height):
     c.drawString(15, height - 160, warehouse_line3)
 
     # Order number (top-right)
+    c.setFont("Helvetica", 9)
     c.drawString(width - 80, height - 130, f"Order: {order_number}")
 
     # Recipient info (bold name)
@@ -101,14 +101,14 @@ 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
+    # 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
+    # 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))
@@ -141,15 +141,15 @@ def process_file(file_obj):
         for f in os.listdir(BARCODES_DIR):
             os.remove(os.path.join(BARCODES_DIR, f))
 
-        # Read the file
+        # Read the file from the Gradio file object
         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."
+            return "Invalid file format. Please upload a .csv, .xls, or .xlsx file.", None
 
-        # Basic check for columns
+        # Basic check for required columns
         needed_cols = [
             "Warehouse Line1", "Warehouse Line2", "Warehouse Line3", "Order Number",
             "Recipient Name", "Recipient Addr1", "Recipient Addr2", "Recipient Addr3",
@@ -157,23 +157,23 @@ def process_file(file_obj):
         ]
         missing = [c for c in needed_cols if c not in df.columns]
         if missing:
-            return f"Missing columns: {', '.join(missing)}"
+            return f"Missing columns: {', '.join(missing)}", None
 
         # Generate a PDF label for each row
         for i, row in df.iterrows():
             create_usps_label(row.to_dict(), index=i+1)
 
-        # Zip all PDFs
+        # Zip all PDFs into one file
         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)
 
-        return zip_filename
+        return "Labels generated successfully.", zip_filename
 
     except Exception as e:
-        return f"Error processing file: {str(e)}"
+        return f"Error processing file: {str(e)}", None
 
 ############################################
 # 5) Gradio Interface
@@ -181,7 +181,7 @@ def process_file(file_obj):
 iface = gr.Interface(
     fn=process_file,
     inputs=gr.File(type="filepath"),
-    outputs=gr.File(type="filepath", label="Download Zipped Labels"),
+    outputs=[gr.Text(label="Status"), gr.File(label="Download Zipped Labels")],
     title="USPS-Style Label Generator (4x6)",
     description=(
         "Upload a CSV/XLS/XLSX with columns:\n"