from .emotion import emotion_analysis
from .ner import ner
from .prompt import llm_prompt
from .sentiment import sentiment_analysis
from .pos_tagging import pos
from .topic_modeling import topic_modeling
from fastapi import APIRouter

router = APIRouter()
router.add_api_route("/emotion-analysis", emotion_analysis.predict, methods=["POST"])
router.add_api_route("/ner", ner.predict, methods=["POST"])
router.add_api_route("/pos-tagging", pos.predict, methods=["POST"])
router.add_api_route("/prompt", llm_prompt.prompt, methods=["POST"])
router.add_api_route("/sentiment-analysis", sentiment_analysis.predict, methods=["POST"])
# router.add_api_route("/topic-modeling", topic_modeling.predict, methods=["POST"])


@router.get('/') 
async def index(req: dict):
    return {'app': 'baLLseM', 'version': '0.0.1'}