Update app.py

app.py

@@ -5,28 +5,25 @@ import os
 
 DATA_FILE = "habits.csv"
 
-# ---------- Custom CSS (Modern Neon Blue UI) ----------
+# ---------- Page Config ----------
+st.set_page_config(page_title="Habit Tracker", layout="centered")
+
+# ---------- Light UI Styling ----------
 st.markdown("""
 <style>
-body {
-    background-color: #0f172a;
-    color: #e2e8f0;
-}
 .stApp {
-    background: linear-gradient(135deg, #0f172a, #1e293b);
-}
-h1, h2, h3 {
-    color: #38bdf8;
+    background-color: #f8fafc;
 }
-div.stButton > button {
-    background-color: #0ea5e9;
-    color: white;
-    border-radius: 10px;
-    transition: 0.3s;
+h1 {
+    color: #1e293b;
+    text-align: center;
 }
-div.stButton > button:hover {
-    background-color: #0284c7;
-    box-shadow: 0px 0px 10px #38bdf8;
+.habit-card {
+    background: white;
+    padding: 15px;
+    border-radius: 12px;
+    margin-bottom: 10px;
+    box-shadow: 0 2px 6px rgba(0,0,0,0.05);
 }
 </style>
 """, unsafe_allow_html=True)
@@ -44,72 +41,79 @@ def save_data(df):
 if "habits" not in st.session_state:
     st.session_state.habits = load_data()
 
-st.title("⚡ Atomic Habit Tracker")
+df = st.session_state.habits
+
+# ---------- Title ----------
+st.title("📌 Atomic Habit Tracker")
 
 # ---------- 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 valid habit")
-    elif new_habit in st.session_state.habits["Habit"].values:
-        st.warning("Habit already exists")
-    else:
-        new_row = pd.DataFrame([[new_habit, 0, "Never"]],
-                               columns=st.session_state.habits.columns)
-        st.session_state.habits = pd.concat(
-            [st.session_state.habits, new_row],
-            ignore_index=True
-        )
-        save_data(st.session_state.habits)
-        st.success("Habit added!")
-
-# ---------- Mark Done (Checkbox UI) ----------
-st.subheader("✅ Daily Tracker")
+
+col1, col2 = st.columns([3, 1])
+with col1:
+    new_habit = st.text_input("Enter habit name", label_visibility="collapsed")
+with col2:
+    if st.button("Add"):
+        if new_habit.strip() == "":
+            st.warning("Enter valid habit")
+        elif new_habit in df["Habit"].values:
+            st.warning("Already exists")
+        else:
+            new_row = pd.DataFrame([[new_habit, 0, "Never"]],
+                                   columns=df.columns)
+            df = pd.concat([df, new_row], ignore_index=True)
+            st.session_state.habits = df
+            save_data(df)
+            st.success("Added!")
+
+# ---------- Habit List ----------
+st.subheader("✅ Today")
 
 today = datetime.now().date()
-df = st.session_state.habits
 
-for i, row in df.iterrows():
-    col1, col2 = st.columns([3, 1])
+if df.empty:
+    st.info("No habits yet. Add one above.")
+else:
+    for i, row in df.iterrows():
+        col1, col2, col3 = st.columns([4, 2, 1])
+
+        with col1:
+            st.markdown(f"**{row['Habit']}**")
 
-    with col1:
-        st.write(f"**{row['Habit']}**  🔥 Streak: {row['Streak']}")
+        with col2:
+            st.write(f"🔥 {row['Streak']}")
 
-    with col2:
-        if st.checkbox("Done", key=f"done_{i}"):
+        with col3:
+            if st.button("Done", key=f"done_{i}"):
 
-            last_date = row["Last Completed"]
+                last_date = row["Last Completed"]
 
-            if last_date != "Never":
-                last_date = pd.to_datetime(last_date).date()
+                if last_date != "Never":
+                    last_date = pd.to_datetime(last_date).date()
 
-                if (today - last_date).days == 1:
-                    df.loc[i, "Streak"] += 1
-                elif (today - last_date).days > 1:
+                    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:
-                df.loc[i, "Streak"] = 1
 
-            df.loc[i, "Last Completed"] = str(today)
-            st.success(f"{row['Habit']} completed!")
+                df.loc[i, "Last Completed"] = str(today)
 
-# Save after loop
-st.session_state.habits = df
-save_data(df)
+                st.session_state.habits = df
+                save_data(df)
 
-# ---------- Graph ----------
-st.subheader("📊 Streak Progress")
+                st.rerun()
+
+# ---------- Chart ----------
+st.subheader("📊 Progress")
 
 if not df.empty:
     chart_data = df.set_index("Habit")["Streak"]
     st.bar_chart(chart_data)
-else:
-    st.info("No data yet")
 
-# ---------- Delete ----------
-st.subheader("🗑️ Manage")
+# ---------- Manage ----------
+st.subheader("⚙️ Manage")
 
 if not df.empty:
     habit_to_delete = st.selectbox("Delete habit", df["Habit"])
@@ -118,7 +122,7 @@ if not df.empty:
         df = df[df["Habit"] != habit_to_delete]
         st.session_state.habits = df
         save_data(df)
-        st.warning("Habit deleted!")
+        st.rerun()
 
 # ---------- Reset ----------
 if st.button("Reset All"):
@@ -126,4 +130,4 @@ if st.button("Reset All"):
         columns=["Habit", "Streak", "Last Completed"]
     )
     save_data(st.session_state.habits)
-    st.error("All habits removed!")
+    st.rerun()