Update app.py

app.py

@@ -19,16 +19,43 @@ def initialize(pdf_file, question):
                 # Process the PDF
                 pdf_loader = PyPDFLoader(file_path)
                 pages = pdf_loader.load_and_split()
-                context = "\n".join(str(page.page_content) for page in pages[:30])  # Limit to first 30 pages
+                processed_context = "\n".join(str(page.page_content) for page in pages[:30])  # Limit to first 30 pages
 
                 # Configure Google Generative AI (replace with your API key)
+                genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+                # Prompt template for formatting context and question
+                prompt_template = """Answer the question as precise as possible using the provided context. If the answer is not contained in the context, say "answer not available in context" 
+
+                Context: 
+                {context}
+
+                Question: 
+                {question} 
+
+                Answer:
+                """
+
+                prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
+
+                # Generate answer using GeminiPro model
+                # Load the GeminiPro model
                 model = genai.GenerativeModel('gemini-pro')
-                model = ChatGoogleGenerativeAI(model="gemini-pro", temperature=0.3)
 
+                # Prepare the input data
+                input_data = {
+                    "context": processed_context,
+                    "question": question
+                }
+
+                # Generate the answer using load_qa_chain
                 stuff_chain = load_qa_chain(model, chain_type="stuff", prompt=prompt)
-                stuff_answer = stuff_chain({"input_documents": pages, "question": question, "context": context}, return_only_outputs=True)
+                stuff_answer = stuff_chain(input_data, return_only_outputs=True)
+
+                # Extract the answer
+                generated_answer = stuff_answer['output_text']
 
-                return stuff_answer['output_text']
+                return generated_answer
             else:
                 return "Error: The uploaded file could not be found."
         else:
@@ -56,3 +83,5 @@ interface.launch()
 
 
 
+
+