Update app.py

app.py

@@ -58,11 +58,17 @@ def generate_label(data, index):
     c.showPage()
     c.save()
 
-def process_csv(file):
+def process_file(file):
     """
-    Read CSV and generate labels.
+    Read .csv or .xlsx file and generate labels.
     """
-    df = pd.read_csv(file.name)
+    # 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():
@@ -79,11 +85,11 @@ def process_csv(file):
 
 # Gradio Interface
 interface = gr.Interface(
-    fn=process_csv,
-    inputs=gr.File(label="Upload CSV File"),
+    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 file with shipping details, and this app will generate professional 4x6 shipping labels with barcodes.",
+    description="Upload a .csv or .xlsx file with shipping details, and this app will generate professional 4x6 shipping labels with barcodes.",
 )
 
 # Run the app