import streamlit as st

st.set_page_config(page_title="Gemma LLM Fine-Tuning UI", layout="wide")

st.title("Gemma LLM Fine-Tuning Suite 🚀")
st.markdown("""
### 🔥 **Multi-page AI Model Trainer**
- **Chat**: Interact with the model.
- **Fine-tuning**: Train on `train_data.csv` or upload new datasets.
- **Conversion**: Export models to TorchScript and ONNX.
- **Dataset Management**: View and add to your training data.
""")

# st.sidebar.title("Navigation")
# st.sidebar.page_link("pages/Chat.py", label="🔹 Chat")
# st.sidebar.page_link("pages/Finetune.py", label="🔹 Fine-tuning")
# st.sidebar.page_link("pages/Conversion.py", label="🔹 Model Conversion")
# st.sidebar.page_link("pages/Dataset_Management.py", label="🔹 Dataset Management")