Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import re
 import unicodedata
 import io
+import tempfile
 
 # ---------- Helper Functions ----------
 
@@ -60,7 +61,7 @@ def flow_address_lines(lines, maxlen=35, maxlines=3):
     return [ln[:maxlen] for ln in out]
 
 def convert_dry_ice_kg(x):
-    """Convert lbs -> kg and round."""
+    """Convert lbs → kg and round."""
     if pd.isna(x) or str(x).strip() == "":
         return ""
     try:
@@ -79,11 +80,11 @@ def clean_csv(file):
 
     df.columns = df.columns.str.strip()
 
-    # --- Cleaning operations ---
+    # ZIP correction
     if "ZipCode" in df.columns:
         df["ZipCode"] = df["ZipCode"].map(format_zip)
 
-    # Address split logic
+    # Split address lines
     addr1, addr2, addr3 = [], [], []
     for _, row in df.iterrows():
         a1, a2, a3 = flow_address_lines([
@@ -96,33 +97,33 @@ def clean_csv(file):
     df["Address2"] = addr2
     df["Address3"] = addr3
 
-    # Clean text fields
-    text_cols = ["Company Name", "Contact Name", "City", "State", "Phone Number", "Email"]
-    for col in text_cols:
+    # Clean key fields
+    for col in ["Company Name", "Contact Name", "City", "State", "Phone Number", "Email"]:
         if col in df.columns:
             df[col] = df[col].map(clean_text)
 
-    # Dry Ice conversion
+    # Convert Dry Ice Weight
     if "Dry Ice Weight" in df.columns:
         df["Dry Ice Weight (kg)"] = df["Dry Ice Weight"].map(convert_dry_ice_kg)
 
-    # Save to BytesIO for Gradio download
-    buffer = io.BytesIO()
-    df.to_csv(buffer, index=False, encoding="utf-8-sig")
-    buffer.seek(0)
-    return buffer, "cleaned_output.csv"
+    # Save to a temp file for download
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
+    df.to_csv(temp_file.name, index=False, encoding="utf-8-sig")
+    temp_file.close()
+
+    return temp_file.name  # ✅ Return single file path
 
 # ---------- Gradio UI ----------
 
 title = "UPS Shipment CSV Cleaner"
 description = """
-Upload your **raw shipment CSV file** below.  
-This tool will:
-- Remove strange characters (e.g. ÿ)
-- Pad ZIP codes to 5 digits
-- Split long addresses into ≤ 35 characters
-- Convert Dry Ice Weight from lbs → kg  
-Then download the cleaned CSV ready for UPS Batch import.
+Upload your **raw shipment CSV file**.  
+The tool will:
+- Remove bad characters (e.g. ÿ)
+- Pad ZIP codes to 5 digits  
+- Split long addresses into 35-character lines  
+- Convert Dry Ice Weight (lbs → kg)
+Then download your cleaned CSV ready for UPS Batch import.
 """
 
 demo = gr.Interface(