Update task_operations.py

task_operations.py

@@ -1,9 +1,8 @@
-# components/task_operations.py
-
 import pandas as pd
 import os
 from datetime import datetime
 import streamlit as st
+import uuid  # For generating unique task IDs
 
 class TaskManager:
     TASKS_FILE = "tasks.csv"
@@ -20,7 +19,7 @@ class TaskManager:
             else:
                 return []
         else:
-            df = pd.DataFrame(columns=["Task Name", "Task Time", "Task Duration (hours)", "Task Duration (minutes)", "Category"])
+            df = pd.DataFrame(columns=["Task ID", "Task Name", "Task Time", "Task Duration (hours)", "Task Duration (minutes)", "Category"])
             df.to_csv(self.TASKS_FILE, index=False)
             return []
 
@@ -30,7 +29,9 @@ class TaskManager:
 
     def add_task(self, task_name, task_time, task_duration_hours, task_duration_minutes, task_category):
         task_time_full = datetime.combine(datetime.today(), task_time)
+        task_id = str(uuid.uuid4())  # Generate a unique ID
         task_entry = {
+            "Task ID": task_id,  # Add Task ID
             "Task Name": task_name,
             "Task Time": task_time_full,
             "Task Duration (hours)": int(task_duration_hours),
@@ -40,9 +41,9 @@ class TaskManager:
         st.session_state.tasks.append(task_entry)
         self.save_tasks(st.session_state.tasks)
 
-    def delete_task_by_name(self, task_name):
+    def delete_task_by_id(self, task_id):  # Modified method to delete by Task ID
         for index, task in enumerate(st.session_state.tasks):
-            if task['Task Name'] == task_name:
+            if task['Task ID'] == task_id:  # Match by Task ID
                 st.session_state.tasks.pop(index)
                 self.save_tasks(st.session_state.tasks)
                 return True