import streamlit as st
from tasks.symbol_match import run_symbol_match
from tasks.trail_making import run_trail_making
from tasks.cpt import run_cpt
from drills.emoji_label import run_emoji_label
from drills.body_scan import run_body_scan
from drills.gratitude import run_gratitude
from personality.big_five import run_big_five
from reports.export_pdf import generate_report

st.set_page_config(page_title="MindMentor", layout="centered")
st.title("🧠 MindMentor - Brain & EQ Coach")

menu = st.sidebar.selectbox("Select a module", [
    "Cognitive Task: Symbol Match",
    "Cognitive Task: Trail Making",
    "Cognitive Task: CPT Task",
    "EI Drill: Emoji Labeling",
    "EI Drill: Body Scan",
    "EI Drill: Gratitude Journal",
    "Personality: Big Five",
    "🧾 Generate Report"
])

if menu == "Cognitive Task: Symbol Match":
    run_symbol_match()
elif menu == "Cognitive Task: Trail Making":
    run_trail_making()
elif menu == "Cognitive Task: CPT Task":
    run_cpt()
elif menu == "EI Drill: Emoji Labeling":
    run_emoji_label()
elif menu == "EI Drill: Body Scan":
    run_body_scan()
elif menu == "EI Drill: Gratitude Journal":
    run_gratitude()
elif menu == "Personality: Big Five":
    run_big_five()
elif menu == "🧾 Generate Report":
    generate_report()