Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import streamlit as st
 from task_operations import TaskManager
 from task_visualization import TaskVisualizer
@@ -36,27 +38,39 @@ def main():
 
     if st.button("Daily Report"):
         report = task_manager.generate_report('daily')
-        st.write("Daily Report:")
-        st.dataframe(report)
-        visualizer.plot_category_performance('daily', task_manager)
+        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)
-        visualizer.plot_category_performance('weekly', task_manager)
+        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)
-        visualizer.plot_category_performance('monthly', task_manager)
+        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)
-        visualizer.plot_category_performance('yearly', task_manager)
+        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.")
 
     visualizer.plot_performance()
     visualizer.plot_overall_category_performance()