Add multiple nested query patterns to check_user_has_category_data for better MongoDB compatibility

app/smart_recommendation.py

@@ -173,6 +173,7 @@ class SmartBudgetRecommender:
         
         # FIRST: Check nested structure (headCategories[].categories[].category) - most common case
         # Budgets have structure: headCategories[].categories[].category (ObjectId)
+        # User ID is stored in createdBy field
         print(f"🔍 Checking nested structure: headCategories[].categories[].category for category_id {category_id}")
         try:
             try:
@@ -180,9 +181,11 @@ class SmartBudgetRecommender:
             except (ValueError, TypeError):
                 category_objid = category_id
             
-            # Try to find budgets with nested category structures
-            # Structure: headCategories[].categories[].category
-            nested_query = {
+            # Try multiple nested query patterns to ensure we catch all cases
+            nested_queries = []
+            
+            # Pattern 1: Using $elemMatch for nested arrays (most accurate)
+            nested_queries.append({
                 "$and": [
                     {"$or": user_conditions},
                     {
@@ -196,14 +199,36 @@ class SmartBudgetRecommender:
                         ]
                     }
                 ]
-            }
-            nested_count = self.db.budgets.count_documents(nested_query)
-            if nested_count > 0:
-                print(f"✅ Found {nested_count} budget(s) with nested category structure for user {user_id} with category_id {category_id}")
-                print(f"   Nested query matched: headCategories[].categories[].category = {category_id}")
-                return True
-            else:
-                print(f"🔍 Nested query found 0 budgets for category_id {category_id}")
+            })
+            
+            # Pattern 2: Using dot notation (simpler, might work for some MongoDB versions)
+            nested_queries.append({
+                "$and": [
+                    {"$or": user_conditions},
+                    {
+                        "$or": [
+                            {"headCategories.categories.category": category_objid},
+                            {"headCategories.categories.category": category_id},
+                            {"headCategories.headCategory": category_objid},
+                            {"headCategories.headCategory": category_id},
+                        ]
+                    }
+                ]
+            })
+            
+            # Try each nested query pattern
+            for i, nested_query in enumerate(nested_queries, 1):
+                try:
+                    nested_count = self.db.budgets.count_documents(nested_query)
+                    if nested_count > 0:
+                        print(f"✅ Found {nested_count} budget(s) with nested category structure for user {user_id} with category_id {category_id}")
+                        print(f"   Nested query pattern {i} matched: headCategories[].categories[].category = {category_id}")
+                        return True
+                except Exception as e:
+                    print(f"⚠️ Error with nested query pattern {i}: {e}")
+                    continue
+            
+            print(f"🔍 Nested query found 0 budgets for category_id {category_id}")
         except Exception as e:
             print(f"⚠️ Error checking nested categories: {e}")
             import traceback