Update app.py

app.py

@@ -128,6 +128,6 @@ if prompt := st.chat_input("Skriv din fråga..."):
     response = f"{generated_text}"
     # Display assistant response in chat message container
     with st.chat_message("assistant"):
-        st.markdown(f"```{prompt}```\n" + response)
+        st.markdown(f"```{query}```\n" + response)
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})