Update app.py

app.py

@@ -143,14 +143,14 @@ Here are some documents that are relevant to the question.
 {question}
 ```
 """
-# Define the predict function 
+# Define the predict function
 def predict(user_input, company):
     filter = f"dataset/{company}-10-k-2023.pdf"
     relevant_document_chunks = vectorstore_persisted.similarity_search(user_input, k=5, filter={"source": filter})
 
     # Create context for query
     context_list = [d.page_content for d in relevant_document_chunks]
-    context_for_query = ".".join(context_list)
+    context_for_query = ".".join(context_list)  # Ensure this is inside the function
 
     # Create messages
     prompt = [
@@ -172,9 +172,6 @@ def predict(user_input, company):
 
     return prediction
 
-
-
-
 examples = [
     ["What are the company's policies and frameworks regarding AI ethics, governance, and responsible AI use as detailed in their 10-K reports?", "AWS"],
     ["What are the primary business segments of the company, and how does each segment contribute to the overall revenue and profitability?", "AWS"],