Update app.py

app.py

@@ -7,21 +7,11 @@ import os
 import streamlit as st
 from PIL import Image
 
-def generate_response(question):
-    response = st.session_state.vq.submit_query(question)
-    return response
-
-def process_prompt(prompt):
-    # Your logic to process the prompt and generate response
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    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)
 
 def launch_bot():
+    def generate_response(question):
+        response = vq.submit_query(question)
+        return response
 
     if 'cfg' not in st.session_state:
         corpus_ids = str(os.environ['corpus_ids']).split(',')
@@ -39,13 +29,14 @@ def launch_bot():
         st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids)
 
     cfg = st.session_state.cfg
+    vq = st.session_state.vq
     st.set_page_config(page_title=cfg.title, layout="wide")
 
     # left side content
     with st.sidebar:
         image = Image.open('Vectara-logo.png')
         st.markdown(f"## Welcome to {cfg.title}\n\n"
-                    f"With this demo uses Retrieval Augmented Generation to ask questions about {cfg.source_data_desc}\n\n")
+                    f"This demo uses Retrieval Augmented Generation to ask questions about {cfg.source_data_desc}\n\n")
 
         st.markdown("---")
         st.markdown(
@@ -62,33 +53,26 @@ def launch_bot():
 
     if "messages" not in st.session_state.keys():
         st.session_state.messages = [{"role": "assistant", "content": "How may I help you?"}]
-    
+
     # Display chat messages
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.write(message["content"])
 
-     # Always allow typing in the chat input box
-    user_input = st.text_input("Type your question here...", key="user_input")
-
-    # Display example questions only during the first round
-    if len(st.session_state.messages) == 0:  # Only show examples in the first round
-        for example in cfg.examples:
-            if st.button(example):
-                user_input = example  # Directly use the example as input
-                # Process the input immediately
-                response = generate_response(user_input)
-                st.session_state.messages.append(user_input)
-                st.session_state.messages.append(response)
-                st.experimental_rerun()  # Rerun to refresh and show the response
-
-    if st.button("Submit"):
-        if user_input:  # Ensure there's something to process
-            response = generate_response(user_input)
-            st.session_state.messages.append(user_input)
-            st.session_state.messages.append(response)
-            # Clear the input (workaround as direct clearing isn't supported)
-            st.experimental_rerun()  # Rerun to refresh and clear the input box
-            
+    # 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()
+    launch_bot()