app.py

from fastapi import FastAPI, Request
from pydantic import BaseModel
from typing import List
from embedder import embed_clauses, embed_chunks
from tagger import clause_match_by_cosine

app = FastAPI()

class ChunkRequest(BaseModel):
    chunks: List[str]
    clause_reference: List[dict]

@app.post("/tag_clauses")
def tag_clauses(data: ChunkRequest):
    embedded_chunks = embed_chunks(data.chunks)
    embedded_references = embed_clauses(data.clause_reference)
    return clause_match_by_cosine(embedded_chunks, embedded_references)