llama-stack-mirror/llama_stack/distribution/ui/app.py
2024-11-27 16:03:58 -08:00

40 lines
1 KiB
Python

# 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()