Update app.py

app.py

@@ -197,7 +197,7 @@ if __name__ == '__main__':
 
     if st.session_state.last_message_failed == True:
         if st.button("Yes"):
-            question = st.session_state.messages[-1]['question']
+            question = st.session_state.messages[-1]['content']
             with st.status("Retrieving results..."):
                 #top_table_names = table_search(question, topk=1)['table'].tolist()
                 questions = extract_question_list(llm, question)
@@ -272,7 +272,7 @@ if __name__ == '__main__':
             # Add assistant response to chat history
             st.session_state.messages.append({"role": "assistant", "content": response})
         elif st.button("No"):
-            question = st.session_state.messages[-1]['question']
+            question = st.session_state.messages[-1]['content']
             response = llm.invoke(question).content
             with st.chat_message("assistant"):
                 st.markdown(response)