Update app.py

app.py

@@ -203,16 +203,17 @@ with col2:
         st.session_state["user_input"] = "How to implement DEI?"
         
 # User input in Streamlit
-user_input = st.text_input("Describe your policy or project to find relevant Lab Lab projects...")
+st.session_state["user_input"] = st.text_input("Describe your policy or project to find relevant Lab Lab projects...")
 
-if st.session_state["user_input"]:
-
-    # Display user message
-    #st.chat_message("user").markdown(st.session_state["user_input"])
+# Display chat history
+#for message in st.session_state.messages:
+#    with st.chat_message(message["role"]):
+#        st.markdown(message["content"])
 
-    grounding = proximity_search(st.session_state["user_input"])
+if st.session_state["user_input"]:
 
     # add the submissions as context (only in prompt, not in history)
+    grounding = proximity_search(st.session_state["user_input"])
     prompt = st.session_state["user_input"] + ". For a project share the image as markdown and mention the url as well. The context for the question: " + grounding;
     messages = st.session_state.messages.copy()
     messages.append({"role": "user", "content": prompt})
@@ -223,5 +224,5 @@ if st.session_state["user_input"]:
         assistant_reply = IBM_chat(messages)
 
     # Display assistant message
-    st.chat_message("assistant").markdown(assistant_reply)
+    #st.chat_message("assistant").markdown(assistant_reply)
     st.session_state.messages.append({"role": "assistant", "content": assistant_reply})