Update app.py

app.py

@@ -52,6 +52,7 @@ def clear_knowledge_base():
     vectors = None  # Reset the vector store
     return "Knowledge base cleared."
 
+# Function to process questions
 # Function to process questions
 def process_question(question):
     global vectors
@@ -60,7 +61,9 @@ def process_question(question):
     
     # Create document retrieval chain
     retriever = vectors.as_retriever(search_type="similarity", search_kwargs={"k": 5})
-    documents = retriever.get_relevant_documents(question)
+    
+    # Use the invoke method instead of get_relevant_documents
+    documents = retriever.invoke(question)
     
     if not documents:
         return "No relevant context found.", "", 0
@@ -68,14 +71,15 @@ def process_question(question):
     # Create context from retrieved documents
     context = "\n\n".join([doc.page_content for doc in documents])
 
-    # Use the LLM to answer the question
-    response = llm({"context": context, "input": question})
+    # Use the invoke method for the LLM
+    response = llm.invoke({"context": context, "input": question})
 
     # Confidence score as average relevance
     confidence_score = sum([doc.metadata.get('score', 0) for doc in documents]) / len(documents)
 
     return response, context, round(confidence_score, 2)
 
+
 # CSS styling
 CSS = """
 .duplicate-button { margin: auto !important; color: white !important; background: black !important; border-radius: 100vh !important;}