import language_tool_python


class CorrectionService:

    def __init__(self):
        self.tools = {
            "fra_Latn": language_tool_python.LanguageTool("fr"),
            "eng_Latn": language_tool_python.LanguageTool("en"),
        }


    def correctText(self, text: str, sourceLang: str) -> str:
        tool = self.tools.get(sourceLang)

        if tool is None:
            raise ValueError(f"Langue non supportée : {sourceLang}")

        return tool.correct(text)