Update app.py

app.py

@@ -12,22 +12,28 @@ BARCODES_DIR = "barcodes"
 os.makedirs(LABELS_DIR, exist_ok=True)
 os.makedirs(BARCODES_DIR, exist_ok=True)
 
+# Define required columns
+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:
         # 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
-        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}"
+        missing_columns = [col for col in REQUIRED_COLUMNS if col not in df.columns]
+        if missing_columns:
+            return f"Missing columns: {', '.join(missing_columns)}"
 
         # Process each row and generate labels
         label_files = []
@@ -39,49 +45,4 @@ def process_file(file_path):
         return label_files if label_files else "No labels generated."
 
     except Exception as e:
-        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 sender & receiver details
-        c.setFont("Helvetica", 12)
-        c.drawString(50, 750, f"From: {row['Sender Name']}")
-        c.drawString(50, 730, f"To: {row['Receiver Name']}")
-        c.drawString(50, 710, f"Address: {row['Address']}, {row['City']}, {row['State']} {row['Zip']}")
-
-        # Add barcode
-        if os.path.exists(barcode_path):
-            c.drawImage(barcode_path, 50, 600, width=200, height=50)
-        else:
-            c.drawString(50, 600, "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()
+        return f"