Update app.py

app.py

@@ -90,7 +90,7 @@ st.subheader('Upload a PDF and ask questions about its content!')
 
 # Display the welcome prompt if chat history is only the initial prompt
 if len(st.session_state.chat_history) == 1:
-    with st.chat_message(SPEAKER_TYPES.BOT, avatar="🔍"):
+    with st.chat_message(SPEAKER_TYPES["BOT"], avatar="🔍"):
         st.write(initial_prompt['content'])
 
 # Get user input
@@ -111,20 +111,20 @@ def get_rag_response(prompt):
 # Handle the user prompt and generate response
 if prompt:
     # Add user prompt to chat history
-    st.session_state.chat_history.append({'role': SPEAKER_TYPES.USER, 'content': prompt})
+    st.session_state.chat_history.append({'role': SPEAKER_TYPES["USER"], 'content': prompt})
     
     # Display chat messages from the chat history
     for message in st.session_state.chat_history[1:]:
-        with st.chat_message(message["role"], avatar="👤" if message['role'] == SPEAKER_TYPES.USER else "🔍"):
+        with st.chat_message(message["role"], avatar="👤" if message['role'] == SPEAKER_TYPES["USER"] else "🔍"):
             st.write(message["content"])
     
     # Get the response using the RAG chain
     with st.spinner(text='Generating response...'):
         response_text = get_rag_response(prompt)
-        st.session_state.chat_history.append({'role': SPEAKER_TYPES.BOT, 'content': response_text})
+        st.session_state.chat_history.append({'role': SPEAKER_TYPES["BOT"], 'content': response_text})
     
     # Display the bot response
-    with st.chat_message(SPEAKER_TYPES.BOT, avatar="🔍"):
+    with st.chat_message(SPEAKER_TYPES["BOT"], avatar="🔍"):
         st.write(response_text)
 
 # Add footer for additional information or credits