Refactor mobile money asset positioning and sync backend engines

ai_assistant/views.py

@@ -31,14 +31,14 @@ class ChatbotView(APIView):
         
         context_prompt = f"""You are Aureon's premium AI Financial Assistant. You give concise, professional, and helpful financial advice.
 User Profile:
-- Available Cash: ${cash_available}
-- Net Worth: ${net_worth}
+- Available Cash: ₹{cash_available}
+- Net Worth: ₹{net_worth}
 Active Budgets:
 {budget_context if budget_context else "No active budgets."}
 Recent Transactions:
 {tx_context if tx_context else "No recent transactions."}
 
-Based strictly on this data, answer the user's question. Do not make up data not present here.
+Based strictly on this data, answer the user's question in rupees (₹). Do not make up data not present here.
 User Question: "{user_message}"
 """
 
@@ -101,8 +101,8 @@ class AIInsightsView(APIView):
 
         context_prompt = f"""You are Aureon's premium AI Financial Assistant. Based on the user's financial profile, generate exactly three helpful financial insights.
 User Profile:
-- Available Cash: ${cash_available}
-- Net Worth: ${net_worth}
+- Available Cash: ₹{cash_available}
+- Net Worth: ₹{net_worth}
 Active Budgets:
 {budget_context if budget_context else "No active budgets."}
 Recent Transactions:
@@ -113,14 +113,14 @@ Provide the output strictly as a JSON array of three objects. Do not include mar
 - "type": one of "success", "warning", "suggestion", "pattern"
 - "title": a very short title
 - "message": a helpful one-sentence message
-- "savings": optional, an integer number of potential savings in dollars (e.g. 42)
+- "savings": optional, an integer number of potential savings in rupees (e.g. 42)
 - "detail": a short secondary detail string (e.g. "Keep it up!")
 - "options": optional array of 2 strings if type is "suggestion"
 
 Example:
 [
   {{"id": 1, "type": "success", "title": "Great news!", "message": "You spent 15% less this week on food.", "savings": 25, "detail": "Keep it up!"}},
-  {{"id": 2, "type": "warning", "title": "Watch out!", "message": "Your subscriptions total $45 this month.", "detail": "Consider cancelling unused ones."}},
+  {{"id": 2, "type": "warning", "title": "Watch out!", "message": "Your subscriptions total ₹45 this month.", "detail": "Consider cancelling unused ones."}},
   {{"id": 3, "type": "suggestion", "title": "Smart Suggestion", "message": "You have extra cash.", "options": ["Invest it", "Add to goal"]}}
 ]
 """

finance/category_engine.py

@@ -0,0 +1,194 @@
+import re
+
+
+CATEGORY_RULES = [
+    {
+        "key": "income",
+        "name": "Income",
+        "patterns": [
+            "salary",
+            "payroll",
+            "paycheck",
+            "credited",
+            "deposit",
+            "interest",
+            "refund",
+        ],
+        "positive_only": True,
+    },
+    {
+        "key": "dining",
+        "name": "Food & Dining",
+        "patterns": [
+            "zomato",
+            "swiggy",
+            "mcdonald",
+            "domino",
+            "pizza",
+            "restaurant",
+            "cafe",
+            "burger",
+            "kfc",
+            "subway",
+            "dining",
+        ],
+    },
+    {
+        "key": "coffee",
+        "name": "Coffee & Cafes",
+        "patterns": ["starbucks", "coffee", "ccd", "barista"],
+    },
+    {
+        "key": "groceries",
+        "name": "Groceries",
+        "patterns": [
+            "grocery",
+            "groceries",
+            "dmart",
+            "bigbasket",
+            "blinkit",
+            "zepto",
+            "walmart",
+            "target",
+            "whole foods",
+            "supermarket",
+        ],
+    },
+    {
+        "key": "transportation",
+        "name": "Transportation",
+        "patterns": [
+            "uber",
+            "ola",
+            "rapido",
+            "metro",
+            "fuel",
+            "petrol",
+            "gas",
+            "shell",
+            "parking",
+            "toll",
+        ],
+    },
+    {
+        "key": "entertainment",
+        "name": "Entertainment",
+        "patterns": [
+            "netflix",
+            "spotify",
+            "prime",
+            "hotstar",
+            "bookmyshow",
+            "cinema",
+            "movie",
+            "youtube",
+        ],
+    },
+    {
+        "key": "shopping",
+        "name": "Shopping",
+        "patterns": [
+            "amazon",
+            "flipkart",
+            "myntra",
+            "ajio",
+            "nykaa",
+            "shopping",
+            "store",
+            "mall",
+        ],
+    },
+    {
+        "key": "healthcare",
+        "name": "Healthcare",
+        "patterns": [
+            "pharmacy",
+            "medical",
+            "hospital",
+            "clinic",
+            "doctor",
+            "apollo",
+            "1mg",
+            "pharmeasy",
+        ],
+    },
+    {
+        "key": "education",
+        "name": "Education",
+        "patterns": [
+            "school",
+            "college",
+            "tuition",
+            "course",
+            "udemy",
+            "coursera",
+            "book",
+            "exam",
+        ],
+    },
+    {
+        "key": "housing",
+        "name": "Housing",
+        "patterns": [
+            "rent",
+            "maintenance",
+            "electricity",
+            "water bill",
+            "gas bill",
+            "mortgage",
+            "society",
+        ],
+    },
+]
+
+
+FALLBACK_CATEGORY = {"key": "miscellaneous", "name": "Miscellaneous"}
+
+
+def normalize_text(*values):
+    text = " ".join(str(value or "") for value in values).lower()
+    return re.sub(r"\s+", " ", text).strip()
+
+
+def categorize_transaction(merchant, note="", amount=None):
+    text = normalize_text(merchant, note)
+    numeric_amount = None
+
+    try:
+        numeric_amount = float(amount) if amount is not None else None
+    except (TypeError, ValueError):
+        numeric_amount = None
+
+    for rule in CATEGORY_RULES:
+        if rule.get("positive_only") and numeric_amount is not None and numeric_amount <= 0:
+            continue
+
+        if any(pattern in text for pattern in rule["patterns"]):
+            return {
+                "category": rule["name"],
+                "category_key": rule["key"],
+                "confidence": "high",
+                "matched_rule": rule["key"],
+            }
+
+    if numeric_amount is not None and numeric_amount > 0:
+        return {
+            "category": "Income",
+            "category_key": "income",
+            "confidence": "medium",
+            "matched_rule": "positive_amount",
+        }
+
+    return {
+        "category": FALLBACK_CATEGORY["name"],
+        "category_key": FALLBACK_CATEGORY["key"],
+        "confidence": "low",
+        "matched_rule": None,
+    }
+
+
+def should_auto_categorize(category=None, category_key=None):
+    generic_values = {"", None, "other", "misc", "miscellaneous", "uncategorized"}
+    normalized_category = category.lower() if isinstance(category, str) else category
+    normalized_key = category_key.lower() if isinstance(category_key, str) else category_key
+    return normalized_key in generic_values or normalized_category in generic_values

finance/serializers.py

@@ -6,6 +6,10 @@ class TransactionSerializer(serializers.ModelSerializer):
         model = Transaction
         fields = '__all__'
         read_only_fields = ('user',)
+        extra_kwargs = {
+            'category': {'required': False, 'allow_blank': True},
+            'category_key': {'required': False, 'allow_blank': True},
+        }
 
 class BudgetSerializer(serializers.ModelSerializer):
     class Meta:

finance/views.py

@@ -4,6 +4,7 @@ from rest_framework.decorators import action
 from django.db.models import Sum
 from .models import Transaction, Budget, Goal, Subscription
 from .serializers import TransactionSerializer, BudgetSerializer, GoalSerializer, SubscriptionSerializer
+from .category_engine import categorize_transaction, should_auto_categorize
 
 class TransactionViewSet(viewsets.ModelViewSet):
     serializer_class = TransactionSerializer
@@ -13,6 +14,22 @@ class TransactionViewSet(viewsets.ModelViewSet):
         return Transaction.objects.filter(user=self.request.user)
 
     def perform_create(self, serializer):
+        category = serializer.validated_data.get('category')
+        category_key = serializer.validated_data.get('category_key')
+
+        if should_auto_categorize(category, category_key):
+            prediction = categorize_transaction(
+                serializer.validated_data.get('merchant'),
+                serializer.validated_data.get('note'),
+                serializer.validated_data.get('amount'),
+            )
+            serializer.save(
+                user=self.request.user,
+                category=prediction['category'],
+                category_key=prediction['category_key'],
+            )
+            return
+
         serializer.save(user=self.request.user)
 
 class BudgetViewSet(viewsets.ModelViewSet):
@@ -59,4 +76,3 @@ class SubscriptionViewSet(viewsets.ModelViewSet):
 
     def perform_create(self, serializer):
         serializer.save(user=self.request.user)
-