Update app.py

app.py

@@ -1,8 +1,8 @@
 import os
 import pandas as pd
 import gradio as gr
-from reportlab.lib.pagesizes import landscape, letter
 from reportlab.pdfgen import canvas
+from reportlab.lib.pagesizes import letter
 import barcode
 from barcode.writer import ImageWriter
 
@@ -13,13 +13,13 @@ os.makedirs(LABELS_DIR, exist_ok=True)
 os.makedirs(BARCODES_DIR, exist_ok=True)
 
 # Function to process the uploaded file
-def process_file(file):
+def process_file(file_path):
     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)
+        if file_path.endswith('.csv'):
+            df = pd.read_csv(file_path)
+        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."
 
@@ -33,9 +33,10 @@ def process_file(file):
         label_files = []
         for i, row in df.iterrows():
             label_path = create_shipping_label(row, i + 1)
-            label_files.append(label_path)
+            if os.path.exists(label_path):
+                label_files.append(label_path)
 
-        return label_files
+        return label_files if label_files else "No labels generated."
 
     except Exception as e:
         return f"Error processing file: {str(e)}"
@@ -53,19 +54,19 @@ def create_shipping_label(row, index):
 
         # 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
+        c = canvas.Canvas(label_path, pagesize=letter)
 
         # 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']}")
+        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, 20, 200, width=200, height=50)
+            c.drawImage(barcode_path, 50, 600, width=200, height=50)
         else:
-            c.drawString(20, 200, "Barcode generation failed.")
+            c.drawString(50, 600, "Barcode generation failed.")
 
         c.showPage()
         c.save()
@@ -77,8 +78,8 @@ def create_shipping_label(row, index):
 # Gradio interface
 iface = gr.Interface(
     fn=process_file,
-    inputs=gr.File(type="file"),
-    outputs=gr.File(type="file", label="Download Labels"),
+    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."
 )