Update app.py

app.py

@@ -1,102 +1,86 @@
-import gradio as gr
-import pandas as pd
 import os
-import zipfile
-from reportlab.lib.pagesizes import inch
+import pandas as pd
+import gradio as gr
+from reportlab.lib.pagesizes import landscape, letter
 from reportlab.pdfgen import canvas
-from reportlab.lib.utils import ImageReader
 import barcode
 from barcode.writer import ImageWriter
 
-# 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))
+# Ensure output directories exist
+LABELS_DIR = "labels"
+BARCODES_DIR = "barcodes"
+os.makedirs(LABELS_DIR, exist_ok=True)
+os.makedirs(BARCODES_DIR, exist_ok=True)
 
-    # Draw sender details
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(30, 500, "From:")
-    c.setFont("Helvetica", 10)
-    c.drawString(30, 485, str(data.get('Sender Name', '')))
-    c.drawString(30, 470, str(data.get('Sender Address', '')))
-    c.drawString(30, 455, str(data.get('Sender City, State ZIP', '')))
-
-    # Draw recipient details
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(30, 400, "To:")
-    c.setFont("Helvetica", 10)
-    c.drawString(30, 385, str(data.get('Recipient Name', '')))
-    c.drawString(30, 370, str(data.get('Recipient Address', '')))
-    c.drawString(30, 355, str(data.get('Recipient City, State ZIP', '')))
-
-    # Draw shipping service
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(30, 320, f"Service: {str(data.get('Shipping Service', ''))}")
+# Function to process the uploaded file
+def process_file(file):
+    try:
+        # Load data from file
+        if file.name.endswith('.csv'):
+            df = pd.read_csv(file.name)
+        elif file.name.endswith(('.xls', '.xlsx')):
+            df = pd.read_excel(file.name)
+        else:
+            return "Invalid file format. Please upload a .csv, .xls, or .xlsx file."
 
-    # Draw tracking number
-    c.setFont("Helvetica-Bold", 12)
-    c.drawString(30, 280, "Tracking #:")
-    c.setFont("Helvetica", 12)
-    c.drawString(30, 260, str(data.get('Tracking Number', '')))
+        # Ensure required columns exist
+        required_cols = ["Sender Name", "Receiver Name", "Address", "City", "State", "Zip", "Tracking Number"]
+        for col in required_cols:
+            if col not in df.columns:
+                return f"Missing column: {col}"
 
-    # Generate barcode
-    barcode_filename = f"labels/barcode_{index+1}.png"
-    tracking_number = str(data.get('Tracking Number', '000000000'))
-    ean = barcode.get_barcode_class('code128')
-    ean_obj = ean(tracking_number, writer=ImageWriter())
-    ean_obj.save(barcode_filename)
+        # Process each row and generate labels
+        label_files = []
+        for i, row in df.iterrows():
+            label_path = create_shipping_label(row, i + 1)
+            label_files.append(label_path)
 
-    # Add barcode to label
-    barcode_img = ImageReader(barcode_filename)
-    c.drawImage(barcode_img, 30, 120, width=300, height=100)
+        return label_files
 
-    # Save PDF
-    c.showPage()
-    c.save()
+    except Exception as e:
+        return f"Error processing file: {str(e)}"
 
-def process_file(file_obj):
-    """
-    Read CSV or Excel file and generate shipping labels.
-    """
+# Function to create a shipping label
+def create_shipping_label(row, index):
     try:
-        # Read file into Pandas (directly from file object)
-        if file_obj.name.endswith('.csv'):
-            df = pd.read_csv(file_obj)  # Directly read from the file object
-        elif file_obj.name.endswith(('.xls', '.xlsx')):
-            df = pd.read_excel(file_obj)  # Read Excel file directly
+        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=(4 * 72, 6 * 72))  # 4x6 inches
+
+        # Add sender & receiver details
+        c.setFont("Helvetica", 10)
+        c.drawString(20, 380, f"From: {row['Sender Name']}")
+        c.drawString(20, 360, f"To: {row['Receiver Name']}")
+        c.drawString(20, 340, f"Address: {row['Address']}, {row['City']}, {row['State']} {row['Zip']}")
+
+        # Add barcode
+        if os.path.exists(barcode_path):
+            c.drawImage(barcode_path, 20, 200, width=200, height=50)
         else:
-            return "Invalid file format. Please upload a .csv, .xls, or .xlsx file."
+            c.drawString(20, 200, "Barcode generation failed.")
 
-        # 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
+        c.showPage()
+        c.save()
 
+        return label_path
     except Exception as e:
-        return f"Error processing file: {str(e)}"
+        return f"Error creating label: {str(e)}"
 
-# Gradio Interface
-interface = gr.Interface(
+# Gradio interface
+iface = gr.Interface(
     fn=process_file,
-    inputs=gr.File(label="Upload CSV or Excel File"),
-    outputs=gr.File(label="Download Shipping Labels (ZIP)"),
-    title="4x6 Shipping Label Generator",
-    description="Upload a .csv or .xlsx file with shipping details, and this app will generate professional 4x6 shipping labels with barcodes.",
+    inputs=gr.File(type="file"),
+    outputs=gr.File(type="file", label="Download Labels"),
+    title="Shipping Label Generator",
+    description="Upload a .csv, .xls, or .xlsx file to generate shipping labels."
 )
 
-# Run the app
-if __name__ == "__main__":
-    interface.launch()
+iface.launch()