Spaces:
Running
Running
File size: 2,295 Bytes
78cdc34 cf9f618 78cdc34 cf9f618 221dfe3 78cdc34 cf9f618 d9ed521 cf9f618 3eda6d3 78cdc34 cf9f618 78cdc34 3b3aaa9 9361457 3b3aaa9 78cdc34 3b3aaa9 9361457 3b3aaa9 78cdc34 3b3aaa9 78cdc34 9361457 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
from collections import defaultdict
import streamlit as st
from mlip_arena.tasks import REGISTRY as TASKS
# if "logged_in" not in st.session_state:
# st.session_state.logged_in = False
# def login():
# if st.button("Log in"):
# st.session_state.logged_in = True
# st.rerun()
# def logout():
# if st.button("Log out"):
# st.session_state.logged_in = False
# st.rerun()
# login_page = st.Page(login, title="Log in", icon=":material/login:")
# logout_page = st.Page(logout, title="Log out", icon=":material/logout:")
leaderboard = st.Page(
"leaderboard.py", title="Leaderboard", icon=":material/trophy:", default=True
)
search = st.Page("tools/search.py", title="Search", icon=":material/search:")
history = st.Page("tools/history.py", title="History", icon=":material/history:")
ptable = st.Page("tools/ptable.py", title="Periodic table", icon=":material/gradient:")
nav = defaultdict(list)
nav[""].append(leaderboard)
wide_pages, centered_pages = [], []
for task in TASKS:
page = st.Page(
f"tasks/{TASKS[task]['task-page']}.py", title=task, icon=":material/target:"
)
nav[TASKS[task]["category"]].append(page)
if TASKS[task]["task-layout"] == "wide":
wide_pages.append(page)
else:
centered_pages.append(page)
pg = st.navigation(nav)
if pg in centered_pages:
st.set_page_config(
layout="centered",
page_title="MLIP Arena",
page_icon=":shark:",
initial_sidebar_state="expanded",
menu_items={
"About": "https://github.com/atomind-ai/mlip-arena",
"Report a bug": "https://github.com/atomind-ai/mlip-arena/issues/new",
},
)
else:
st.set_page_config(
layout="wide",
page_title="MLIP Arena",
page_icon=":shark:",
initial_sidebar_state="expanded",
menu_items={
"About": "https://github.com/atomind-ai/mlip-arena",
"Report a bug": "https://github.com/atomind-ai/mlip-arena/issues/new",
},
)
st.toast(
"MLIP Arena is currently in **pre-alpha**. The results are not stable. Please interpret them with care. Contributions are welcome. For more information, visit https://github.com/atomind-ai/mlip-arena.",
icon="🍞",
)
pg.run()
|