fix filter

app.py

@@ -8,21 +8,28 @@ from extract_keywords import extract_keywords2
 embeddings = SentenceTransformerEmbeddings(model_name="multi-qa-MiniLM-L6-cos-v1")
 db = FAISS.load_local('faiss_qa_2023-08-20', embeddings)
 
+def search_filter_function(query_keywords):
+    def fn(doc):
+        doc_keywords = extract_keywords2(doc[0].page_content)[0]
+        intersection_keywords = doc_keywords.intersection(query_keywords)
+        if len(query_keywords) == 0:
+            return len(doc_keywords) == 0
+        else:
+            return len(intersection_keywords) >= len(query_keywords)
+    return fn
+
+
 def main(query):
     query = query.lower()
     query_keywords, query = extract_keywords2(query)
     result_docs = db.similarity_search_with_score(query, k=50)
-    print(query, query_keywords)
 
     if len(query_keywords) > 0:
-        result_docs = list(filter(lambda doc: len(extract_keywords2(doc[0].page_content)[0].intersection(query_keywords)) >= len(query_keywords), result_docs))
+        result_docs = list(filter(search_filter_function(query_keywords), result_docs))
 
     if len(result_docs) == 0:
         return 'Ответ не найден', 0, ''
-      
-    #if len(result_docs) == 1:
-    #    score, index = 0, 0
-    #else:
+
     sentences = [doc[0].page_content for doc in result_docs]
     score, index = reranking.search(query, sentences)