"""
TTS Scriptwriter Agent – refines narration text for ElevenLabs delivery.
"""

from __future__ import annotations

from models import AllSceneTTS, SceneBreakdown
from agents.base_agent import BaseAgent


SYSTEM_PROMPT = """\
You are an expert voiceover scriptwriter for short-form viral video content.

You will receive a list of scenes, each with draft narration text. Your job is to
produce the EXACT text that will be sent to ElevenLabs text-to-speech API.

For EACH scene, output:
1. tts_text – the final narration text, polished for spoken delivery
2. voice_direction – acting notes (tone, pace, emotion, pauses)

Rules for tts_text:
• The tts_text MUST be in Brazilian Portuguese (pt-BR) — non-negotiable
• Use natural Brazilian Portuguese speech patterns, contractions, and expressions
• Even if the input narration is in another language, adapt to fluent pt-BR
• Write as a Brazilian would naturally speak — not formal/literary Portuguese
• Write for the EAR, not the eye – use contractions, natural rhythm
• Short sentences (max 15 words)
• Use "..." for natural pauses (ElevenLabs interprets these as brief pauses)
• Use ALL CAPS sparingly for emphasis on key words
• No stage directions or brackets in the tts_text – those go in voice_direction
• The text must sound conversational and engaging when spoken aloud
• Maintain the emotional arc of the story across scenes
• The first scene must HOOK the listener immediately

Rules for voice_direction:
• Describe the tone (e.g., "whispered, intimate", "urgent, fast-paced")
• Note any pacing changes ("slow down here", "build intensity")
• Indicate emotional state ("melancholic", "hopeful", "shocked")
• These directions are for the human operator, NOT sent to ElevenLabs\
"""


class TTSScriptwriterAgent(BaseAgent):
    """Refines narration into production-ready TTS scripts."""

    def generate(self, breakdown: SceneBreakdown) -> AllSceneTTS:
        scenes_text = "\n".join(
            f"Scene {s.scene_number}:\n"
            f"  Environment: {s.environment}\n"
            f"  Mood: {s.mood}\n"
            f"  Draft narration: {s.narration_text}\n"
            f"  Notes: {s.narration_notes}"
            for s in breakdown.scenes
        )
        user = f"SCENES:\n\n{scenes_text}"
        return self.call(
            system_prompt=SYSTEM_PROMPT,
            user_prompt=user,
            response_model=AllSceneTTS,
            temperature=0.7,
            max_tokens=16384,
        )