Update app.py

app.py

@@ -8,10 +8,13 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 import google.generativeai as genai
 from langchain.chains.question_answering import load_qa_chain  # Import load_qa_chain
 
-# Initialize an empty list to store chat history
+# Initialize an empty list to store chat history and context
 chat_history = []
+context_history = ""
 
 async def initialize(file_path, question):
+    global context_history
+    
     genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
     model = genai.GenerativeModel('gemini-pro')
     model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3)
@@ -41,8 +44,11 @@ async def initialize(file_path, question):
         # Load the question-answering chain
         stuff_chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
         
+        # Combine previous context with the new context
+        combined_context = context_history + "\n" + context
+        
         # Get the answer from the model
-        stuff_answer = await stuff_chain.ainvoke({"input_documents": pages, "question": question, "context": context})
+        stuff_answer = await stuff_chain.ainvoke({"input_documents": pages, "question": question, "context": combined_context})
         answer = stuff_answer.get('output_text', '').strip()
         
         # Identify key sentences or phrases
@@ -82,6 +88,9 @@ async def initialize(file_path, question):
             'document_link': source_link
         })
         
+        # Update context history
+        context_history += f"\nQ: {question}\nA: {answer}"
+        
         return f"Answer: {answer}\n{source_str}\n{source_link}"
     else:
         return "Error: Unable to process the document. Please ensure the PDF file is valid."