Update src/streamlit_app.py

src/streamlit_app.py

@@ -123,6 +123,7 @@ if prompt := st.chat_input("Ask your research question..."):
     st.chat_message("user").write(prompt)
     with st.chat_message("assistant"):
         st_callback = StreamlitCallbackHandler(st.container())
-        response = agent.invoke({"messages": [{"role": "user", "content": "What is langgraph?"}]},{"callbacks": [st_callback]})
-        output = response.get("output", str(response))
-        st.write(output)
+        response = agent.invoke({"messages": [{"role": "user", "content": prompt}]},{"callbacks": [st_callback]})
+        displayed_response = response["messages"][-1].content
+        #output = response.get("output", str(response))
+        st.write(displayed_response)