forked from stanford-oval/storm
-
Notifications
You must be signed in to change notification settings - Fork 0
/
storm.py
64 lines (52 loc) · 2.49 KB
/
storm.py
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
import os
import sys
script_dir = os.path.dirname(os.path.abspath(__file__))
wiki_root_dir = os.path.dirname(os.path.dirname(script_dir))
sys.path.append(os.path.normpath(os.path.join(script_dir, '../../src/storm_wiki')))
sys.path.append(os.path.normpath(os.path.join(script_dir, '../../src')))
import demo_util
from pages_util import MyArticles, CreateNewArticle
from streamlit_float import *
from streamlit_option_menu import option_menu
def main():
global database
st.set_page_config(layout='wide')
if "first_run" not in st.session_state:
st.session_state['first_run'] = True
# set api keys from secrets
if st.session_state['first_run']:
for key, value in st.secrets.items():
if type(value) == str:
os.environ[key] = value
# initialize session_state
if "selected_article_index" not in st.session_state:
st.session_state["selected_article_index"] = 0
if "selected_page" not in st.session_state:
st.session_state["selected_page"] = 0
if st.session_state.get("rerun_requested", False):
st.session_state["rerun_requested"] = False
st.rerun()
st.write('<style>div.block-container{padding-top:2rem;}</style>', unsafe_allow_html=True)
menu_container = st.container()
with menu_container:
pages = ["My Articles", "Create New Article"]
menu_selection = option_menu(None, pages,
icons=['house', 'search'],
menu_icon="cast", default_index=0, orientation="horizontal",
manual_select=st.session_state.selected_page,
styles={
"container": {"padding": "0.2rem 0", "background-color": "#22222200"},
},
key='menu_selection')
if st.session_state.get("manual_selection_override", False):
menu_selection = pages[st.session_state["selected_page"]]
st.session_state["manual_selection_override"] = False
st.session_state["selected_page"] = None
if menu_selection == "My Articles":
demo_util.clear_other_page_session_state(page_index=2)
MyArticles.my_articles_page()
elif menu_selection == "Create New Article":
demo_util.clear_other_page_session_state(page_index=3)
CreateNewArticle.create_new_article_page()
if __name__ == "__main__":
main()