import streamlit as st

# Set page configuration
st.set_page_config(page_title="🤖 Innomatics Online Trainer Bot", page_icon="📚", layout="centered")

# Custom CSS styling
st.markdown("""
    <style>
    .main {
        background: linear-gradient(135deg, #2c3e50 0%, #3498db 100%);
        padding: 2rem;
        font-family: 'Segoe UI', sans-serif;
        color: #ffffff !important;
    }
    h1, h2, h3, h4, p, li {
        color: #ffffff !important;
        text-align: center !important;
    }
    .stButton>button {
        background: rgba(255, 255, 255, 0.1);
        border: 2px solid rgba(255, 255, 255, 0.3);
        color: #ffffff;
        font-size: 18px;
        font-weight: 600;
        padding: 0.8em 1.2em;
        border-radius: 12px;
        width: 100%;
        transition: 0.3s ease;
        box-shadow: 0 4px 10px rgba(0, 0, 0, 0.25);
    }
    .stButton>button:hover {
        background: rgba(255, 255, 255, 0.3);
        border-color: #fff;
        color: #ffffff;
    }
    hr {
        border: 1px solid rgba(255, 255, 255, 0.4);
        margin: 2em 0;
    }
    </style>
""", unsafe_allow_html=True)

# Header content
st.title("🤖 Innomatics Online Trainer Bot")
st.markdown("### 👋 Welcome to your personalized learning companion!")
st.markdown("This smart assistant helps you navigate your doubts in **Data Science**, **AI**, and related fields.")

st.markdown("## 📚 Choose a module where you need help:")

# Grid of subject buttons with emojis
subjects = [
    ("🐍 Python", "pages/1_Python.py"),
    ("🤖 Machine Learning", "pages/machine_learning.py"),
    ("🧠 Deep Learning", "pages/deep_learning.py"),
    ("📊 Statistics", "pages/statistics.py"),
    ("🪄 GenAI", "pages/gen_ai.py"),
    ("🗄️ SQL", "pages/sql.py")
]

# Render buttons in a grid (2 per row)
for i in range(0, len(subjects), 2):
    cols = st.columns(2)
    for j in range(2):
        if i + j < len(subjects):
            with cols[j]:
                if st.button(subjects[i + j][0]):
                    st.switch_page(subjects[i + j][1])

# Footer note
st.markdown("---")
st.markdown("🧠 *Empowering your journey through every concept, one question at a time.*")