Update app.py

app.py

@@ -93,16 +93,9 @@ with st.sidebar:
 user_prompt = st.chat_input("Ask me anything about the content of the PDF:")
 if user_prompt:
     st.session_state.messages.append({'role': 'user', "content": user_prompt})
-    for message in st.session_state.messages:
-        with st.chat_message(message['role']):
-            st.write(message['content'])
     response = handle_query(user_prompt)
-    if st.session_state.messages[-1]["role"] != "assistant":
-        with st.chat_message("assistant"):
-            with st.spinner("Thinking..."):
-                st.write(response)
     st.session_state.messages.append({'role': 'assistant', "content": response})
 
-# for message in st.session_state.messages:
-#     with st.chat_message(message['role']):
-#         st.write(message['content'])
+for message in st.session_state.messages:
+    with st.chat_message(message['role']):
+        st.write(message['content'])