Update app.py

app.py

@@ -198,12 +198,26 @@ def main_app():
         net_savings = 0
         max_savings = total_income - total_expense - net_savings
         
+        # Initialize session state for buttons if not already done
+        if "show_set_goal_form" not in st.session_state:
+            st.session_state["show_set_goal_form"] = False
+        if "show_allocate_savings_form" not in st.session_state:
+            st.session_state["show_allocate_savings_form"] = False
+        
         # Buttons to toggle between setting goal and allocating savings
         set_goal_button = st.button("Set Saving Goal")
         allocate_savings_button = st.button("Allocate Saving to Goal")
         
-        # Set Saving Goal Form
+        # Toggle the display state for each form
         if set_goal_button:
+            st.session_state["show_set_goal_form"] = True
+            st.session_state["show_allocate_savings_form"] = False  # Hide other form
+        if allocate_savings_button:
+            st.session_state["show_allocate_savings_form"] = True
+            st.session_state["show_set_goal_form"] = False  # Hide other form
+        
+        # Set Saving Goal Form
+        if st.session_state["show_set_goal_form"]:
             st.subheader("Set Saving Goal")
             goal_name = st.text_input("Goal Name")
             target_amount = st.number_input("Target Amount", min_value=0, step=100)
@@ -225,7 +239,7 @@ def main_app():
                     st.error("Please provide valid goal name and target amount.")
         
         # Allocate Savings Form
-        if allocate_savings_button:
+        if st.session_state["show_allocate_savings_form"]:
             st.subheader("Allocate Savings to Goals")
             if not st.session_state["saving_goals"].empty:
                 # Add a dropdown to select a goal
@@ -279,6 +293,7 @@ def main_app():
 
 
 
+
         else:
             st.write("No saving goals set yet.")