Update app.py

app.py

@@ -5,9 +5,6 @@ from task_operations import TaskManager
 from task_visualization import TaskVisualizer
 import pandas as pd
 from datetime import datetime
-import os
-
-
 
 def main():
     st.title("Daily Task Tracker")
@@ -19,7 +16,7 @@ def main():
     if 'tasks' not in st.session_state:
         st.session_state.tasks = task_manager.load_tasks()
 
-    # Task input fields for adding new tasks
+    # Task input fields
     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")
@@ -36,13 +33,12 @@ def main():
 
         # Convert session tasks to DataFrame and filter tasks for today
         df = pd.DataFrame(st.session_state.tasks)
-        df['Task Time'] = pd.to_datetime(df['Task Time'])  # Ensure 'Task Time' is in datetime format
-        today_tasks = df[df['Task Time'].dt.date == datetime.today().date()]  # Filter tasks for today
+        df['Task Time'] = pd.to_datetime(df['Task Time'])
+        today_tasks = df[df['Task Time'].dt.date == datetime.today().date()]
 
         if not today_tasks.empty:
-            today_tasks['Task Duration (hours)'] = today_tasks['Task Duration (hours)'].astype(int)
-            today_tasks['Task Duration (minutes)'] = today_tasks['Task Duration (minutes)'].astype(int)
-            st.table(today_tasks[['Task ID', 'Task Name', 'Task Time', 'Task Duration (hours)', 'Task Duration (minutes)', 'Category']])
+            st.table(today_tasks[['Task ID', 'Task Name', 'Task Time', 'Task Duration (hours)', 
+                                  'Task Duration (minutes)', 'Category']])
         else:
             st.write("No tasks for today.")
 
@@ -54,14 +50,13 @@ def main():
             else:
                 st.error(f"Task with ID '{task_id_to_delete}' not found.")
 
-
-
-    # Report options and other visualizations are unchanged
+    # Report options and other visualizations remain 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.")
 
@@ -70,6 +65,7 @@ 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.")
 
@@ -78,6 +74,7 @@ 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.")
 
@@ -86,6 +83,7 @@ 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.")