Update app/pdf_handler.py

app/pdf_handler.py

@@ -1,6 +1,6 @@
 import os
 from fastapi import UploadFile
-from app.vector_store import store_pdf
+from app.vector_store import store_pdf, store_pdf_image
 
 UPLOAD_DIR = "data/uploaded_pdfs"
 
@@ -12,3 +12,12 @@ def process_pdf(file: UploadFile):
         f.write(file.file.read())
     store_pdf(filepath)
     return {"status": "uploaded", "filename": file.filename}
+
+
+def process_image_pdf(file: UploadFile):
+    os.makedirs(UPLOAD_DIR, exist_ok=True)
+    filepath = os.path.join(UPLOAD_DIR, file.filename)
+    with open(filepath, "wb") as f:
+        f.write(file.file.read())
+    store_pdf_image(filepath)
+    return {"status": "uploaded", "filename": file.filename}