Update app.py

app.py

@@ -12,7 +12,7 @@ BARCODES_DIR = "barcodes"
 os.makedirs(LABELS_DIR, exist_ok=True)
 os.makedirs(BARCODES_DIR, exist_ok=True)
 
-# Define required columns
+# 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)",
@@ -23,17 +23,17 @@ REQUIRED_COLUMNS = [
 def process_file(file_path):
     try:
         # Load data from file
-        if file_path.endswith(".csv"):
+        if file_path.endswith('.csv'):
             df = pd.read_csv(file_path)
-        elif file_path.endswith((".xls", ".xlsx")):
+        elif file_path.endswith(('.xls', '.xlsx')):
             df = pd.read_excel(file_path)
         else:
             return "Invalid file format. Please upload a .csv, .xls, or .xlsx file."
 
-        # Ensure required columns exist
-        missing_columns = [col for col in REQUIRED_COLUMNS if col not in df.columns]
-        if missing_columns:
-            return f"Missing columns: {', '.join(missing_columns)}"
+        # Ensure all required columns exist
+        missing_cols = [col for col in REQUIRED_COLUMNS if col not in df.columns]
+        if missing_cols:
+            return f"Missing columns: {', '.join(missing_cols)}"
 
         # Process each row and generate labels
         label_files = []
@@ -45,4 +45,58 @@ def process_file(file_path):
         return label_files if label_files else "No labels generated."
 
     except Exception as e:
-        return 
+        return f"Error processing file: {str(e)}"
+
+# Function to create a 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)
+
+        # Add shipping details
+        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']}")
+
+        # Add barcode
+        if os.path.exists(barcode_path):
+            c.drawImage(barcode_path, 50, 400, width=200, height=50)
+        else:
+            c.drawString(50, 400, "Barcode generation failed.")
+
+        c.showPage()
+        c.save()
+
+        return label_path
+    except Exception as e:
+        return f"Error creating label: {str(e)}"
+
+# Gradio interface
+iface = gr.Interface(
+    fn=process_file,
+    inputs=gr.File(type="filepath"),  # FIXED: Use 'filepath' instead of 'file'
+    outputs=gr.File(type="filepath", label="Download Labels"),
+    title="Shipping Label Generator",
+    description="Upload a .csv, .xls, or .xlsx file to generate shipping labels."
+)
+
+iface.launch()