import os

os.environ["KERAS_BACKEND"] = "jax"

import streamlit as st
from vision_models import vision_page



def main():
    # Set up the main layout and title
    st.set_page_config(page_title="ModelLens", layout="centered")
    st.title("ModelLens")

    # Sidebar for navigation
    st.sidebar.title("Navigation")
    options = ["Vision", "NLP", "About"]
    choice = st.sidebar.radio("Go to", options)

    # Route to the selected page
    if choice == "Vision":
        vision_page()
    elif choice == "NLP":
        nlp_page()
    elif choice == "About":
        about_page()


def nlp_page():
    st.header("Natural Language Processing")
    st.write("This section is for exploring NLP models.")
    # Add your NLP model visualization or interaction code here


def about_page():
    st.header("About Page")
    st.write(
        "This app was created to demonstrate a basic Streamlit application layout."
    )


if __name__ == "__main__":
    main()