Update app.py

app.py

@@ -6,13 +6,9 @@ import os
 
 import streamlit as st
 from PIL import Image
-from functools import partial
-
-def set_query(q: str):
-    st.session_state['query'] = q
 
 def launch_bot():
-    def get_answer(question):
+    def generate_response(question):
         response = vq.submit_query(question)
         return response
 
@@ -41,36 +37,36 @@ def launch_bot():
             "## How this works?\n"
             "This app was built with [Vectara](https://vectara.com).\n"
             "Vectara's [Indexing API](https://docs.vectara.com/docs/api-reference/indexing-apis/indexing) was used to ingest the data into a Vectara corpus (or index).\n\n"
-            "This app uses Vectara API to query the corpus and present the results to you, answering your question.\n\n"
+            "This app uses Vectara Chat API to query the corpus and present the results to you, answering your question.\n\n"
         )
         st.markdown("---")
         st.image(image, width=250)
 
-    st.markdown(f"<center> <h2> Vectara demo app: {cfg.title} </h2> </center>", unsafe_allow_html=True)
+    st.markdown(f"<center> <h2> Vectara chat demo: {cfg.title} </h2> </center>", unsafe_allow_html=True)
     st.markdown(f"<center> <h4> {cfg.description} <h4> </center>", unsafe_allow_html=True)
 
-    # Setup a split column layout
-    main_col, questions_col = st.columns([4, 2], gap="medium")
-    with main_col:        
-        cols = st.columns([1, 8], gap="small")
-        cols[0].markdown("""<h5>Search</h5>""", unsafe_allow_html=True)             
-        cols[1].text_input(label="search", key='query', max_chars=256, label_visibility='collapsed', help="Enter your question here")
-
-        st.markdown("<h5>Response</h5>", unsafe_allow_html=True)
-        response_text = st.empty()
-        response_text.text_area(f" ", placeholder="The answer will appear here.", disabled=True,
-                                key="response", height=1, label_visibility='collapsed')
-    with questions_col:
-        st.markdown("<h5 style='text-align:center; color: red'> Sample questions </h5>", unsafe_allow_html=True)
-        for q in list(cfg.examples):
-            st.button(q, on_click=partial(set_query, q), use_container_width=True)
-
+    if "messages" not in st.session_state.keys():
+        st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
 
-    # run the main flow
-    if st.session_state.get('query'):
-        query = st.session_state['query']
-        response = get_answer(query)
-        response_text.markdown(response)
+    # Display chat messages
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.write(message["content"])
 
+    # User-provided prompt
+    if prompt := st.chat_input():
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.write(prompt)
+    
+    # Generate a new response if last message is not from assistant
+    if st.session_state.messages[-1]["role"] != "assistant":
+        with st.chat_message("assistant"):
+            with st.spinner("Thinking..."):
+                response = generate_response(prompt) 
+                st.write(response) 
+        message = {"role": "assistant", "content": response}
+        st.session_state.messages.append(message)
+    
 if __name__ == "__main__":
     launch_bot()