Update app.py

app.py

@@ -18,7 +18,7 @@ def init_file(path, cols):
     if not os.path.exists(path):
         pd.DataFrame(columns=cols).to_csv(path, index=False)
 
-init_file(HABIT_FILE, ["habit"])
+init_file(HABIT_FILE, ["username", "habit"])
 init_file(DATA_FILE, ["username", "date", "mood", "mood_note", "habit", "completed"])
 
 # ---------- USER LOGIN ----------
@@ -27,21 +27,31 @@ if not username:
     st.warning("Please enter your name to continue.")
     st.stop()
 
-# ---------- LOAD DATA ----------
-user_habits = pd.read_csv(HABIT_FILE)
-data = pd.read_csv(DATA_FILE)
+# ---------- LOAD USER HABITS ----------
+user_habits_df = pd.read_csv(HABIT_FILE)
+user_habits = user_habits_df[user_habits_df["username"] == username]["habit"].tolist()
 
-# ---------- SIDEBAR: ADD HABIT ----------
-st.sidebar.markdown("## ➕ Add Your Custom Habit")
-new_habit = st.sidebar.text_input("Enter new habit:")
+# ---------- FIRST-TIME HABIT SETUP ----------
+if not user_habits:
+    st.markdown("### 🛠️ Setup Your Habits")
+    st.info("Please enter at least 3 habits you want to track:")
 
-if st.sidebar.button("Add Habit"):
-    if new_habit and new_habit not in user_habits["habit"].values:
-        user_habits.loc[len(user_habits)] = [new_habit]
-        user_habits.to_csv(HABIT_FILE, index=False)
-        st.sidebar.success(f"Habit '{new_habit}' added!")
-    elif new_habit:
-        st.sidebar.warning("Habit already exists.")
+    habit_inputs = []
+    for i in range(1, 6):
+        habit = st.text_input(f"Habit {i} (optional):")
+        if habit:
+            habit_inputs.append(habit)
+
+    if st.button("✅ Save My Habits"):
+        if len(habit_inputs) < 3:
+            st.warning("Please enter at least 3 habits.")
+        else:
+            new_df = pd.DataFrame([{"username": username, "habit": h} for h in habit_inputs])
+            new_df.to_csv(HABIT_FILE, mode="a", index=False, header=False)
+            st.success("✅ Your habits are saved! Please reload the app.")
+            st.stop()
+    else:
+        st.stop()
 
 # ---------- MOOD & NOTE ----------
 st.markdown(f"### Hello, {username}! Today is **{TODAY}**")
@@ -51,7 +61,7 @@ mood_note = st.text_area("📝 Describe how you're feeling today (optional):")
 # ---------- HABIT CHECKLIST ----------
 st.markdown("### ✅ Select completed habits:")
 completed_today = []
-for habit in user_habits["habit"].tolist():
+for habit in user_habits:
     if st.checkbox(habit):
         completed_today.append(habit)
 
@@ -126,4 +136,3 @@ if os.path.exists(DATA_FILE):
         st.dataframe(df.sort_values(by="date", ascending=False), use_container_width=True)
 else:
     st.info("Start tracking to view analytics.")
-