Update app.py

app.py

@@ -10,9 +10,16 @@ STATUS_COLORS = {
     "Done": "#e6ffe6"
 }
 
-TASKS_FILE = "/data/tasks.csv"  # Hugging Face Spaces persistent storage
+# Adjust these file paths according to your environment
+TASKS_FILE = "tasks.csv"  # Changed from /data/tasks.csv for local testing
 ASSIGNEE_FILE = "Assignee.txt"
 
+def create_empty_df():
+    """Create an empty DataFrame with the required columns"""
+    return pd.DataFrame(columns=[
+        'Title', 'Description', 'Assignee', 'Status', 'Due Date'
+    ])
+
 def load_tasks():
     """Load tasks from CSV file"""
     try:
@@ -24,7 +31,7 @@ def load_tasks():
         return create_empty_df()
 
 def save_tasks():
-    """Save tasks to CSV file with proper permissions"""
+    """Save tasks to CSV file with proper error handling"""
     try:
         st.session_state.tasks.to_csv(TASKS_FILE, index=False)
     except PermissionError:
@@ -32,15 +39,6 @@ def save_tasks():
     except Exception as e:
         st.error(f"Error saving tasks: {str(e)}")
 
-def create_empty_df():
-    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 = []
@@ -53,40 +51,12 @@ def load_assignees():
                     if ' ' in name_part:
                         assignees.append(name_part)
     except FileNotFoundError:
-        st.error("Assignee.txt file not found!")
+        st.error(f"Assignee file '{ASSIGNEE_FILE}' not found!")
+        # Return a default list to prevent app from crashing
+        return ["User 1", "User 2"]
     return assignees
 
-# Initialize session state ONLY ONCE
-if 'tasks' not in st.session_state:
-    st.session_state.tasks = load_tasks()  # This line was duplicated before
-
-# Load assignees once and reuse
-assignee_list = load_assignees()
-
-# Sidebar for new tasks
-with st.sidebar:
-    st.header("➕ Add New Task")
-    title = st.text_input("Task Title")
-    description = st.text_area("Description")
-    assignee = st.selectbox("Assignee", load_assignees())
-    status = st.selectbox("Status", ["To Do", "In Progress", "Done"])
-    due_date = st.date_input("Due Date")
-    
-    if st.button("Add Task"):
-        new_task = {
-            'Title': title,
-            'Description': description,
-            'Assignee': assignee,
-            'Status': status,
-            'Due Date': due_date
-        }
-        st.session_state.tasks = pd.concat([
-            st.session_state.tasks,
-            pd.DataFrame([new_task])
-        ], ignore_index=True)
-        save_tasks()
-
-# Function to create styled task cards (same as before)
+# Function to create styled task cards with proper HTML
 def task_card(title, description, assignee, status, due_date):
     color = STATUS_COLORS.get(status, "#ffffff")
     card = f"""
@@ -101,39 +71,77 @@ def task_card(title, description, assignee, status, due_date):
         <p><strong>Assignee:</strong> {assignee}</p>
         <p><strong>Due Date:</strong> {due_date}</p>
         <p><strong>Description:</strong> {description}</p>
-        <div style="margin-top: 15px;">
+        <div style="margin-top: 15px;"></div>
+    </div>
     """
     return card
 
-# Main display
+# Initialize session state ONLY ONCE
+if 'tasks' not in st.session_state:
+    st.session_state.tasks = load_tasks()
+
+# Load assignees once and reuse
+assignee_list = load_assignees()
+
+# App UI starts here
 st.title("Company Task Board 🗒️")
+
+# Sidebar for new tasks
+with st.sidebar:
+    st.header("➕ Add New Task")
+    title = st.text_input("Task Title")
+    description = st.text_area("Description")
+    assignee = st.selectbox("Assignee", assignee_list)  # Use the preloaded list
+    status = st.selectbox("Status", ["To Do", "In Progress", "Done"])
+    due_date = st.date_input("Due Date")
+    
+    if st.button("Add Task"):
+        if title:  # Basic validation
+            new_task = {
+                'Title': title,
+                'Description': description,
+                'Assignee': assignee,
+                'Status': status,
+                'Due Date': due_date
+            }
+            st.session_state.tasks = pd.concat([
+                st.session_state.tasks,
+                pd.DataFrame([new_task])
+            ], ignore_index=True)
+            save_tasks()
+            st.success("Task added successfully!")
+        else:
+            st.warning("Please enter a task title")
+
 st.divider()
 
 # Display tasks in cards
-for idx, task in st.session_state.tasks.iterrows():
-    card = task_card(
-        task['Title'],
-        task['Description'],
-        task['Assignee'],
-        task['Status'],
-        task['Due Date']
-    )
-    
-    # Display the card
-    st.markdown(card, unsafe_allow_html=True)
-    
-    # Status update dropdown below the card
-    new_status = st.selectbox(
-        "Update Status",
-        ["To Do", "In Progress", "Done"],
-        index=["To Do", "In Progress", "Done"].index(task['Status']),
-        key=f"status_{idx}"
-    )
-    
-    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)
-    st.write("")  # Add some spacing between cards
+if st.session_state.tasks.empty:
+    st.info("No tasks yet. Add your first task using the sidebar.")
+else:
+    for idx, task in st.session_state.tasks.iterrows():
+        card = task_card(
+            task['Title'],
+            task['Description'],
+            task['Assignee'],
+            task['Status'],
+            task['Due Date'].strftime('%Y-%m-%d') if isinstance(task['Due Date'], datetime) else task['Due Date']
+        )
+        
+        # Display the card
+        st.markdown(card, unsafe_allow_html=True)
+        
+        # Status update dropdown below the card
+        new_status = st.selectbox(
+            "Update Status",
+            ["To Do", "In Progress", "Done"],
+            index=["To Do", "In Progress", "Done"].index(task['Status']),
+            key=f"status_{idx}"
+        )
+        
+        if new_status != task['Status']:
+            st.session_state.tasks.at[idx, 'Status'] = new_status
+            save_tasks()  # Save after updating
+            st.rerun()
+        
+        st.write("")