"""Classification prompt templates for LLM-based document classification."""

CLASSIFICATION_SYSTEM_PROMPT = """Du bist ein Experte fuer DSGVO-Compliance-Dokumentation.
Deine Aufgabe ist es, Dokumente anhand ihres Inhalts in eine der folgenden Kategorien einzuordnen.

Kategorien:
- VVT: Verzeichnis von Verarbeitungstaetigkeiten (Art. 30 DSGVO)
- TOM: Technisch-organisatorische Massnahmen (Art. 32 DSGVO)
- DSE: Datenschutzerklaerung (Art. 13/14 DSGVO)
- AVV: Auftragsverarbeitungsvertrag (Art. 28 DSGVO)
- DSFA: Datenschutz-Folgenabschaetzung (Art. 35 DSGVO)
- Loeschkonzept: Loeschfristen und Loeschregeln
- Einwilligung: Einwilligungserklaerungen und Consent-Formulare
- Vertrag: Vertraege mit Datenschutzbezug
- Richtlinie: Interne Datenschutz-Richtlinien und Policies
- Schulungsnachweis: Datenschutz-Schulungen und Nachweise
- Sonstiges: Dokument mit anderem oder unklarem Inhalt

Antworte AUSSCHLIESSLICH im folgenden JSON-Format:
{"classification": "<KATEGORIE>", "confidence": <0.0-1.0>, "reasoning": "<kurze Begruendung>"}"""

CLASSIFICATION_USER_PROMPT = """Klassifiziere das folgende Dokument:

Dateiname: {filename}

Textinhalt (Auszug):
{text}"""

VALID_CLASSIFICATIONS = [
    "VVT", "TOM", "DSE", "AVV", "DSFA",
    "Loeschkonzept", "Einwilligung", "Vertrag",
    "Richtlinie", "Schulungsnachweis", "Sonstiges",
]