Update app.py

app.py

@@ -5,30 +5,54 @@ import os
 
 DATA_FILE = "habits.csv"
 
-# Load data
+# ---------- Custom CSS (Modern Neon Blue UI) ----------
+st.markdown("""
+<style>
+body {
+    background-color: #0f172a;
+    color: #e2e8f0;
+}
+.stApp {
+    background: linear-gradient(135deg, #0f172a, #1e293b);
+}
+h1, h2, h3 {
+    color: #38bdf8;
+}
+div.stButton > button {
+    background-color: #0ea5e9;
+    color: white;
+    border-radius: 10px;
+    transition: 0.3s;
+}
+div.stButton > button:hover {
+    background-color: #0284c7;
+    box-shadow: 0px 0px 10px #38bdf8;
+}
+</style>
+""", unsafe_allow_html=True)
+
+# ---------- Load / Save ----------
 def load_data():
     if os.path.exists(DATA_FILE):
         return pd.read_csv(DATA_FILE)
-    else:
-        return pd.DataFrame(columns=["Habit", "Streak", "Last Completed"])
+    return pd.DataFrame(columns=["Habit", "Streak", "Last Completed"])
 
-# Save data
 def save_data(df):
     df.to_csv(DATA_FILE, index=False)
 
-# Initialize
+# ---------- Init ----------
 if "habits" not in st.session_state:
     st.session_state.habits = load_data()
 
-st.title("📌 Atomic Habit Tracker")
+st.title("⚡ Atomic Habit Tracker")
 
-# --- Add Habit ---
-st.subheader("Add New Habit")
-new_habit = st.text_input("Enter habit name")
+# ---------- Add Habit ----------
+st.subheader("➕ Add Habit")
+new_habit = st.text_input("Habit name")
 
 if st.button("Add Habit"):
     if new_habit.strip() == "":
-        st.warning("Enter a valid habit name")
+        st.warning("Enter valid habit")
     elif new_habit in st.session_state.habits["Habit"].values:
         st.warning("Habit already exists")
     else:
@@ -39,65 +63,67 @@ if st.button("Add Habit"):
             ignore_index=True
         )
         save_data(st.session_state.habits)
-        st.success(f"Habit '{new_habit}' added!")
+        st.success("Habit added!")
 
-# --- Mark Habit Done ---
-st.subheader("Mark Habit as Done")
-habit_list = st.session_state.habits["Habit"].tolist()
+# ---------- Mark Done (Checkbox UI) ----------
+st.subheader("✅ Daily Tracker")
 
-if habit_list:
-    selected_habit = st.selectbox("Select Habit", habit_list)
+today = datetime.now().date()
+df = st.session_state.habits
 
-    if st.button("Mark Done"):
-        today = datetime.now().date()
-        df = st.session_state.habits
+for i, row in df.iterrows():
+    col1, col2 = st.columns([3, 1])
 
-        idx = df[df["Habit"] == selected_habit].index[0]
-        last_date = df.loc[idx, "Last Completed"]
+    with col1:
+        st.write(f"**{row['Habit']}**  🔥 Streak: {row['Streak']}")
 
-        # Streak logic
-        if last_date != "Never":
-            last_date = pd.to_datetime(last_date).date()
-            if (today - last_date).days == 1:
-                df.loc[idx, "Streak"] += 1
-            elif (today - last_date).days > 1:
-                df.loc[idx, "Streak"] = 1
-        else:
-            df.loc[idx, "Streak"] = 1
+    with col2:
+        if st.checkbox("Done", key=f"done_{i}"):
 
-        df.loc[idx, "Last Completed"] = str(today)
+            last_date = row["Last Completed"]
 
-        st.session_state.habits = df
-        save_data(df)
+            if last_date != "Never":
+                last_date = pd.to_datetime(last_date).date()
 
-        st.success(f"Habit '{selected_habit}' marked as done!")
+                if (today - last_date).days == 1:
+                    df.loc[i, "Streak"] += 1
+                elif (today - last_date).days > 1:
+                    df.loc[i, "Streak"] = 1
+            else:
+                df.loc[i, "Streak"] = 1
 
-else:
-    st.info("No habits added yet.")
+            df.loc[i, "Last Completed"] = str(today)
+            st.success(f"{row['Habit']} completed!")
 
-# --- Display Data ---
-st.subheader("Your Habits")
-st.dataframe(st.session_state.habits)
+# Save after loop
+st.session_state.habits = df
+save_data(df)
 
-# --- Delete Habit ---
-st.subheader("Delete Habit")
+# ---------- Graph ----------
+st.subheader("📊 Streak Progress")
 
-if habit_list:
-    delete_habit = st.selectbox("Select Habit to Delete", habit_list, key="delete")
+if not df.empty:
+    chart_data = df.set_index("Habit")["Streak"]
+    st.bar_chart(chart_data)
+else:
+    st.info("No data yet")
 
-    if st.button("Delete Habit"):
-        st.session_state.habits = st.session_state.habits[
-            st.session_state.habits["Habit"] != delete_habit
-        ]
-        save_data(st.session_state.habits)
-        st.success(f"Habit '{delete_habit}' deleted!")
+# ---------- Delete ----------
+st.subheader("🗑️ Manage")
+
+if not df.empty:
+    habit_to_delete = st.selectbox("Delete habit", df["Habit"])
 
-# --- Reset All ---
-st.subheader("Danger Zone ⚠️")
+    if st.button("Delete"):
+        df = df[df["Habit"] != habit_to_delete]
+        st.session_state.habits = df
+        save_data(df)
+        st.warning("Habit deleted!")
 
-if st.button("Reset All Habits"):
+# ---------- Reset ----------
+if st.button("Reset All"):
     st.session_state.habits = pd.DataFrame(
         columns=["Habit", "Streak", "Last Completed"]
     )
     save_data(st.session_state.habits)
-    st.warning("All habits cleared!")
+    st.error("All habits removed!")