Update app.py

app.py

@@ -61,10 +61,16 @@ def process_file(file):
     if "Shipping Province" in df.columns:
         output_df["Province/State"] = df["Shipping Province"]
 
-    # 6. ZipCode (pad 4-digit US zips, leave others as-is)
+    # 6. ZipCode (pad 4-digit US zips with a leading zero + apostrophe for text)
     if "Shipping ZIP" in df.columns:
-        zip_raw = df["Shipping ZIP"].astype(str).str.zfill(5)  # zfill makes '1234' → '01234'
-        output_df["ZipCode"] = zip_raw
+        zip_raw = df["Shipping ZIP"].astype(str).str.strip()
+        mask_us = output_df["CountryCode"] == "US"
+        mask_4  = zip_raw.str.len() == 4
+        # pad to 5 digits
+        zip_padded = zip_raw.where(~(mask_us & mask_4), "0" + zip_raw)
+        # prefix apostrophe so Excel treats it as text and preserves leading zero
+        zip_final  = zip_padded.where(~(mask_us & mask_4), "'" + zip_padded)
+        output_df["ZipCode"] = zip_final
 
     # 7. CountryCode
     if "Shipping Country Code" in df.columns:
@@ -110,18 +116,8 @@ def process_file(file):
     today_hk = datetime.now(hk_tz).strftime("%y%m%d")
     output_file_name = f"yunexpress {today_hk}.xlsx"
 
-    # 14. Save to Excel with formatting for ZipCode
-    with pd.ExcelWriter(output_file_name, engine='xlsxwriter') as writer:
-        output_df.to_excel(writer, index=False, sheet_name='YunExpress')
-        workbook  = writer.book
-        worksheet = writer.sheets['YunExpress']
-
-        # Find the zero-based column index of "ZipCode"
-        zip_col_idx = output_df.columns.get_loc("ZipCode")
-        # Create a 5-digit text format
-        fmt_text = workbook.add_format({'num_format': '00000', 'align': 'left'})
-        # Apply to the entire column
-        worksheet.set_column(zip_col_idx, zip_col_idx, 8, fmt_text)
+    # Save to Excel
+    output_df.to_excel(output_file_name, index=False)
 
     return output_df, output_file_name
 
@@ -156,4 +152,4 @@ with gr.Blocks(title="Shipping - YunExpress") as demo:
         """
     )
 
-demo.launch()
+demo.launch()