import re
import os
from utils.prompts import DOCUMENT_CLASSIFY_PROMPT
from llm_bedrock import model


def document_classifier(record_details):

    FINAL_DOCUMENT_CLASSIFY_PROMPT = DOCUMENT_CLASSIFY_PROMPT.format(document=record_details)

    json_string = model.invoke(FINAL_DOCUMENT_CLASSIFY_PROMPT).content

    print(json_string)

    pattern = r'<output>(.*?)</output>'
    match = re.search(pattern, json_string, re.DOTALL)

    if match:
        return match.group(1).strip()
    else:
        return ""