Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 from datetime import datetime
+import os
 
 # Color mapping
 STATUS_COLORS = {
@@ -9,24 +10,41 @@ STATUS_COLORS = {
     "Done": "#e6ffe6"
 }
 
-# Read assignees from text file
+# File paths
+TASKS_FILE = "tasks.csv"
+ASSIGNEE_FILE = "Assignee.txt"
+
+def load_tasks():
+    """Load tasks from CSV file"""
+    if os.path.exists(TASKS_FILE):
+        return pd.read_csv(TASKS_FILE, parse_dates=['Due Date'])
+    return pd.DataFrame(columns=[
+        'Title', 'Description', 'Assignee', 'Status', 'Due Date'
+    ])
+
+def save_tasks():
+    """Save tasks to CSV file"""
+    st.session_state.tasks.to_csv(TASKS_FILE, index=False)
+
 def load_assignees():
+    """Load assignees from text file"""
     assignees = []
     try:
-        with open("Assignee.txt", "r") as f:
+        with open(ASSIGNEE_FILE, "r") as f:
             lines = [line.strip() for line in f.readlines() if line.strip()]
-            
-            # Parse the format from your example
             for line in lines:
                 if '-' in line:
                     name_part = line.split('-')[0].strip()
                     if ' ' in name_part:
-                        name = name_part
-                        assignees.append(name)
+                        assignees.append(name_part)
     except FileNotFoundError:
         st.error("Assignee.txt file not found!")
     return assignees
 
+# Initialize session state
+if 'tasks' not in st.session_state:
+    st.session_state.tasks = load_tasks()
+
 # Initialize task database
 if 'tasks' not in st.session_state:
     st.session_state.tasks = pd.DataFrame(columns=[
@@ -41,10 +59,7 @@ with st.sidebar:
     st.header("➕ Add New Task")
     title = st.text_input("Task Title")
     description = st.text_area("Description")
-    
-    # Assignee dropdown from txt file
-    assignee = st.selectbox("Assignee", assignee_list)
-    
+    assignee = st.selectbox("Assignee", load_assignees())
     status = st.selectbox("Status", ["To Do", "In Progress", "Done"])
     due_date = st.date_input("Due Date")
     
@@ -60,6 +75,7 @@ with st.sidebar:
             st.session_state.tasks,
             pd.DataFrame([new_task])
         ], ignore_index=True)
+        save_tasks()
 
 # Function to create styled task cards (same as before)
 def task_card(title, description, assignee, status, due_date):
@@ -107,6 +123,7 @@ for idx, task in st.session_state.tasks.iterrows():
     
     if new_status != task['Status']:
         st.session_state.tasks.at[idx, 'Status'] = new_status
+        save_tasks()  # Save after updating
         st.rerun()
     
     st.markdown("</div>", unsafe_allow_html=True)