Update app.py

app.py

@@ -7,11 +7,21 @@ import os
 import streamlit as st
 from PIL import Image
 
+def generate_response(question):
+    response = 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(',')
@@ -27,7 +37,6 @@ def launch_bot():
         })
         st.session_state.cfg = cfg
         st.session_state.vq = VectaraQuery(cfg.api_key, cfg.customer_id, cfg.corpus_ids)
-        st.session_state.first_round = True
 
     cfg = st.session_state.cfg
     vq = st.session_state.vq
@@ -60,26 +69,33 @@ def launch_bot():
         with st.chat_message(message["role"]):
             st.write(message["content"])
 
-    if st.session_state.first_round:
-        st.markdown("**Example questions:**")
+     # 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) <= 1:  # Adjust based on when you want examples to disappear
+        st.markdown("**Or choose an example question:**")
         for example in cfg.examples:
-            if st.button(example):
-                prompt = example
-                st.session_state.first_round = False
-                process_prompt(prompt)
-    else:
-        if prompt := st.chat_input("Ask me anything..."):
-            process_prompt(prompt)  # Process the user-provided prompt
+            if st.button(example, key=example):
+                user_input = example
+                st.session_state['user_input'] = user_input
+
+    # Check if there's input to process (either typed or selected from examples)
+    if user_input:
+        # This block prevents immediate processing when the page loads. Adjust as necessary.
+        if "processed" not in st.session_state or not st.session_state["processed"]:
+            st.session_state.messages.append({"role": "user", "content": user_input})
+            with st.chat_message("assistant"):
+                with st.spinner("Thinking..."):
+                    response = generate_response(user_input) 
+                    st.write(response) 
+            st.session_state.messages.append({"role": "assistant", "content": response})
+            st.session_state["processed"] = True  # Prevent re-processing the same input on refresh
             
-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)
-    
+            # Clear the input for the next message
+            st.session_state['user_input'] = ''
+            st.experimental_rerun()  # Rerun the app to clear the input box
+            
+                
 if __name__ == "__main__":
     launch_bot()