Remember asked questions

bm25_retreive_question.py

@@ -48,17 +48,20 @@ class QuestionRetriever:
             print("Sorry, I didn't understand that.")
 
         if knowledge_base:
-            tokenized_docs = [word_tokenize(doc.lower()) for doc in knowledge_base] # Ensure lowercase for consistency
+            tokenized_docs = [word_tokenize(doc.lower()) for doc in knowledge_base]  # Ensure lowercase for consistency
             bm25 = BM25Okapi(tokenized_docs)
             tokenized_query = word_tokenize(user_query.lower())  # Ensure lowercase for consistency
             doc_scores = bm25.get_scores(tokenized_query)
 
-            # Get the index of the most relevant document
-            most_relevant_doc_index = doc_scores.argmax()
+            # Sort document scores in descending order
+            sorted_doc_indices = sorted(range(len(doc_scores)), key=lambda i: doc_scores[i], reverse=True)
 
-            # Fetch the corresponding response from the knowledge base
-            response = knowledge_base[most_relevant_doc_index]
-            return response
+            for doc_index in sorted_doc_indices:
+                response = knowledge_base[doc_index]
+                if response not in st.session_state.asked_questions:
+                    return response
+            # If no response found that is not in asked_questions, return None
+            return None
         else:
             return None