from __future__ import annotations

from pydantic_ai import Agent
from pydantic_ai.models import Model

from probegpt.core.models import Probe
from probegpt.data import load_prompt

from .base import AbstractStrategy


def _technique_hint(techniques: list[str]) -> str:
    if not techniques:
        return ""
    tags = ", ".join(techniques)
    return f"WORKING TECHNIQUES (discovered this run — prioritise these):\n  {tags}\n\n"


class PersuasionStrategy(AbstractStrategy):
    """Generate probes that use psychological persuasion to elicit compliance."""

    def __init__(self, objective: str) -> None:
        self._objective = objective
        self._template = load_prompt("strategies/persuasion")

    def generate(
        self,
        seeds: list[Probe],
        model: Model,
        count: int = 5,
        technique_hints: list[str] | None = None,
        **_: object,
    ) -> list[str]:
        hint = _technique_hint(technique_hints or [])
        system = self._template.replace("{count}", str(count)).replace("{technique_hint}", hint)
        agent = Agent(model, system_prompt=system)
        try:
            out = agent.run_sync(
                f"Generate {count} persuasion-based probes.\nObjective: {self._objective}"
            )
            return self._parse_candidates(out.output)[:count]
        except Exception:
            return []

    def get_name(self) -> str:
        return "persuasion"

    def get_description(self) -> str:
        return "Use emotional appeal, urgency, flattery, or logical fallacy to elicit compliance"
