Update task_operations.py

task_operations.py

@@ -3,24 +3,34 @@ import pandas as pd
 from datetime import datetime
 import streamlit as st
 import uuid  # For generating unique IDs
+import shutil  # To create a backup of the file
 
 class TaskManager:
     TASKS_FILE = "tasks.csv"
+    BACKUP_FILE = "tasks_backup.csv"
     CATEGORIES = ['Learning', 'Gym', 'Personal', 'Family', 'Work', 'Prayer']
 
     def __init__(self):
         if 'tasks' not in st.session_state:
             st.session_state.tasks = self.load_tasks()
 
+    def backup_csv(self):
+        """Create a backup of the current tasks file."""
+        if os.path.exists(self.TASKS_FILE):
+            shutil.copy(self.TASKS_FILE, self.BACKUP_FILE)
+
+    def restore_from_backup(self):
+        """Restore tasks from backup if needed."""
+        if os.path.exists(self.BACKUP_FILE):
+            shutil.copy(self.BACKUP_FILE, self.TASKS_FILE)
+
     def load_tasks(self):
         # 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 []
+                return []  # Return empty list if CSV is empty
         else:
             # 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"])
@@ -28,12 +38,15 @@ class TaskManager:
             return []  # Return an empty list to the session state
 
     def save_tasks(self, tasks):
+        """Save tasks to the CSV file after making a backup."""
         if tasks:  # Only save if tasks are not empty
+            # Backup the current file before saving
+            self.backup_csv()
             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.")
+            # Prevent overwriting the file with an empty DataFrame
+            st.warning("No tasks to save. File write skipped to prevent data loss.")
 
     def add_task(self, task_name, task_time, task_duration_hours, task_duration_minutes, task_category):
         # Ensure valid task details are being added
@@ -51,6 +64,8 @@ class TaskManager:
             st.session_state.tasks.append(task_entry)
             self.save_tasks(st.session_state.tasks)
             st.success(f"Task '{task_name}' added successfully!")
+            # Reload tasks to ensure consistency
+            st.session_state.tasks = self.load_tasks()
         else:
             st.error("Task name and category are required.")
 
@@ -64,9 +79,10 @@ class TaskManager:
         if task_found:
             self.save_tasks(st.session_state.tasks)
             st.success(f"Task with ID '{task_id}' deleted.")
+            # Reload tasks to ensure consistency
+            st.session_state.tasks = self.load_tasks()
         else:
             st.error(f"Task with ID '{task_id}' not found.")
-        return task_found
 
     def generate_report(self, timeframe):
         df = pd.DataFrame(st.session_state.tasks)