Update app.py
Browse files
app.py
CHANGED
@@ -11,11 +11,12 @@ from langchain.prompts import PromptTemplate
|
|
11 |
from langchain_community.document_loaders import PyPDFLoader
|
12 |
from langchain_openai import ChatOpenAI
|
13 |
from langchain.callbacks import get_openai_callback
|
|
|
14 |
|
15 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
16 |
|
17 |
def process_pdf(file):
|
18 |
-
loader = PyPDFLoader(file)
|
19 |
documents = loader.load()
|
20 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|
21 |
splits = text_splitter.split_documents(documents)
|
|
|
11 |
from langchain_community.document_loaders import PyPDFLoader
|
12 |
from langchain_openai import ChatOpenAI
|
13 |
from langchain.callbacks import get_openai_callback
|
14 |
+
from io import BytesIO
|
15 |
|
16 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
17 |
|
18 |
def process_pdf(file):
|
19 |
+
loader = PyPDFLoader(BytesIO(file.read()))
|
20 |
documents = loader.load()
|
21 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
|
22 |
splits = text_splitter.split_documents(documents)
|