Update app.py

app.py

@@ -8,6 +8,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 import google.generativeai as genai
 from langchain.chains.question_answering import load_qa_chain  # Import load_qa_chain
 
+
 async def initialize(file_path, question):
     genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
     model = genai.GenerativeModel('gemini-pro')
@@ -36,9 +37,8 @@ async def initialize(file_path, question):
         # Use ainvoke to get the result
         stuff_answer = await stuff_chain.ainvoke({"input_documents": pages, "question": question, "context": context})
         
-        # Post-process the answer to remove any unnecessary text or system messages
-        answer = stuff_answer.strip()
-        answer = answer.split("Answer:")[1] if "Answer:" in answer else answer
+        # Access the correct key for the answer
+        answer = stuff_answer.get('output_text', '').strip()
 
         # Extract the page number where the context was found
         sources = []