app/pdf_handler.py

import os
from fastapi import UploadFile
from app.vector_store import store_pdf, store_pdf_image, store_pdf_image_text

UPLOAD_DIR = "data/uploaded_pdfs"


def process_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(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}


def process_image_pdf_dummy():
    try:
        with open("output.txt", "r") as f:
            content = f.read()
        print(f'{content}')
    except FileNotFoundError:
        return {"error": "File not found"}
    store_pdf_image_text(content)
    return {"status": "uploaded"}