Fix category budget update

The category budget was not being updated correctly after setting the budget. This commit fixes the issue.

src/components/BudgetTabContent.tsx

@@ -1,25 +1,31 @@
+
 import React, { useState } from 'react';
 import { Button } from '@/components/ui/button';
 import { Check, ChevronDown, ChevronUp, Calculator } from 'lucide-react';
 import { Collapsible, CollapsibleContent, CollapsibleTrigger } from '@/components/ui/collapsible';
 import BudgetProgress from './BudgetProgress';
 import CategoryBudgetInputs from './CategoryBudgetInputs';
+import { toast } from '@/components/ui/use-toast';
+
 interface BudgetData {
   targetAmount: number;
   spentAmount: number;
   remainingAmount: number;
 }
+
 interface CategoryBudget {
   식비: number;
   생활비: number;
   교통비: number;
 }
+
 interface BudgetTabContentProps {
   data: BudgetData;
   formatCurrency: (amount: number) => string;
   calculatePercentage: (spent: number, target: number) => number;
-  onSaveBudget: (amount: number) => void;
+  onSaveBudget: (amount: number, categoryBudgets?: CategoryBudget) => void;
 }
+
 const BudgetTabContent: React.FC<BudgetTabContentProps> = ({
   data,
   formatCurrency,
@@ -29,16 +35,25 @@ const BudgetTabContent: React.FC<BudgetTabContentProps> = ({
   const percentage = calculatePercentage(data.spentAmount, data.targetAmount);
   const [isOpen, setIsOpen] = useState(false);
   const [budgetInput, setBudgetInput] = useState(data.targetAmount.toString());
-  const [categoryBudgets, setCategoryBudgets] = useState<CategoryBudget>({
-    식비: Math.round(data.targetAmount * 0.4),
-    생활비: Math.round(data.targetAmount * 0.4),
-    교통비: Math.round(data.targetAmount * 0.2)
-  });
+  
+  // 저장된 카테고리 예산을 불러옵니다
+  const savedCategoryBudgets = localStorage.getItem('categoryBudgets');
+  const initialCategoryBudgets = savedCategoryBudgets 
+    ? JSON.parse(savedCategoryBudgets) 
+    : {
+        식비: Math.round(data.targetAmount * 0.4),
+        생활비: Math.round(data.targetAmount * 0.4),
+        교통비: Math.round(data.targetAmount * 0.2)
+      };
+  
+  const [categoryBudgets, setCategoryBudgets] = useState<CategoryBudget>(initialCategoryBudgets);
+
   const handleInputChange = (value: string) => {
     // Remove all non-numeric characters
     const numericValue = value.replace(/[^0-9]/g, '');
     setBudgetInput(numericValue);
   };
+
   const handleCategoryInputChange = (value: string, category: keyof CategoryBudget) => {
     // Remove all non-numeric characters
     const numericValue = value.replace(/[^0-9]/g, '');
@@ -47,10 +62,19 @@ const BudgetTabContent: React.FC<BudgetTabContentProps> = ({
       [category]: parseInt(numericValue) || 0
     }));
   };
+
   const handleSave = () => {
     // Calculate total from all categories
     const totalAmount = Object.values(categoryBudgets).reduce((sum, value) => sum + value, 0);
-    onSaveBudget(totalAmount);
+    
+    // 카테고리 예산도 함께 전달합니다
+    onSaveBudget(totalAmount, categoryBudgets);
+    
+    toast({
+      title: "예산 설정 완료",
+      description: "카테고리별 예산이 성공적으로 저장되었습니다.",
+    });
+    
     setIsOpen(false);
   };
 
@@ -58,6 +82,7 @@ const BudgetTabContent: React.FC<BudgetTabContentProps> = ({
   const formatWithCommas = (amount: string) => {
     return amount.replace(/\B(?=(\d{3})+(?!\d))/g, ',');
   };
+
   return <div className="space-y-4">
       <BudgetProgress spentAmount={data.spentAmount} targetAmount={data.targetAmount} percentage={percentage} formatCurrency={formatCurrency} />
       
@@ -97,4 +122,5 @@ const BudgetTabContent: React.FC<BudgetTabContentProps> = ({
       </div>
     </div>;
 };
-export default BudgetTabContent;
+
+export default BudgetTabContent;