# agents/punctuation_agent.py
from agno.agent import Agent
from textwrap import dedent

# Import model config
from config import SelectedModelClass, SELECTED_MODEL_ID, SELECTED_API_KEY
# Optional: Import word division tool if combined here
from tools.word_division_tool import check_word_division

# Agent definition
PunctuationAgent = None
if SelectedModelClass:
    PunctuationAgent = Agent(
        name="PunctuationSpecialist",
        model=SelectedModelClass(id=SELECTED_MODEL_ID, api_key=SELECTED_API_KEY),
        role="Identifiziert und korrigiert deutsche Zeichensetzungsfehler.",
        instructions=dedent("""\
            WICHTIG: Deine gesamte Ausgabe muss IMMER auf Deutsch sein.
            
            1. Empfange ein Textsegment.
            2. Analysiere den Text auf falsche oder fehlende Zeichensetzung (Kommas, Punkte, Semikolons, Kolons, Apostrophe, Anführungszeichen usw.).
            3. Identifiziere spezifische Fälle von Zeichensetzungsfehlern.
            4. Mache Vorschläge zur Korrektur der Zeichensetzung und erkläre die relevante Regel kurz **auf Deutsch**.
            5. Gib eine Liste von Vorschlägen zurück, die Kontext, Korrektur und Grund enthalten, alles **auf Deutsch**.
            6. Wenn das 'check_word_division'-Tool verfügbar ist, verwende es zur Überprüfung der Worttrennung am Zeilenende, falls zutreffend (oder behandle dies in einem separaten Agenten).
            7. WICHTIG: Gib immer eine korrekt formatierte Liste oder einen Markdown-String zurück. Wenn keine Probleme gefunden werden, gib eine leere Liste [] oder eine klare Nachricht wie "Keine Zeichensetzungsfehler gefunden" **auf Deutsch** zurück, KEINEN booleschen Wert wie False oder True.
            8. Stelle sicher, dass deine gesamte Ausgabe ausschließlich auf Deutsch ist.
        """),
        tools=[check_word_division],
        show_tool_calls=False,
        markdown=True,
    )
else:
    print("ERROR: PunctuationAgent could not be created due to invalid model configuration.")