Update app.py

app.py

@@ -16,7 +16,11 @@ def initialize(pdf_file, question):
             # Construct potential file path based on temporary directory and filename
             file_path = os.path.join("/tmp", file_info.name)  # Adjust temporary directory if needed
             if os.path.exists(file_path):
-                # ... rest of your code for processing the PDF using the file path
+                # 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
+                # ... rest of your code for processing the PDF using context and question
             else:
                 return "Error: The uploaded file could not be found."
         else:
@@ -25,6 +29,23 @@ def initialize(pdf_file, question):
     except Exception as e:
         return f"An error occurred: {e}"  # Generic error handling
 
+# 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"])
+
 # Create a Gradio interface
 interface = gr.Interface(
     fn=initialize,