import streamlit as st

# Set page config
st.set_page_config(page_title="Innomatics Online Trainer Bot", layout="centered")

# Inject custom CSS for background and buttons
st.markdown("""
    <style>
    .main {
        background-color: #8B0000;
        padding: 20px;
    }
    .stButton>button {
        background-color: #B76F27;
        color: #8B0000;
        border: 2px solid #8B0000;
        border-radius: 10px;
        padding: 10px 20px;
        font-size: 18px;
        font-weight: bold;
        width: 100%;
        transition: 0.3s ease-in-out;
    }
    .stButton>button:hover {
        background-color: #8B0000;
        color: #ffffff;
        border: 2px solid white;
    }
    h1, h3, p {
        color: white;
    }
    </style>
""", unsafe_allow_html=True)

# Title and intro
st.title("Innomatics Online Trainer Bot")
st.markdown("### 👋 Welcome to the Innomatics Online Trainer Bot!")
st.markdown("This dashboard will guide you through your doubts in various modules.")
st.markdown("## In which module do you have doubt?")

# Button layout - 2 per row
col1, col2 = st.columns(2)
with col1:
    if st.button("Python"):
        st.switch_page("pages/python.py")
with col2:
    if st.button("Machine Learning"):
        st.switch_page("pages/machine_learning.py")

col3, col4 = st.columns(2)
with col3:
    if st.button("Deep Learning"):
        st.switch_page("pages/deep_learning.py")
with col4:
    if st.button("Statistics"):
        st.switch_page("pages/statistics.py")

col5, col6 = st.columns(2)
with col5:
    if st.button("Excel"):
        st.switch_page("pages/excel.py")
with col6:
    if st.button("SQL"):
        st.switch_page("pages/sql.py")