Update app.py

app.py

@@ -144,12 +144,13 @@ Here are some documents that are relevant to the question.
 
 def predict(user_input,company):
 
-    #filter = "Dataset-10k/"+company+"-10-k-2023.pdf"
-    relevant_document_chunks = vectorstore_persisted.similarity_search(user_input, k=5)
+    #filter = "dataset/"+company+"-10-k-2023.pdf"
+    #relevant_document_chunks = vectorstore_persisted.similarity_search(user_input, k=5, filter={"source":filter})
 
     # Create context_for_query
 
-
+    relevant_document_chunks = retriever.invoke(user_input)
+    
     context_list = [d.page_content for d in relevant_document_chunks]
     context_for_query = ". ".join(context_list)
    
@@ -195,7 +196,7 @@ def predict(user_input,company):
         with log_file.open("a") as f:
             f.write(json.dumps(
                 {
-                    'user_input': user_input,
+                    'user_input': user_input, company
                     'retrieved_context': context_for_query,
                     'model_response': prediction
                 }