Update app.py

app.py

@@ -2,21 +2,30 @@
 
 import streamlit as st
 from task_operations import TaskManager
-from task_visualization import TaskVisualizer
 import pandas as pd
 from datetime import datetime
+import os
+
+def download_file(filepath):
+    """Allows downloading of the specified file."""
+    with open(filepath, "rb") as f:
+        st.download_button(
+            label="Download Database",
+            data=f,
+            file_name=os.path.basename(filepath),
+            mime="application/octet-stream"
+        )
 
 def main():
     st.title("Daily Task Tracker")
 
     task_manager = TaskManager()
-    visualizer = TaskVisualizer()
 
     # Ensure tasks are loaded from the database into session state
     if 'tasks' not in st.session_state:
         st.session_state.tasks = task_manager.load_tasks()
 
-    # Task input fields
+    # Task input fields for adding new tasks
     task_name = st.text_input("Task Name")
     task_time = st.time_input("Task Time")
     task_duration_hours = st.number_input("Task Duration (Hours)", min_value=0, step=1, format="%d")
@@ -51,13 +60,16 @@ def main():
             else:
                 st.error(f"Task with ID '{task_id_to_delete}' not found.")
 
+    # Add a button to download the task.db file
+    if st.button("Download Database"):
+        download_file("tasks.db")  # Adjust the filename if needed
+
     # Report options and other visualizations are unchanged
     if st.button("Daily Report"):
         report = task_manager.generate_report('daily')
         if not report.empty:
             st.write("Daily Report:")
             st.dataframe(report)
-            visualizer.plot_category_performance('daily', task_manager)
         else:
             st.warning("No tasks for today.")
 
@@ -66,7 +78,6 @@ def main():
         if not report.empty:
             st.write("Weekly Report:")
             st.dataframe(report)
-            visualizer.plot_category_performance('weekly', task_manager)
         else:
             st.warning("No tasks for this week.")
 
@@ -75,7 +86,6 @@ def main():
         if not report.empty:
             st.write("Monthly Report:")
             st.dataframe(report)
-            visualizer.plot_category_performance('monthly', task_manager)
         else:
             st.warning("No tasks for this month.")
 
@@ -84,16 +94,8 @@ def main():
         if not report.empty:
             st.write("Yearly Report:")
             st.dataframe(report)
-            visualizer.plot_category_performance('yearly', task_manager)
         else:
             st.warning("No tasks for this year.")
 
-    # Performance visualizations
-    visualizer.plot_performance()
-    visualizer.plot_overall_category_performance()
-
-    # Option to download the report
-    visualizer.download_report()
-
 if __name__ == "__main__":
     main()