georad commited on
Commit
5cdef11
·
verified ·
1 Parent(s): f925f06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -14
app.py CHANGED
@@ -4,7 +4,7 @@ import uuid
4
  import nest_asyncio
5
  import asyncio
6
  import os
7
- import omegaConf
8
 
9
  torch.classes.__path__ = []
10
 
@@ -43,16 +43,16 @@ def launch_bot():
43
 
44
  if 'cfg' not in st.session_state:
45
  #corpus_keys = str(os.environ["first, last"]).split(',')
46
- cfg = omegaConf.create({
47
  # 'corpus_keys': corpus_keys,
48
  # 'api_key': str(os.environ['api_key']),
49
- 'title': os.environ['title'],
50
  # 'source_data_desc': os.environ['source_data_desc'],
51
  # 'streaming': isTrue(os.environ.get('streaming', False)),
52
  # 'prompt_name': os.environ.get('prompt_name', None),
53
  # 'examples': os.environ.get('examples', None),
54
  # 'language': 'English'
55
- })
56
  #st.session_state.cfg = cfg
57
  #st.session_state.ex_prompt = None
58
  #st.session_state.first_turn = True
@@ -62,9 +62,9 @@ def launch_bot():
62
 
63
  #st.session_state.vq = VectaraQuery(cfg.api_key, cfg.corpus_keys, cfg.prompt_name)
64
 
65
- cfg = st.session_state.cfg
66
  #vq = st.session_state.vq
67
- st.set_page_config(page_title=cfg.title, layout="wide")
68
 
69
  # left side content
70
  with st.sidebar:
@@ -73,11 +73,11 @@ def launch_bot():
73
  st.markdown(f"## About\n\n"
74
  f"This demo uses outside RAG to ask questions about {cfg.source_data_desc}\n")
75
 
76
- cfg.language = st.selectbox('Language:', languages.keys())
77
- if st.session_state.language != cfg.language:
78
- st.session_state.language = cfg.language
79
- reset()
80
- st.rerun()
81
 
82
  st.markdown("\n")
83
  bc1, _ = st.columns([1, 1])
@@ -91,7 +91,7 @@ def launch_bot():
91
  "## Temporary test demo only\n"
92
  )
93
 
94
- st.markdown(f"<center> <h2> Header Demo Test: {cfg.title} </h2> </center>", unsafe_allow_html=True)
95
 
96
  if "messages" not in st.session_state.keys():
97
  reset()
@@ -137,7 +137,7 @@ def launch_bot():
137
  send_amplitude_data(
138
  user_query=st.session_state.messages[-2]["content"],
139
  chat_response=st.session_state.messages[-1]["content"],
140
- demo_name=cfg["title"],
141
  language=st.session_state.language
142
  )
143
  st.rerun()
@@ -146,7 +146,7 @@ def launch_bot():
146
  streamlit_feedback(feedback_type="thumbs", on_submit = thumbs_feedback, key = st.session_state.feedback_key,
147
  kwargs = {"user_query": st.session_state.messages[-2]["content"],
148
  "chat_response": st.session_state.messages[-1]["content"],
149
- "demo_name": cfg["title"],
150
  "response_language": st.session_state.language})
151
 
152
 
 
4
  import nest_asyncio
5
  import asyncio
6
  import os
7
+ #import omegaConf
8
 
9
  torch.classes.__path__ = []
10
 
 
43
 
44
  if 'cfg' not in st.session_state:
45
  #corpus_keys = str(os.environ["first, last"]).split(',')
46
+ cfg = "SBS MMMMAPPPPEEERR" # omegaConf.create({
47
  # 'corpus_keys': corpus_keys,
48
  # 'api_key': str(os.environ['api_key']),
49
+ # 'title': os.environ['title'],
50
  # 'source_data_desc': os.environ['source_data_desc'],
51
  # 'streaming': isTrue(os.environ.get('streaming', False)),
52
  # 'prompt_name': os.environ.get('prompt_name', None),
53
  # 'examples': os.environ.get('examples', None),
54
  # 'language': 'English'
55
+ #})
56
  #st.session_state.cfg = cfg
57
  #st.session_state.ex_prompt = None
58
  #st.session_state.first_turn = True
 
62
 
63
  #st.session_state.vq = VectaraQuery(cfg.api_key, cfg.corpus_keys, cfg.prompt_name)
64
 
65
+ #cfg = st.session_state.cfg
66
  #vq = st.session_state.vq
67
+ st.set_page_config(page_title=cfg, layout="wide")
68
 
69
  # left side content
70
  with st.sidebar:
 
73
  st.markdown(f"## About\n\n"
74
  f"This demo uses outside RAG to ask questions about {cfg.source_data_desc}\n")
75
 
76
+ #cfg.language = st.selectbox('Language:', languages.keys())
77
+ #if st.session_state.language != cfg.language:
78
+ # st.session_state.language = cfg.language
79
+ # reset()
80
+ # st.rerun()
81
 
82
  st.markdown("\n")
83
  bc1, _ = st.columns([1, 1])
 
91
  "## Temporary test demo only\n"
92
  )
93
 
94
+ st.markdown(f"<center> <h2> Header Demo Test: {cfg} </h2> </center>", unsafe_allow_html=True)
95
 
96
  if "messages" not in st.session_state.keys():
97
  reset()
 
137
  send_amplitude_data(
138
  user_query=st.session_state.messages[-2]["content"],
139
  chat_response=st.session_state.messages[-1]["content"],
140
+ demo_name=cfg #["title"],
141
  language=st.session_state.language
142
  )
143
  st.rerun()
 
146
  streamlit_feedback(feedback_type="thumbs", on_submit = thumbs_feedback, key = st.session_state.feedback_key,
147
  kwargs = {"user_query": st.session_state.messages[-2]["content"],
148
  "chat_response": st.session_state.messages[-1]["content"],
149
+ "demo_name": cfg #["title"],
150
  "response_language": st.session_state.language})
151
 
152