Update app.py

app.py

@@ -9,10 +9,6 @@ import os
 st.set_page_config(page_title="Task Manager with Procrastination Alerts", page_icon="📝")
 st.title("📝 Task Manager with Procrastination Alerts")
 
-# Temporary storage for user tasks
-user_tasks = {}
-user_preferences = {"name": "John Doe", "email": "john@gmail.com"}
-
 # Logging setup
 logging.basicConfig(level=logging.INFO)
 
@@ -36,7 +32,7 @@ def filter_tasks(section):
     end_of_week = start_of_week + timedelta(days=6)
 
     filtered_tasks = []
-    for task in user_tasks.values():
+    for task in st.session_state.user_tasks.values():
         task_date = datetime.strptime(task['date'], '%Y-%m-%d').date()
         if section == "myDay" and task_date == today:
             filtered_tasks.append(task)
@@ -48,7 +44,7 @@ def filter_tasks(section):
 
 def check_deadlines():
     today = datetime.now().date()
-    return [task for task in user_tasks.values() if today <= datetime.strptime(task['date'], '%Y-%m-%d').date() <= today + timedelta(days=2)]
+    return [task for task in st.session_state.user_tasks.values() if today <= datetime.strptime(task['date'], '%Y-%m-%d').date() <= today + timedelta(days=2)]
 
 def alert_user(upcoming_tasks):
     procrastination_messages = []
@@ -69,52 +65,86 @@ def alert_user(upcoming_tasks):
         procrastination_messages.append(procrastination_message)
     return procrastination_messages
 
-# Streamlit UI
-st.sidebar.title("Task Filters")
-section = st.sidebar.radio("View Tasks", ["myDay", "thisWeek", "all"])
-
+# Initialize session state variables
+if "logged_in" not in st.session_state:
+    st.session_state.logged_in = False
 if "user_tasks" not in st.session_state:
     st.session_state.user_tasks = {}
 
-# Task Input Form
-st.subheader("Add a New Task")
-with st.form("add_task_form"):
-    task_text = st.text_input("Task Description")
-    due_date = st.date_input("Due Date")
-    if st.form_submit_button("Add Task"):
-        task_id = generate_numeric_id()
-        st.session_state.user_tasks[task_id] = {
-            "id": task_id,
-            "text": task_text,
-            "date": due_date.strftime('%Y-%m-%d'),
-            "completed": False
-        }
-        st.success("Task added successfully.")
-
-# Display Tasks
-st.subheader("Your Tasks")
-tasks_to_display = filter_tasks(section)
-for task in tasks_to_display:
-    col1, col2, col3 = st.columns([5, 1, 1])
-    col1.text(f"{task['text']} (Due: {task['date']})")
-    if col2.button("Toggle Complete", key=f"toggle_{task['id']}"):
-        task["completed"] = not task["completed"]
-        st.success(f"Task marked as {'completed' if task['completed'] else 'incomplete'}.")
-    if col3.button("Delete Task", key=f"delete_{task['id']}"):
-        del st.session_state.user_tasks[task["id"]]
-        st.success("Task deleted.")
-
-# Procrastination Alerts
-st.subheader("Upcoming Deadlines")
-upcoming_tasks = check_deadlines()
-if upcoming_tasks:
-    if st.button("Get Procrastination Message"):
-        messages = alert_user(upcoming_tasks)
-        st.info(random.choice(messages))
-else:
-    st.info("No tasks with urgent deadlines.")
+# Login Page
+def login():
+    st.subheader("Login")
+    with st.form("login_form"):
+        name = st.text_input("Name")
+        email = st.text_input("Email")
+        if st.form_submit_button("Login"):
+            if name and email:
+                st.session_state.logged_in = True
+                st.session_state.user_name = name
+                st.session_state.user_email = email
+                st.success("Logged in successfully.")
+                st.experimental_rerun()
+            else:
+                st.warning("Please enter both name and email.")
+
+# Logout function
+def logout():
+    st.session_state.logged_in = False
+    st.session_state.user_name = None
+    st.session_state.user_email = None
+    st.success("Logged out successfully.")
+    st.experimental_rerun()
+
+# Main App
+if st.session_state.logged_in:
+    st.sidebar.title(f"Welcome, {st.session_state.user_name}!")
+    if st.sidebar.button("Logout"):
+        logout()
 
-# Clear All Tasks
-if st.button("Delete All Tasks"):
-    st.session_state.user_tasks.clear()
-    st.success("All tasks deleted.")
+    st.sidebar.title("Task Filters")
+    section = st.sidebar.radio("View Tasks", ["myDay", "thisWeek", "all"])
+
+    # Task Input Form
+    st.subheader("Add a New Task")
+    with st.form("add_task_form"):
+        task_text = st.text_input("Task Description")
+        due_date = st.date_input("Due Date")
+        if st.form_submit_button("Add Task"):
+            task_id = generate_numeric_id()
+            st.session_state.user_tasks[task_id] = {
+                "id": task_id,
+                "text": task_text,
+                "date": due_date.strftime('%Y-%m-%d'),
+                "completed": False
+            }
+            st.success("Task added successfully.")
+
+    # Display Tasks
+    st.subheader("Your Tasks")
+    tasks_to_display = filter_tasks(section)
+    for task in tasks_to_display:
+        col1, col2, col3 = st.columns([5, 1, 1])
+        col1.text(f"{task['text']} (Due: {task['date']})")
+        if col2.button("Toggle Complete", key=f"toggle_{task['id']}"):
+            task["completed"] = not task["completed"]
+            st.success(f"Task marked as {'completed' if task['completed'] else 'incomplete'}.")
+        if col3.button("Delete Task", key=f"delete_{task['id']}"):
+            del st.session_state.user_tasks[task["id"]]
+            st.success("Task deleted.")
+
+    # Procrastination Alerts
+    st.subheader("Upcoming Deadlines")
+    upcoming_tasks = check_deadlines()
+    if upcoming_tasks:
+        if st.button("Get Procrastination Message"):
+            messages = alert_user(upcoming_tasks)
+            st.info(random.choice(messages))
+    else:
+        st.info("No tasks with urgent deadlines.")
+
+    # Clear All Tasks
+    if st.button("Delete All Tasks"):
+        st.session_state.user_tasks.clear()
+        st.success("All tasks deleted.")
+else:
+    login()