Update app.py

app.py

@@ -1,20 +1,21 @@
+# app.py
+
 import streamlit as st
-import pandas as pd
 from task_operations import TaskManager
 from task_visualization import TaskVisualizer
-
+import pandas as pd
 
 def main():
-    
-    st.title("Daily Task Tracker")
+    st.title("Jony Daily Task Tracker")
 
     task_manager = TaskManager()
-    task_visualizer = TaskVisualizer()
+    visualizer = TaskVisualizer()
 
     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")
     task_duration_minutes = st.number_input("Task Duration (Minutes)", min_value=0, max_value=59, step=1, format="%d")
+
     task_category = st.selectbox("Task Category", TaskManager.CATEGORIES)
 
     if st.button("Add Task"):
@@ -37,31 +38,43 @@ def main():
 
     if st.button("Daily Report"):
         report = task_manager.generate_report('daily')
-        st.write("Daily Report:")
-        st.dataframe(report)
-        task_visualizer.plot_category_performance('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.")
 
     if st.button("Weekly Report"):
         report = task_manager.generate_report('weekly')
-        st.write("Weekly Report:")
-        st.dataframe(report)
-        task_visualizer.plot_category_performance('weekly')
+        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.")
 
     if st.button("Monthly Report"):
         report = task_manager.generate_report('monthly')
-        st.write("Monthly Report:")
-        st.dataframe(report)
-        task_visualizer.plot_category_performance('monthly')
+        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.")
 
     if st.button("Yearly Report"):
         report = task_manager.generate_report('yearly')
-        st.write("Yearly Report:")
-        st.dataframe(report)
-        task_visualizer.plot_category_performance('yearly')
+        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.")
 
-    task_visualizer.plot_performance()
-    task_visualizer.plot_overall_category_performance()
-    task_manager.download_report()
+    visualizer.plot_performance()
+    visualizer.plot_overall_category_performance()
+    visualizer.download_report()
 
 if __name__ == "__main__":
     main()