from fastapi import FastAPI
from embedding import PDFEmbedding

app = FastAPI()

@app.get("/")
def greet_json():
    return {"Hello": "World!"}


# API 엔드포인트로 이동됨
# app.py의 다음 코드 참조:
@app.post("/process")
def process_pdfs():
    pdf_embedding = PDFEmbedding(
        model_path="../ai_models/hf/BGE-m3-ko", 
        pdf_dir="./data/pdf"
    )
    pdf_embedding.process_and_store()
    return {"status": "success"}