Update app.py

app.py

@@ -72,29 +72,23 @@ def launch_bot():
     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:**")
+    if len(st.session_state.messages) == 0:  # Only show examples in the first round
         for example in cfg.examples:
-            if st.button(example, key=example):
-                user_input = example
-                st.session_state['user_input'] = user_input
+            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
 
-    # 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
+    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
             
-            # 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()