Add check for budget _id when category_id is provided - handle case where user passes budget ID instead of category ID

app/smart_recommendation.py

@@ -127,6 +127,57 @@ class SmartBudgetRecommender:
         """
         print(f"🔍 check_user_has_category_data: Checking for user_id={user_id}, category_id={category_id}")
         
+        # FIRST: Check if category_id is actually a budget _id
+        # If so, find the budget and check if it belongs to the user and has categories
+        try:
+            try:
+                budget_id_objid = ObjectId(category_id)
+                # Try to find a budget with this _id
+                budget_by_id = self.db.budgets.find_one({"_id": budget_id_objid})
+                if budget_by_id:
+                    budget_created_by = budget_by_id.get("createdBy")
+                    # Check if this budget belongs to the user
+                    budget_user_match = (
+                        str(budget_created_by) == str(user_id) or 
+                        budget_created_by == ObjectId(user_id) if isinstance(budget_created_by, ObjectId) else False
+                    )
+                    
+                    if budget_user_match:
+                        print(f"✅ category_id {category_id} is a budget _id for user {user_id}")
+                        # Extract all category IDs from this budget's headCategories
+                        head_categories = budget_by_id.get("headCategories", [])
+                        category_ids_in_budget = []
+                        for hc in head_categories:
+                            if isinstance(hc, dict):
+                                # Check headCategory itself
+                                hc_id = hc.get("headCategory")
+                                if hc_id:
+                                    category_ids_in_budget.append(str(hc_id))
+                                # Check nested categories
+                                nested_cats = hc.get("categories", [])
+                                for nc in nested_cats:
+                                    if isinstance(nc, dict):
+                                        nc_id = nc.get("category")
+                                        if nc_id:
+                                            category_ids_in_budget.append(str(nc_id))
+                        
+                        print(f"🔍 Found {len(category_ids_in_budget)} category IDs in budget {category_id}")
+                        # If budget has categories, consider it as having previous data
+                        if category_ids_in_budget:
+                            print(f"✅ Budget {category_id} contains categories: {category_ids_in_budget[:5]}...")
+                            print(f"   Budget name: {budget_by_id.get('name', 'Unknown')}, maxAmount: {budget_by_id.get('maxAmount', 0)}")
+                            return True
+                        else:
+                            print(f"⚠️ Budget {category_id} exists but has no categories")
+                    else:
+                        print(f"⚠️ Budget {category_id} exists but belongs to different user")
+            except (ValueError, TypeError):
+                pass  # category_id is not a valid ObjectId, continue with normal check
+        except Exception as e:
+            print(f"⚠️ Error checking if category_id is a budget _id: {e}")
+            import traceback
+            traceback.print_exc()
+        
         # Build comprehensive user query
         user_conditions = []
         try: