Update app.py

app.py

@@ -10,6 +10,19 @@ from langchain.memory import ConversationBufferMemory
 from langchain_core.prompts import PromptTemplate
 import streamlit as st
 import os
+from io import BytesIO
+
+class InMemoryPDFLoader(BaseLoader):
+    def __init__(self, file_bytes: bytes):
+        self.file_bytes = file_bytes
+
+    def load(self):
+        pdf_stream = BytesIO(self.file_bytes)
+        with pdfplumber.open(pdf_stream) as pdf:
+            text = ""
+            for page in pdf.pages:
+                text += page.extract_text()
+        return [Document(page_content=text)]
 
 # Access the OpenAI API key from the environment
 open_ai_key = os.getenv("OPENAI_API_KEY")
@@ -31,10 +44,16 @@ prompt = PromptTemplate(template=template, input_variables=["context", "question
 # upload PDF
 pdf_file = st.file_uploader("Upload your pdf",type="pdf")
 question = st.text_input("Ask a question:")
+
+
+
+
 if pdf_file is not None:
-    try:       
+    try:   
+        pdf_bytes = pdf_file.read()
+        loader = InMemoryPDFLoader(file_bytes=pdf_bytes)
         # Load and process the PDF
-        loader = PDFPlumberLoader(pdf_file)
+        # loader = PDFPlumberLoader(pdf_file)
         pdf_data = loader.load()
         
         # Split the text into chunks