Update app.py

app.py

@@ -134,15 +134,19 @@ if __name__ == '__main__':
 
     if 'history' not in st.session_state:
         st.session_state['history'] = []
+
+    if "previous_response" not in st.session_state:
+        st.session_state['previous_response'] = ""
         
     if question is not None and question != "":
-        q_relevant = extract_question_type(llm, question)
+        #q_relevant = extract_question_type(llm, question)
         with st.chat_message("user"):
             st.markdown(question)
         # Add user message to chat history
         st.session_state.messages.append({"role": "user", "content": question})
     
-        if 'yes' in q_relevant.lower():
+        #if 'yes' in q_relevant.lower():
+        if st.session_state['previous_response'] == "Sorry I may not have answer to this question.":
             st.session_state.last_message_failed = False
             with st.status("Retrieving results..."):
                 #top_table_names = table_search(question, topk=1)['table'].tolist()
@@ -178,6 +182,7 @@ if __name__ == '__main__':
                     response = "\n\n".join(responses)
                     if response == "":
                         response = "Sorry I may not have answer to this question."
+                        
                 else:
                     top_table_names = extract_table_name(llm, question) #[extract_table_name(llm, question)] 
                     print (top_table_names)
@@ -210,7 +215,9 @@ if __name__ == '__main__':
                     if response == "Sorry I may not have answer to this question.":
                         st.session_state.ask_user_selection = True
                         st.session_state.prev_selection = [mapping[tab] for tab in top_table_names if tab in mapping]
-                        
+
+            st.session_state['previous_response'] = response
+            
             with st.chat_message("assistant"):
                 st.markdown(response)
             # Add assistant response to chat history
@@ -289,7 +296,8 @@ if __name__ == '__main__':
                     if response == "Sorry I may not have answer to this question.":
                         st.session_state.ask_user_selection = True
                         st.session_state.prev_selection = [mapping[tab] for tab in top_table_names if tab in mapping]
-                        
+
+            st.session_state['previous_response'] = response
             with st.chat_message("assistant"):
                 st.markdown(response)
             # Add assistant response to chat history
@@ -332,7 +340,8 @@ if __name__ == '__main__':
 
         st.session_state.ask_user_selection = False
         st.session_state.prev_selection = []
-        
+
+        st.session_state['previous_response'] = response
         with st.chat_message("assistant"):
             st.markdown(response)