updated

requirements.txt

@@ -1,9 +1,9 @@
 omegaconf==2.3.0
 python-dotenv==1.0.1
-streamlit==1.39.0
+streamlit==1.41.1
 streamlit_pills==0.3.0
 streamlit_feedback==0.1.3
 uuid==1.30
 langdetect==1.0.9
 langcodes==3.4.0
-vectara-agentic==0.1.19
+vectara-agentic==0.1.21

st_app.py

@@ -9,7 +9,7 @@ from streamlit_feedback import streamlit_feedback
 from utils import thumbs_feedback, escape_dollars_outside_latex, send_amplitude_data
 
 from vectara_agentic.agent import AgentStatusType
-from agent import initialize_agent, get_agent_config
+from agent import initialize_agent, get_agent_config, teaching_styles, languages
 
 initial_prompt = "How can I help you today?"
 
@@ -59,15 +59,15 @@ async def launch_bot():
         st.session_state.ex_prompt = None
         st.session_state.first_turn = True
         st.session_state.show_logs = False
-        if 'agent' not in st.session_state:
-            st.session_state.agent = initialize_agent(cfg, agent_progress_callback=agent_progress_callback)
-        else:
-            st.session_state.agent.clear_memory()
+        st.session_state.agent = initialize_agent(cfg, agent_progress_callback=agent_progress_callback)
 
     if 'cfg' not in st.session_state:
         cfg = get_agent_config()
         st.session_state.cfg = cfg
         st.session_state.ex_prompt = None
+        st.session_state.language = cfg.language
+        st.session_state.student_age = cfg.student_age
+        st.session_state.style = cfg.style
         example_messages = [example.strip() for example in cfg.examples.split(";")] if cfg.examples else []
         st.session_state.example_messages = [em for em in example_messages if len(em)>0]
         reset()
@@ -81,6 +81,28 @@ async def launch_bot():
         st.markdown(f"## {cfg['demo_welcome']}")
         st.markdown(f"{cfg['demo_description']}")
 
+        st.markdown("\n")
+        cfg.style = st.selectbox('Teacher Style:', teaching_styles)
+        if st.session_state.style != cfg.style:
+            st.session_state.style = cfg.style
+            reset()
+            st.rerun()
+
+        cfg.language = st.selectbox('Language:', languages.keys())
+        if st.session_state.language != cfg.language:
+            st.session_state.language = cfg.language
+            reset()
+            st.rerun()
+
+        cfg.student_age = st.number_input(
+            'Student age:',  min_value=13, max_value=99, value=cfg.student_age,
+            step=1, format='%i'
+        )
+        if st.session_state.student_age != cfg.student_age:
+            st.session_state.student_age = cfg.student_age
+            reset()
+            st.rerun()
+
         st.markdown("\n\n")
         bc1, bc2 = st.columns([1, 1])
         with bc1:

utils.py

@@ -31,8 +31,11 @@ def thumbs_feedback(feedback, **kwargs):
 
 def send_amplitude_data(user_query, bot_response, demo_name, feedback=None):
     # Send query and response to Amplitude Analytics
+    amplitude_token = os.environ.get('AMPLITUDE_TOKEN', None)
+    if amplitude_token is None:
+        return
     data = {
-        "api_key": os.getenv('AMPLITUDE_TOKEN'),
+        "api_key": amplitude_token,
         "events": [{
             "device_id": st.session_state.device_id,
             "event_type": "submitted_query",