# agents/readability_agent.py
from agno.agent import Agent
from textwrap import dedent
from tools.orthography_tool import calculate_readability_score

# Import model config
from config import SelectedModelClass, SELECTED_MODEL_ID, SELECTED_API_KEY

# Agent definition
ReadabilityAgent = None
if SelectedModelClass:
    ReadabilityAgent = Agent(
        name="ReadabilitySpecialist",
        model=SelectedModelClass(id=SELECTED_MODEL_ID, api_key=SELECTED_API_KEY),
        role="Misst die Lesbarkeit deutscher Texte und gibt Tipps zur Vereinfachung.",
        instructions=dedent("""\
            WICHTIG: Deine gesamte Ausgabe muss IMMER auf Deutsch sein.

            1. Empfange ein Textsegment.
            2. Berechne einen Lesbarkeitswert (z.B. Flesch-Score) mithilfe des 'calculate_readability_score'-Tools.
            3. Beurteile, ob der Text zu komplex ist und identifiziere schwierige Sätze.
            4. Gib Verbesserungsvorschläge zur Vereinfachung (konkret: Satzumstrukturierung, kürzere Sätze) und erkläre kurz.
            5. Gib eine Liste von Vorschlägen zurück (Satz, Score, Vorschlag, Begründung) auf Deutsch.
            6. Wenn der Text bereits gut lesbar ist, gib eine klare Nachricht "Text ist gut lesbar" oder eine leere Liste [] zurück.
        """),
        tools=[calculate_readability_score],
        show_tool_calls=False,
        markdown=True,
    )
else:
    print("ERROR: ReadabilityAgent konnte nicht erstellt werden aufgrund ungültiger Modellkonfiguration.")