Update app.py

app.py

@@ -97,19 +97,22 @@ with st.sidebar:
             st.success("Done")
 
 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})
-    # response = handle_query(user_prompt)
-    # st.session_state.messages.append({'role': 'assistant', "content": response})
 
 if user_prompt and uploaded_file:
     st.session_state.messages.append({'role': 'user', "content": user_prompt})
     with st.chat_message("user", avatar="human-logo.jpeg"):
         st.write(user_prompt)
 
-if st.session_state.messages[-1]["role"] != "assistant":
-    with st.chat_message("assistant", avatar="👽"):
+    # Trigger assistant's response retrieval and update UI
+    with st.spinner("Assistant is typing..."):
         response = handle_query(user_prompt)
-        full_response = st.write(response)
-        message = {"role": "assistant", "content": full_response}
-        st.session_state.messages.append(message)
+        st.write(response)
+        st.session_state.messages.append({'role': 'assistant', "content": response})
+        
+
+# if st.session_state.messages[-1]["role"] != "assistant":
+#     with st.chat_message("assistant", avatar="👽"):
+#         response = handle_query(user_prompt)
+#         full_response = st.write(response)
+#         message = {"role": "assistant", "content": full_response}
+#         st.session_state.messages.append(message)