from fastapi import Body, APIRouter
from typing_extensions import Annotated
from src.models.request import Request
from sentence_transformers import SentenceTransformer
from src.processors.dialog.dialog_management import process_input_bn, reset_chatbot
from src.processors.downloader.model_downloader import download
from src.utils.identifier import Identifier
from definitions import ROOT_DIR

router = APIRouter()

@router.post('/query')
async def query(data: Annotated[Request, Body(embed=True)]):
    query = data.query.lstrip()
    identifier = Identifier(query)
    if (identifier.is_bangla):
        result = process_input_bn(query, model)
        return {'data': result}
    else:
        return {'data': 'I can\'t understand you!'}