# app.py

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')
        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')
        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')
        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')
        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()
    visualizer.download_report()

if __name__ == "__main__":
    main()