Update app.py

app.py

@@ -8,7 +8,7 @@ import json
 from auditqa.sample_questions import QUESTIONS
 from auditqa.engine.prompts import audience_prompts
 from auditqa.reports import files, report_list
-from auditqa.doc_process import process_pdf
+from auditqa.doc_process import process_pdf, get_local_qdrant
 from langchain_core.messages import (
     HumanMessage,
     SystemMessage,
@@ -24,7 +24,6 @@ HF_token = os.environ["HF_TOKEN"]
 # vectorestore colection are stored on persistent storage so this needs to be run only once
 # hence, comment out line below when creating for first time
 # vectorstores = process_pdf()
-#------now we will only use the existing embedding----------
 
 # -------------------------------------------------------------
 # Functions
@@ -94,9 +93,9 @@ async def chat(query,history,sources,reports,subtype,year):
 
     ##------------------------decide which collection to fetch------------------------------
     if len(reports) == 0:
-        vectorstore = vectorstores[sources]
+        vectorstore = get_local_qdrant(source)
     else:
-        vectorstore = vectorstores["allreports"]
+        vectorstore = get_local_qdrant("allreports")
 
     ##------------------------------get context----------------------------------------------------   
     context_retrieved_lst = []