Update task_visualization.py

task_visualization.py

@@ -1,7 +1,6 @@
-import streamlit as st
-import pandas as pd
 import matplotlib.pyplot as plt
-from task_operations import TaskManager  # Import TaskManager
+import pandas as pd
+import streamlit as st
 
 class TaskVisualizer:
     def plot_performance(self):
@@ -18,8 +17,7 @@ class TaskVisualizer:
         plt.tight_layout()
         st.pyplot(plt)
 
-    def plot_category_performance(self, timeframe):
-        task_manager = TaskManager()  # Use the imported TaskManager class
+    def plot_category_performance(self, timeframe, task_manager):
         report = task_manager.generate_report(timeframe)
         if not report.empty:
             category_times = report.groupby('Category')['Total Duration'].sum()
@@ -47,3 +45,8 @@ class TaskVisualizer:
         plt.xticks(rotation=45)
         plt.tight_layout()
         st.pyplot(plt)
+
+    def download_report(self):
+        df = pd.DataFrame(st.session_state.tasks)
+        csv = df.to_csv(index=False)
+        st.download_button("Download CSV", data=csv, file_name="task_report.csv", mime='text/csv')