Update task_operations.py

task_operations.py

@@ -1,8 +1,7 @@
-import pandas as pd
 import os
+import pandas as pd
 from datetime import datetime
 import streamlit as st
-import uuid  # For generating unique task IDs
 
 class TaskManager:
     TASKS_FILE = "tasks.csv"
@@ -15,10 +14,14 @@ class TaskManager:
     def load_tasks(self):
         if os.path.exists(self.TASKS_FILE):
             if os.path.getsize(self.TASKS_FILE) > 0:
-                return pd.read_csv(self.TASKS_FILE, parse_dates=['Task Time']).to_dict(orient='records')
+                try:
+                    return pd.read_csv(self.TASKS_FILE, parse_dates=['Task Time']).to_dict(orient='records')
+                except pd.errors.EmptyDataError:
+                    return []  # Return an empty list if CSV is empty
             else:
                 return []
         else:
+            # If the file doesn't exist, create it with the correct columns
             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 []
@@ -29,9 +32,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_id = len(st.session_state.tasks) + 1  # Generate a unique task ID
         task_entry = {
-            "Task ID": task_id,  # Add Task ID
+            "Task ID": task_id,
             "Task Name": task_name,
             "Task Time": task_time_full,
             "Task Duration (hours)": int(task_duration_hours),
@@ -41,13 +44,15 @@ class TaskManager:
         st.session_state.tasks.append(task_entry)
         self.save_tasks(st.session_state.tasks)
 
-    def delete_task_by_id(self, task_id):  # Modified method to delete by Task ID
+    def delete_task_by_id(self, task_id):
+        task_found = False
         for index, task in enumerate(st.session_state.tasks):
-            if task['Task ID'] == task_id:  # Match by Task ID
+            if task['Task ID'] == int(task_id):
                 st.session_state.tasks.pop(index)
-                self.save_tasks(st.session_state.tasks)
-                return True
-        return False
+                task_found = True
+                break
+        self.save_tasks(st.session_state.tasks)
+        return task_found
 
     def generate_report(self, timeframe):
         df = pd.DataFrame(st.session_state.tasks)