Update src/streamlit_app.py

src/streamlit_app.py

@@ -24,14 +24,13 @@ page = st.sidebar.radio("Navigation", ["To-Do List", "Habit Tracker"])
 if page == "To-Do List":
     st.header("📌 Your Tasks")
 
-    # Use a temporary variable to hold text input
-    new_task_input = st.text_input("Add a new task", key="new_task_input")
+    # Use temporary variable for input
+    new_task = st.text_input("Add a new task")
 
     if st.button("Add Task"):
-        if new_task_input.strip():
-            st.session_state.tasks.append({"task": new_task_input.strip(), "completed": False})
-            # Clear input by resetting the temporary variable
-            st.experimental_rerun()
+        if new_task.strip():
+            st.session_state.tasks.append({"task": new_task.strip(), "completed": False})
+            new_task = ""  # just clear local variable (widget keeps its value until next rerun)
 
     st.subheader("Pending Tasks")
     for i, task in enumerate(st.session_state.tasks):
@@ -41,11 +40,10 @@ if page == "To-Do List":
                 checked = st.checkbox(task["task"], key=f"task_{i}")
                 if checked:
                     st.session_state.tasks[i]["completed"] = True
-                    st.experimental_rerun()
             with col2:
                 if st.button("Delete", key=f"delete_{i}"):
                     st.session_state.tasks.pop(i)
-                    st.experimental_rerun()
+                    break  # break to avoid index issues
 
     st.subheader("Completed Tasks")
     for task in st.session_state.tasks:
@@ -58,12 +56,12 @@ if page == "To-Do List":
 if page == "Habit Tracker":
     st.header("🔥 Daily Habit Tracker")
 
-    new_habit_input = st.text_input("Add a new habit", key="new_habit_input")
+    new_habit = st.text_input("Add a new habit")
 
     if st.button("Add Habit"):
-        if new_habit_input.strip():
-            st.session_state.habits.append({"habit": new_habit_input.strip(), "dates": []})
-            st.experimental_rerun()
+        if new_habit.strip():
+            st.session_state.habits.append({"habit": new_habit.strip(), "dates": []})
+            new_habit = ""  # clear local variable
 
     today = datetime.now().strftime("%Y-%m-%d")
     completed_today = 0