from __future__ import annotations
from services.model_router import ModelRouter
from config.prompts import LANGUAGE_AGENT_SYSTEM

class LanguageAgent:
    def __init__(self, router: ModelRouter): self._router = router

    def translate(self, text: str, target_lang: str) -> str:
        prompt = f"Translate to {target_lang}:\n\n{text}"
        return self._router.translate(prompt, LANGUAGE_AGENT_SYSTEM).strip()

    def explain_in_language(self, concept: str, explanation: str, target_lang: str) -> str:
        prompt = (f"Explain '{concept}' in {target_lang}.\n"
                  f"Original explanation: {explanation}")
        return self._router.translate(prompt, LANGUAGE_AGENT_SYSTEM).strip()