Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,36 +1,19 @@
 import streamlit as st
-import json
-import os
 from datetime import datetime
 
 # =============================
-# Configuration
+# Initialize session state
 # =============================
-DATA_FILE = "data.json"  # Stored in container filesystem
+if "tasks" not in st.session_state:
+    st.session_state.tasks = []
 
-st.set_page_config(
-    page_title="To-Do & Habit Tracker",
-    layout="centered"
-)
-
-# =============================
-# Load / Save Functions
-# =============================
-def load_data():
-    if os.path.exists(DATA_FILE):
-        with open(DATA_FILE, "r") as f:
-            return json.load(f)
-    return {"tasks": [], "habits": []}
-
-def save_data(data):
-    with open(DATA_FILE, "w") as f:
-        json.dump(data, f)
-
-data = load_data()
+if "habits" not in st.session_state:
+    st.session_state.habits = []
 
 # =============================
 # App Title and Navigation
 # =============================
+st.set_page_config(page_title="To-Do & Habit Tracker", layout="centered")
 st.title("📝 To-Do List & Habit Tracker")
 
 page = st.sidebar.radio("Navigation", ["To-Do List", "Habit Tracker"])
@@ -41,32 +24,28 @@ page = st.sidebar.radio("Navigation", ["To-Do List", "Habit Tracker"])
 if page == "To-Do List":
     st.header("📌 Your Tasks")
 
-    new_task = st.text_input("Add a new task")
+    new_task = st.text_input("Add a new task", key="new_task")
 
     if st.button("Add Task"):
         if new_task.strip():
-            data["tasks"].append({"task": new_task, "completed": False})
-            save_data(data)
-            st.experimental_rerun()  # This works in latest Streamlit versions
+            st.session_state.tasks.append({"task": new_task, "completed": False})
+            st.session_state.new_task = ""
 
     st.subheader("Pending Tasks")
-    for i, task in enumerate(data["tasks"]):
+    for i, task in enumerate(st.session_state.tasks):
         if not task["completed"]:
             col1, col2 = st.columns([0.8, 0.2])
             with col1:
                 checked = st.checkbox(task["task"], key=f"task_{i}")
                 if checked:
-                    data["tasks"][i]["completed"] = True
-                    save_data(data)
-                    st.experimental_rerun()
+                    st.session_state.tasks[i]["completed"] = True
             with col2:
                 if st.button("Delete", key=f"delete_{i}"):
-                    data["tasks"].pop(i)
-                    save_data(data)
-                    st.experimental_rerun()
+                    st.session_state.tasks.pop(i)
+                    break  # break to avoid index error
 
     st.subheader("Completed Tasks")
-    for task in data["tasks"]:
+    for task in st.session_state.tasks:
         if task["completed"]:
             st.write("✅", task["task"])
 
@@ -76,34 +55,30 @@ if page == "To-Do List":
 if page == "Habit Tracker":
     st.header("🔥 Daily Habit Tracker")
 
-    new_habit = st.text_input("Add a new habit")
+    new_habit = st.text_input("Add a new habit", key="new_habit")
 
     if st.button("Add Habit"):
         if new_habit.strip():
-            data["habits"].append({"habit": new_habit, "dates": []})
-            save_data(data)
-            st.experimental_rerun()
+            st.session_state.habits.append({"habit": new_habit, "dates": []})
+            st.session_state.new_habit = ""
 
     today = datetime.now().strftime("%Y-%m-%d")
     completed_today = 0
 
-    for i, habit in enumerate(data["habits"]):
+    for i, habit in enumerate(st.session_state.habits):
         checked = today in habit["dates"]
 
-        if st.checkbox(habit["habit"], value=checked, key=f"habit_{i}"):
-            if today not in habit["dates"]:
-                data["habits"][i]["dates"].append(today)
-                save_data(data)
-        else:
-            if today in habit["dates"]:
-                data["habits"][i]["dates"].remove(today)
-                save_data(data)
+        new_checked = st.checkbox(habit["habit"], value=checked, key=f"habit_{i}")
+        if new_checked and today not in habit["dates"]:
+            st.session_state.habits[i]["dates"].append(today)
+        elif not new_checked and today in habit["dates"]:
+            st.session_state.habits[i]["dates"].remove(today)
 
-        if today in habit["dates"]:
+        if today in st.session_state.habits[i]["dates"]:
             completed_today += 1
 
-    if data["habits"]:
-        progress = completed_today / len(data["habits"])
+    if st.session_state.habits:
+        progress = completed_today / len(st.session_state.habits)
         st.subheader("📊 Today's Progress")
         st.progress(progress)
         st.write(f"{int(progress * 100)}% habits completed today")