Update task_operations.py

task_operations.py

@@ -13,37 +13,46 @@ class TaskManager:
             st.session_state.tasks = self.load_tasks()
 
     def load_tasks(self):
-        if os.path.exists(self.TASKS_FILE):
-            if os.path.getsize(self.TASKS_FILE) > 0:
-                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:
+        # Check if the file exists and has data
+        if os.path.exists(self.TASKS_FILE) and os.path.getsize(self.TASKS_FILE) > 0:
+            try:
+                # Load tasks into session state from CSV
+                return pd.read_csv(self.TASKS_FILE, parse_dates=['Task Time']).to_dict(orient='records')
+            except pd.errors.EmptyDataError:
+                # If the file is empty, return an empty list
                 return []
         else:
-            # If the file doesn't exist, create it with the correct columns
+            # If file doesn't exist, initialize it with an empty DataFrame and 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 []
+            return []  # Return an empty list to the session state
 
     def save_tasks(self, tasks):
-        df = pd.DataFrame(tasks)
-        df.to_csv(self.TASKS_FILE, index=False)
+        if tasks:  # Only save if tasks are not empty
+            df = pd.DataFrame(tasks)
+            df.to_csv(self.TASKS_FILE, index=False)
+        else:
+            # Ensure the file is not accidentally overwritten with an empty DataFrame
+            st.warning("No tasks to save. Skipping file write to prevent data loss.")
 
     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 using uuid4
-        task_entry = {
-            "Task ID": task_id,
-            "Task Name": task_name,
-            "Task Time": task_time_full,
-            "Task Duration (hours)": int(task_duration_hours),
-            "Task Duration (minutes)": int(task_duration_minutes),
-            "Category": task_category
-        }
-        st.session_state.tasks.append(task_entry)
-        self.save_tasks(st.session_state.tasks)
+        # Ensure valid task details are being added
+        if task_name and task_category:
+            task_time_full = datetime.combine(datetime.today(), task_time)
+            task_id = str(uuid.uuid4())  # Generate a unique ID using uuid4
+            task_entry = {
+                "Task ID": task_id,
+                "Task Name": task_name,
+                "Task Time": task_time_full,
+                "Task Duration (hours)": int(task_duration_hours),
+                "Task Duration (minutes)": int(task_duration_minutes),
+                "Category": task_category
+            }
+            st.session_state.tasks.append(task_entry)
+            self.save_tasks(st.session_state.tasks)
+            st.success(f"Task '{task_name}' added successfully!")
+        else:
+            st.error("Task name and category are required.")
 
     def delete_task_by_id(self, task_id):
         task_found = False
@@ -52,7 +61,11 @@ class TaskManager:
                 st.session_state.tasks.pop(index)
                 task_found = True
                 break
-        self.save_tasks(st.session_state.tasks)
+        if task_found:
+            self.save_tasks(st.session_state.tasks)
+            st.success(f"Task with ID '{task_id}' deleted.")
+        else:
+            st.error(f"Task with ID '{task_id}' not found.")
         return task_found
 
     def generate_report(self, timeframe):