# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
import streamlit as st


def main():
    # Evaluation pages
    application_evaluation_page = st.Page(
        "page/evaluations/app_eval.py",
        title="Application Evaluation",
        icon="🦙",
        default=False,
    )

    # Playground pages
    chat_page = st.Page(
        "page/playground/chat.py", title="Chat", icon="💬", default=True
    )
    rag_page = st.Page("page/playground/rag.py", title="RAG", icon="💬", default=False)

    # Distribution pages
    distribution_page = st.Page(
        "page/distribution/distro.py", title="Distribution", icon="🔍", default=False
    )

    pg = st.navigation(
        {
            "Evaluations": [application_evaluation_page],
            "Playground": [chat_page, rag_page],
            "Distribution": [distribution_page],
        }
    )
    pg.run()


if __name__ == "__main__":
    main()