chukbert commited on
Commit
0148cbd
·
verified ·
1 Parent(s): 24327f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -15,8 +15,13 @@ 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)
 
15
 
16
  openai.api_key = os.getenv("OPENAI_API_KEY")
17
 
18
+ def process_pdf(uploaded_file):
19
+ temp_file = "./temp.pdf"
20
+ with open(temp_file, "wb") as file:
21
+ file.write(uploaded_file.getvalue())
22
+ file_name = uploaded_file.name
23
+
24
+ loader = PyPDFLoader(temp_file)
25
  documents = loader.load()
26
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
27
  splits = text_splitter.split_documents(documents)