Add LaTeX and Start Over

app.py

@@ -8,7 +8,7 @@ import streamlit as st
 from streamlit_pills import pills
 from streamlit_feedback import streamlit_feedback
 
-from utils import thumbs_feedback, send_amplitude_data
+from utils import thumbs_feedback, send_amplitude_data, escape_dollars_outside_latex
 
 
 max_examples = 6
@@ -31,6 +31,12 @@ def isTrue(x) -> bool:
     return x.strip().lower() == 'true'
 
 def launch_bot():
+    def reset():
+        st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
+        st.session_state.ex_prompt = None
+        st.session_state.first_turn = True
+
+
     def generate_response(question):
         response = vq.submit_query(question, languages[st.session_state.language])
         return response
@@ -83,9 +89,16 @@ def launch_bot():
         cfg.language = st.selectbox('Language:', languages.keys())
         if st.session_state.language != cfg.language:
             st.session_state.language = cfg.language
-            print(f"DEBUG: Language changed to {st.session_state.language}")
+            reset()
             st.rerun()
 
+        st.markdown("\n")
+        bc1, _ = st.columns([1, 1])
+        with bc1:
+            if st.button('Start Over'):
+                reset()
+                st.rerun()
+
         st.markdown("---")
         st.markdown(
             "## How this works?\n"
@@ -99,7 +112,7 @@ def launch_bot():
     st.markdown(f"<center> <h2> Vectara AI Assistant: {cfg.title} </h2> </center>", unsafe_allow_html=True)
 
     if "messages" not in st.session_state.keys():
-        st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
+        reset()
                 
     # Display chat messages
     for message in st.session_state.messages:
@@ -128,11 +141,13 @@ def launch_bot():
         with st.chat_message("assistant"):
             if cfg.streaming:
                 stream = generate_streaming_response(prompt)
-                response = st.write_stream(stream) 
+                response = st.write_stream(stream)
             else:
                 with st.spinner("Thinking..."):
                     response = generate_response(prompt)
                     st.write(response)
+
+            response = escape_dollars_outside_latex(response)
             message = {"role": "assistant", "content": response}
             st.session_state.messages.append(message)
 

utils.py

@@ -45,3 +45,23 @@ def send_amplitude_data(user_query, chat_response, demo_name, language, feedback
     response = requests.post('https://api2.amplitude.com/2/httpapi', headers=headers, data=json.dumps(data))
     if response.status_code != 200:
         print(f"Amplitude request failed with status code {response.status_code}. Response Text: {response.text}")
+
+def escape_dollars_outside_latex(text):
+    # Define a regex pattern to find LaTeX equations (either single $ or double $$)
+    pattern = re.compile(r'(\$\$.*?\$\$|\$.*?\$)')
+    latex_matches = pattern.findall(text)
+    
+    # Placeholder to temporarily store LaTeX equations
+    placeholders = {}
+    for i, match in enumerate(latex_matches):
+        placeholder = f'__LATEX_PLACEHOLDER_{i}__'
+        placeholders[placeholder] = match
+        text = text.replace(match, placeholder)
+    
+    # Escape dollar signs in the rest of the text
+    text = text.replace('$', '\\$')
+    
+    # Replace placeholders with the original LaTeX equations
+    for placeholder, original in placeholders.items():
+        text = text.replace(placeholder, original)
+    return text