Update app.py

app.py

@@ -1,66 +1,66 @@
-import streamlit as st
-from components.task_operations import TaskManager
-from components.task_visualization import TaskVisualizer
-import pandas as pd
-
-def main():
-    st.title("Jony Daily Task Tracker")
-
-    task_manager = TaskManager()
-    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"):
-        task_manager.add_task(task_name, task_time, task_duration_hours, task_duration_minutes, task_category)
-        st.success(f"Task '{task_name}' added!")
-
-    if st.session_state.tasks:
-        st.write("Today's Tasks:")
-        df = pd.DataFrame(st.session_state.tasks)
-        df['Task Duration (hours)'] = df['Task Duration (hours)'].astype(int)
-        df['Task Duration (minutes)'] = df['Task Duration (minutes)'].astype(int)
-        st.table(df[['Task Name', 'Task Time', 'Task Duration (hours)', 'Task Duration (minutes)', 'Category']])
-
-        task_to_delete = st.text_input("Enter Task Name to Delete")
-        if st.button("Delete Task"):
-            if task_manager.delete_task_by_name(task_to_delete):
-                st.success(f"Task '{task_to_delete}' deleted!")
-            else:
-                st.error(f"Task '{task_to_delete}' not found.")
-
-    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 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 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 st.button("Yearly Report"):
-        report = task_manager.generate_report('yearly')
-        st.write("Yearly Report:")
-        st.dataframe(report)
-        visualizer.plot_category_performance('yearly', task_manager)
-
-    visualizer.plot_performance()
-    visualizer.plot_overall_category_performance()
-    visualizer.download_report()
-
-if __name__ == "__main__":
-    main()
+import streamlit as st
+from task_operations import TaskManager
+from task_visualization import TaskVisualizer
+import pandas as pd
+
+def main():
+    st.title("Jony Daily Task Tracker")
+
+    task_manager = TaskManager()
+    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"):
+        task_manager.add_task(task_name, task_time, task_duration_hours, task_duration_minutes, task_category)
+        st.success(f"Task '{task_name}' added!")
+
+    if st.session_state.tasks:
+        st.write("Today's Tasks:")
+        df = pd.DataFrame(st.session_state.tasks)
+        df['Task Duration (hours)'] = df['Task Duration (hours)'].astype(int)
+        df['Task Duration (minutes)'] = df['Task Duration (minutes)'].astype(int)
+        st.table(df[['Task Name', 'Task Time', 'Task Duration (hours)', 'Task Duration (minutes)', 'Category']])
+
+        task_to_delete = st.text_input("Enter Task Name to Delete")
+        if st.button("Delete Task"):
+            if task_manager.delete_task_by_name(task_to_delete):
+                st.success(f"Task '{task_to_delete}' deleted!")
+            else:
+                st.error(f"Task '{task_to_delete}' not found.")
+
+    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 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 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 st.button("Yearly Report"):
+        report = task_manager.generate_report('yearly')
+        st.write("Yearly Report:")
+        st.dataframe(report)
+        visualizer.plot_category_performance('yearly', task_manager)
+
+    visualizer.plot_performance()
+    visualizer.plot_overall_category_performance()
+    visualizer.download_report()
+
+if __name__ == "__main__":
+    main()