# app.py

import streamlit as st
from task_operations import TaskManager
from task_visualization import TaskVisualizer
import pandas as pd
from datetime import datetime

def main():
    st.title("Daily Task Tracker")

    task_manager = TaskManager()
    visualizer = TaskVisualizer()

    # Ensure tasks are loaded from the database into session state
    if 'tasks' not in st.session_state:
        st.session_state.tasks = task_manager.load_tasks()

    # Task input fields
    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!")

    # Display tasks for today only
    if st.session_state.tasks:
        st.write("Today's Tasks:")

        # Convert session tasks to DataFrame and filter tasks for today
        df = pd.DataFrame(st.session_state.tasks)
        df['Task Time'] = pd.to_datetime(df['Task Time'])  # Ensure 'Task Time' is in datetime format
        today_tasks = df[df['Task Time'].dt.date == datetime.today().date()]  # Filter tasks for today

        if not today_tasks.empty:
            today_tasks['Task Duration (hours)'] = today_tasks['Task Duration (hours)'].astype(int)
            today_tasks['Task Duration (minutes)'] = today_tasks['Task Duration (minutes)'].astype(int)
            st.table(today_tasks[['Task ID', 'Task Name', 'Task Time', 'Task Duration (hours)', 'Task Duration (minutes)', 'Category']])
        else:
            st.write("No tasks for today.")

        # Task deletion option by ID
        task_id_to_delete = st.number_input("Enter Task ID to Delete", min_value=0, step=1)
        if st.button("Delete Task by ID"):
            if task_manager.delete_task_by_id(task_id_to_delete):
                st.success(f"Task with ID '{task_id_to_delete}' deleted!")
            else:
                st.error(f"Task with ID '{task_id_to_delete}' not found.")

    # Report options and other visualizations are unchanged
    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.")

    # Performance visualizations
    visualizer.plot_performance()
    visualizer.plot_overall_category_performance()

    # Option to download the report
    visualizer.download_report()

if __name__ == "__main__":
    main()