Update app.py

app.py

@@ -48,11 +48,17 @@ def check_deadlines():
 
 def alert_user(upcoming_tasks):
     procrastination_messages = []
+    # Sample procrastination prompts
+    procrastination_prompts = [
+        "Take a break! You can do '{task}' later.",
+        "Why not relax for a while? '{task}' isn't going anywhere.",
+        "You've earned a rest! Save '{task}' for later.",
+        "Maybe watch a quick video before starting '{task}'?",
+        "A quick snack could be just what you need before tackling '{task}'.",
+    ]
+    
     for task in upcoming_tasks:
-        prompt = (
-            f"You are my task manager who will prevent me from completing tasks. "
-            f"Please make me procrastinate on '{task['text']}' due on {task['date']}."
-        )
+        prompt = random.choice(procrastination_prompts).format(task=task['text'])
         try:
             response = client.chat_completion(
                 messages=[{"role": "user", "content": prompt}],
@@ -71,6 +77,21 @@ if "logged_in" not in st.session_state:
 if "user_tasks" not in st.session_state:
     st.session_state.user_tasks = {}
 
+# Display tasks function
+def display_tasks(tasks):
+    for task in tasks:
+        col1, col2, col3 = st.columns([5, 1, 1])
+        with col1:
+            st.markdown(f"**{task['text']}** - Due: {task['date']} - {'Completed' if task['completed'] else 'Pending'}")
+        with col2:
+            if st.button("Toggle Complete", key=f"toggle_{task['id']}"):
+                task["completed"] = not task["completed"]
+                st.experimental_rerun()
+        with col3:
+            if st.button("Delete", key=f"delete_{task['id']}"):
+                del st.session_state.user_tasks[task["id"]]
+                st.experimental_rerun()
+
 # Login Page
 def login():
     st.subheader("Login")
@@ -122,25 +143,14 @@ if st.session_state.logged_in:
     # 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.")
+    display_tasks(tasks_to_display)
 
     # 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.")
+    if upcoming_tasks and st.button("Get Procrastination Message"):
+        messages = alert_user(upcoming_tasks)
+        st.info(random.choice(messages))
 
     # Clear All Tasks
     if st.button("Delete All Tasks"):