Handle category field as string name (not ObjectId) in budgets

app/smart_recommendation.py

@@ -483,13 +483,33 @@ class SmartBudgetRecommender:
             # Get category name from headCategories or categories collection using category ID
             if category_id:
                 print(f"🔍 Looking up category ID: {category_id} (type: {type(category_id).__name__})")
-                category_name = self._get_category_name(category_id)
-                if category_name == str(category_id):
-                    # Category name lookup failed, still showing ID
-                    print(f"⚠️ Category ID not resolved: {category_id} (not found in headCategories or categories collections)")
-                    print(f"   This means the category ID doesn't exist in the database. Please check if the category exists.")
+                
+                # Check if category_id is already a string name (not a valid ObjectId)
+                if isinstance(category_id, str):
+                    # Check if it's a valid ObjectId format (24 hex characters)
+                    is_valid_objectid = len(category_id) == 24 and all(c in '0123456789abcdefABCDEF' for c in category_id)
+                    if not is_valid_objectid:
+                        # It's already a category name, use it directly
+                        category_name = category_id
+                        print(f"✅ Using category name directly (not an ObjectId): '{category_name}'")
+                    else:
+                        # It's a valid ObjectId string, try to look it up
+                        category_name = self._get_category_name(category_id)
+                        if category_name == str(category_id):
+                            # Category name lookup failed, still showing ID
+                            print(f"⚠️ Category ID not resolved: {category_id} (not found in headCategories or categories collections)")
+                            print(f"   This means the category ID doesn't exist in the database. Please check if the category exists.")
+                        else:
+                            print(f"✅ Found category ID: {category_id} -> Name: '{category_name}'")
                 else:
-                    print(f"✅ Found category ID: {category_id} -> Name: '{category_name}'")
+                    # It's an ObjectId object, look it up
+                    category_name = self._get_category_name(category_id)
+                    if category_name == str(category_id):
+                        # Category name lookup failed, still showing ID
+                        print(f"⚠️ Category ID not resolved: {category_id} (not found in headCategories or categories collections)")
+                        print(f"   This means the category ID doesn't exist in the database. Please check if the category exists.")
+                    else:
+                        print(f"✅ Found category ID: {category_id} -> Name: '{category_name}'")
             else:
                 # Fallback to budget name if no category ID found
                 category_name = b.get("name", "Uncategorized")