from typing import Dict

from ielts_be.controllers import ITrainingController
from ielts_be.services import ITrainingService
from ielts_be.dtos.training import FetchTipsDTO


class TrainingController(ITrainingController):

    def __init__(self, training_service: ITrainingService):
        self._service = training_service

    async def fetch_tips(self, data: FetchTipsDTO):
        return await self._service.fetch_tips(data.context, data.question, data.answer, data.correct_answer)

    async def get_training_content(self, data: Dict):
        return await self._service.get_training_content(data)