Update meeprofit.py

meeprofit.py

@@ -63,6 +63,28 @@ def step1_generate_sheet2(files):
     # -------------------------
     # Filter Required Statuses
     # -------------------------
+    # Auto detect status column
+    status_column = None
+
+    for col in merged_data.columns:
+      if "status" in str(col).lower():
+        status_column = col
+        break
+
+    if status_column is None:
+        return None, f"❌ Status column not found.\nColumns: {list(merged_data.columns)}"
+
+    # Auto detect SKU column
+    sku_column = None
+
+    for col in merged_data.columns:
+        if "sku" in str(col).lower():
+           sku_column = col
+           break
+
+    if sku_column is None:
+        return None, f"❌ SKU column not found.\nColumns: {list(merged_data.columns)}"
+
     status_keep = [
         "Delivered",
         "Return",
@@ -71,12 +93,10 @@ def step1_generate_sheet2(files):
     ]
 
     merged_data = merged_data[
-        merged_data["Live Order Status"].isin(status_keep)
+       merged_data[status_column].astype(str).isin(status_keep)
     ]
 
-    merged_data = merged_data.sort_values(
-        by="Supplier SKU"
-    )
+    merged_data = merged_data.sort_values(by=sku_column)
 
     # -------------------------
     # Create Sheet2
@@ -84,9 +104,10 @@ def step1_generate_sheet2(files):
     sheet2 = pd.DataFrame()
 
     sheet2["Supplier SKU"] = (
-        merged_data["Supplier SKU"]
-        .dropna()
-        .unique()
+         merged_data[sku_column]
+         .dropna()
+         .unique()
+)
     )
 
     sheet2["Cost"] = ""