Update app.py

app.py

@@ -1,21 +1,20 @@
-# app.py
-
 import streamlit as st
-from task_operations import TaskManager
-from task_visualization import TaskVisualizer
 import pandas as pd
+from components.task_operations import TaskManager
+from components.task_visualization import TaskVisualizer
+
 
 def main():
-    st.title("Jony Daily Task Tracker")
+    
+    st.title("Daily Task Tracker")
 
     task_manager = TaskManager()
-    visualizer = TaskVisualizer()
+    task_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"):
@@ -38,43 +37,31 @@ def main():
 
     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.")
+        st.write("Daily Report:")
+        st.dataframe(report)
+        task_visualizer.plot_category_performance('daily')
 
     if st.button("Weekly Report"):
         report = task_manager.generate_report('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.")
+        st.write("Weekly Report:")
+        st.dataframe(report)
+        task_visualizer.plot_category_performance('weekly')
 
     if st.button("Monthly Report"):
         report = task_manager.generate_report('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.")
+        st.write("Monthly Report:")
+        st.dataframe(report)
+        task_visualizer.plot_category_performance('monthly')
 
     if st.button("Yearly Report"):
         report = task_manager.generate_report('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.")
+        st.write("Yearly Report:")
+        st.dataframe(report)
+        task_visualizer.plot_category_performance('yearly')
 
-    visualizer.plot_performance()
-    visualizer.plot_overall_category_performance()
-    visualizer.download_report()
+    task_visualizer.plot_performance()
+    task_visualizer.plot_overall_category_performance()
+    task_manager.download_report()
 
 if __name__ == "__main__":
     main()