Create categorizer.py

categorizer.py

@@ -0,0 +1,20 @@
+import pandas as pd
+
+CATEGORIES = {
+    "DINING": ["RESTAURANT", "CAFE", "FOOD TRUCK"],
+    "TRANSPORT": ["UBER", "LYFT", "TAXI", "PETROL"],
+    "SHOPPING": ["AMAZON", "WALMART", "TARGET"],
+    "UTILITIES": ["ELECTRIC", "WATER", "INTERNET"]
+}
+
+def categorize_expenses(file):
+    df = pd.read_csv(file.name)
+    df['Category'] = "OTHER"
+    
+    for idx, row in df.iterrows():
+        description = row['Description'].upper()
+        for cat, keywords in CATEGORIES.items():
+            if any(kw in description for kw in keywords):
+                df.at[idx, 'Category'] = cat
+                break
+    return df