Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
 import pandas as pd
+import os
+import zipfile
 from reportlab.lib.pagesizes import inch
 from reportlab.pdfgen import canvas
 from reportlab.lib.utils import ImageReader
 import barcode
 from barcode.writer import ImageWriter
-import os
-import zipfile
 
 # Ensure 'labels' directory exists
 os.makedirs("labels", exist_ok=True)
@@ -18,7 +18,7 @@ def generate_label(data, index):
     pdf_filename = f"labels/shipping_label_{index+1}.pdf"
     c = canvas.Canvas(pdf_filename, pagesize=(4 * inch, 6 * inch))
 
-    # Draw sender details (top-left)
+    # Draw sender details
     c.setFont("Helvetica-Bold", 12)
     c.drawString(30, 500, "From:")
     c.setFont("Helvetica", 10)
@@ -26,7 +26,7 @@ def generate_label(data, index):
     c.drawString(30, 470, data['Sender Address'])
     c.drawString(30, 455, data['Sender City, State ZIP'])
 
-    # Draw recipient details (middle)
+    # Draw recipient details
     c.setFont("Helvetica-Bold", 12)
     c.drawString(30, 400, "To:")
     c.setFont("Helvetica", 10)
@@ -58,30 +58,35 @@ def generate_label(data, index):
     c.showPage()
     c.save()
 
-def process_file(file):
+def process_file(file_obj):
     """
-    Read .csv or .xlsx file and generate labels.
+    Read CSV or Excel file and generate shipping labels.
     """
-    # Determine file format
-    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."
-
-    # 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
+    try:
+        # Read file into Pandas
+        file_path = file_obj.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."
+
+        # 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
+
+    except Exception as e:
+        return f"Error processing file: {str(e)}"
 
 # Gradio Interface
 interface = gr.Interface(
@@ -94,4 +99,4 @@ interface = gr.Interface(
 
 # Run the app
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)