Fix endpoint to use user-provided budget_amount even when has_previous_data is true

app/main.py

@@ -208,15 +208,25 @@ async def check_and_get_recommendations(request: RecommendationRequest, month: O
     has_data = recommender.check_user_has_category_data(request.user_id, request.category_id)
     
     if has_data:
-        # User has previous data - use it for recommendations (ignore budget_amount if provided)
-        print(f"✅ User {request.user_id} has previous data for category {request.category_id} - using historical data")
-        recommendations = recommender.get_recommendation_for_category(
-            request.user_id, 
-            request.category_id, 
-            month, 
-            year,
-            budget_amount=None  # Don't use budget_amount if user has historical data
-        )
+        # User has previous data - but if they provided budget_amount, use it (they want to set a new budget)
+        if request.budget_amount and request.budget_amount > 0:
+            print(f"✅ User {request.user_id} has previous data for category {request.category_id}, but provided budget_amount {request.budget_amount} - using provided amount")
+            recommendations = recommender.get_recommendation_for_category(
+                request.user_id, 
+                request.category_id, 
+                month, 
+                year,
+                budget_amount=request.budget_amount  # Use user's provided budget_amount
+            )
+        else:
+            print(f"✅ User {request.user_id} has previous data for category {request.category_id} - using historical data")
+            recommendations = recommender.get_recommendation_for_category(
+                request.user_id, 
+                request.category_id, 
+                month, 
+                year,
+                budget_amount=None  # Use historical data if no budget_amount provided
+            )
         
         if recommendations:
             return RecommendationResponse(