Update app.py

app.py

@@ -1,124 +1,102 @@
-import streamlit as st
-import pandas as pd
-import plotly.express as px
-from datetime import datetime, timedelta
-import json
-import os
+def main():
+    import streamlit as st
+    import pandas as pd
+    import plotly.express as px
+    from datetime import datetime, timedelta
+    import json
+    import os
 
-# ----------------------------
-# App Config
-# ----------------------------
-st.set_page_config(page_title="Habit & To-Do Tracker", layout="wide")
-st.title("📋 Habit & To-Do Tracker")
+    st.set_page_config(page_title="Habit & To-Do Tracker", layout="wide")
+    st.title("📋 Habit & To-Do Tracker")
 
-# ----------------------------
-# File Storage
-# ----------------------------
-DATA_FILE = "data.json"
+    DATA_FILE = "data.json"
 
-def load_data():
+    # 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 = json.load(f)
+    else:
+        data = {"tasks": [], "habits": {}}
 
-data = load_data()
+    # Sidebar: Add Task
+    st.sidebar.header("➕ Add Task")
+    task_name = st.sidebar.text_input("Task Name")
+    task_completed = st.sidebar.checkbox("Completed")
+    if st.sidebar.button("Add Task") and task_name.strip():
+        today = datetime.today().strftime("%Y-%m-%d")
+        data["tasks"].append({
+            "Task": task_name,
+            "Completed": task_completed,
+            "Date": today
+        })
+        with open(DATA_FILE, "w") as f:
+            json.dump(data, f)
+        st.sidebar.success("Task added!")
 
-# ----------------------------
-# Sidebar: Add Task
-# ----------------------------
-st.sidebar.header("➕ Add Task")
-task_name = st.sidebar.text_input("Task Name")
-task_completed = st.sidebar.checkbox("Completed")
-if st.sidebar.button("Add Task") and task_name.strip():
-    today = datetime.today().strftime("%Y-%m-%d")
-    data["tasks"].append({
-        "Task": task_name,
-        "Completed": task_completed,
-        "Date": today
-    })
-    save_data(data)
-    st.sidebar.success("Task added!")
+    # Sidebar: Track Habit
+    st.sidebar.header("🔥 Track Habit")
+    habit_name = st.sidebar.text_input("Habit Name")
+    if st.sidebar.button("Mark Habit Done") and habit_name.strip():
+        today = datetime.today().strftime("%Y-%m-%d")
+        if habit_name not in data["habits"]:
+            data["habits"][habit_name] = []
+        if today not in data["habits"][habit_name]:
+            data["habits"][habit_name].append(today)
+            with open(DATA_FILE, "w") as f:
+                json.dump(data, f)
+            st.sidebar.success("Habit marked for today!")
+        else:
+            st.sidebar.warning("Already marked today.")
 
-# ----------------------------
-# Sidebar: Track Habit
-# ----------------------------
-st.sidebar.header("🔥 Track Habit")
-habit_name = st.sidebar.text_input("Habit Name")
-if st.sidebar.button("Mark Habit Done") and habit_name.strip():
-    today = datetime.today().strftime("%Y-%m-%d")
-    if habit_name not in data["habits"]:
-        data["habits"][habit_name] = []
-    if today not in data["habits"][habit_name]:
-        data["habits"][habit_name].append(today)
-        save_data(data)
-        st.sidebar.success("Habit marked for today!")
+    # Display Tasks
+    st.subheader("📌 Tasks")
+    if data["tasks"]:
+        task_df = pd.DataFrame(data["tasks"])
+        st.dataframe(task_df, use_container_width=True)
     else:
-        st.sidebar.warning("Already marked today.")
+        st.info("No tasks yet.")
 
-# ----------------------------
-# Display Tasks
-# ----------------------------
-st.subheader("📌 Tasks")
-if data["tasks"]:
-    task_df = pd.DataFrame(data["tasks"])
-    st.dataframe(task_df, use_container_width=True)
-else:
-    st.info("No tasks yet.")
+    # Display Habits
+    st.subheader("🔥 Habits")
+    habit_data = []
+    for habit, dates in data["habits"].items():
+        dates_sorted = sorted(dates)
+        streak = 0
+        today = datetime.today().date()
+        for i in range(21):
+            day = today - timedelta(days=i)
+            if day.strftime("%Y-%m-%d") in dates_sorted:
+                streak += 1
+            else:
+                break
+        habit_type = "🌟 Permanent Habit" if streak >= 21 else "⏳ Temporary Habit"
+        habit_data.append({"Habit": habit, "Days Followed": len(dates), "Streak": streak, "Type": habit_type})
 
-# ----------------------------
-# Display Habits
-# ----------------------------
-st.subheader("🔥 Habits")
-habit_data = []
-for habit, dates in data["habits"].items():
-    # Sort dates for streak calculation
-    dates_sorted = sorted(dates)
-    streak = 0
-    # Calculate consecutive days streak ending today
-    today = datetime.today().date()
-    for i in range(21):
-        day = today - timedelta(days=i)
-        if day.strftime("%Y-%m-%d") in dates_sorted:
-            streak += 1
-        else:
-            break
-    habit_type = "🌟 Permanent Habit" if streak >= 21 else "⏳ Temporary Habit"
-    habit_data.append({"Habit": habit, "Days Followed": len(dates), "Streak": streak, "Type": habit_type})
-
-if habit_data:
-    habit_df = pd.DataFrame(habit_data)
-    st.dataframe(habit_df, use_container_width=True)
+    if habit_data:
+        habit_df = pd.DataFrame(habit_data)
+        st.dataframe(habit_df, use_container_width=True)
+        st.subheader("📊 Habit Analytics")
+        st.plotly_chart(px.bar(habit_df, x="Habit", y="Days Followed", color="Type"), use_container_width=True)
+        st.plotly_chart(px.pie(habit_df, names="Habit", values="Days Followed"), use_container_width=True)
+    else:
+        st.info("No habits tracked yet.")
 
-    # ----------------------------
-    # Visualizations
-    # ----------------------------
-    st.subheader("📊 Habit Analytics")
-    st.plotly_chart(px.bar(habit_df, x="Habit", y="Days Followed", color="Type"), use_container_width=True)
-    st.plotly_chart(px.pie(habit_df, names="Habit", values="Days Followed"), use_container_width=True)
-else:
-    st.info("No habits tracked yet.")
+    # Daily Report
+    st.subheader("📑 Daily Report")
+    today_str = datetime.today().strftime("%Y-%m-%d")
+    today_tasks = [t for t in data["tasks"] if t["Date"] == today_str]
+    report_df = pd.DataFrame([{
+        "Date": today_str,
+        "Total Tasks Today": len(today_tasks),
+        "Completed Tasks Today": len([t for t in today_tasks if t["Completed"]]),
+        "Total Habits": len(data["habits"])
+    }])
+    st.table(report_df)
 
-# ----------------------------
-# Daily Report
-# ----------------------------
-st.subheader("📑 Daily Report")
-today_str = datetime.today().strftime("%Y-%m-%d")
-today_tasks = [t for t in data["tasks"] if t["Date"] == today_str]
-report_df = pd.DataFrame([{
-    "Date": today_str,
-    "Total Tasks Today": len(today_tasks),
-    "Completed Tasks Today": len([t for t in today_tasks if t["Completed"]]),
-    "Total Habits": len(data["habits"])
-}])
-st.table(report_df)
+    # Download CSV
+    csv_data = report_df.to_csv(index=False).encode("utf-8")
+    st.download_button("⬇ Download CSV", csv_data, "daily_report.csv", "text/csv")
 
-# ----------------------------
-# Download CSV
-# ----------------------------
-csv_data = report_df.to_csv(index=False).encode("utf-8")
-st.download_button("⬇ Download CSV", csv_data, "daily_report.csv", "text/csv")
+# Run the app
+if __name__ == "__main__":
+    main()