app.py

import streamlit as st

st.set_page_config(
    page_title="Brain Tumor AI Suite",
    page_icon="🧠",
    layout="wide",
)

st.markdown('''
<style>
    .hero-title {
        font-size: 3rem; font-weight: 800;
        background: linear-gradient(135deg, #667eea, #764ba2, #f093fb);
        -webkit-background-clip: text; -webkit-text-fill-color: transparent;
        margin-bottom: 0;
    }
</style>
''', unsafe_allow_html=True)

st.markdown('<p class="hero-title">🧠 Brain Tumor AI Suite</p>', unsafe_allow_html=True)
st.markdown("**Federated Learning Classification & 3D Segmentation**")
st.markdown("---")

col1, col2 = st.columns(2)

with col1:
    st.markdown("### 📊 Tumor Classification (Federated Learning)")
    st.markdown("Predict the class of a brain tumor (Glioma, Meningioma, Pituitary, No Tumor) using a SimpleCNN model trained across simulated hospitals with Layer-by-Layer QPSO aggregation.")
    st.page_link("pages/1_Classification.py", label="Open Classification App →", icon="📊")

with col2:
    st.markdown("### 🔬 3D Tumor Segmentation")
    st.markdown("View MRI slices and 3D volumetric renderings of brain tumors with segmentation overlays (Whole Tumor, Tumor Core, Enhancing Tumor) predicted by a 3D Attention U-Net.")
    st.page_link("pages/2_Slice_Viewer.py", label="Open Slice Viewer →", icon="🔬")
    st.page_link("pages/3_3D_Visualization.py", label="Open 3D Viewer →", icon="🌐")