Dewasheesh commited on
Commit
99ab820
·
verified ·
1 Parent(s): 64ce359

Update app/pdf_handler.py

Browse files
Files changed (1) hide show
  1. app/pdf_handler.py +12 -1
app/pdf_handler.py CHANGED
@@ -1,6 +1,6 @@
1
  import os
2
  from fastapi import UploadFile
3
- from app.vector_store import store_pdf, store_pdf_image
4
 
5
  UPLOAD_DIR = "data/uploaded_pdfs"
6
 
@@ -21,3 +21,14 @@ def process_image_pdf(file: UploadFile):
21
  f.write(file.file.read())
22
  store_pdf_image(filepath)
23
  return {"status": "uploaded", "filename": file.filename}
 
 
 
 
 
 
 
 
 
 
 
 
1
  import os
2
  from fastapi import UploadFile
3
+ from app.vector_store import store_pdf, store_pdf_image, store_pdf_image_text
4
 
5
  UPLOAD_DIR = "data/uploaded_pdfs"
6
 
 
21
  f.write(file.file.read())
22
  store_pdf_image(filepath)
23
  return {"status": "uploaded", "filename": file.filename}
24
+
25
+
26
+ def process_image_pdf_dummy():
27
+ try:
28
+ with open("output.txt", "r") as f:
29
+ content = f.read()
30
+ print(f'{content}')
31
+ except FileNotFoundError:
32
+ return {"error": "File not found"}
33
+ store_pdf_image_text(content)
34
+ return {"status": "uploaded"}