File size: 740 Bytes
ae2ef1b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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!'} |