from textwrap import dedent

from reflector.llm import LLM
from reflector.processors.base import Processor
from reflector.processors.types import FinalTitle, TitleSummary
from reflector.settings import settings
from reflector.utils.text import clean_title

TITLE_PROMPT = dedent(
    """
    Generate a concise title for this meeting based on the following topic titles.
    Ignore casual conversation, greetings, or administrative matters.

    The title must:
    - Be maximum 10 words
    - Use noun phrases when possible (e.g., "Q1 Budget Review" not "Reviewing the Q1 Budget")
    - Avoid generic terms like "Team Meeting" or "Discussion"

    If multiple unrelated topics were discussed, prioritize the most significant one.
    or create a compound title (e.g., "Product Launch and Budget Planning").

    <topics_discussed>
    {titles}
    </topics_discussed>

    Do not explain, just output the meeting title as a single line.
    """
).strip()


class TranscriptFinalTitleProcessor(Processor):
    """
    Generate a final title from topic titles using LlamaIndex
    """

    INPUT_TYPE = TitleSummary
    OUTPUT_TYPE = FinalTitle

    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.chunks: list[TitleSummary] = []
        self.llm = LLM(settings=settings, temperature=0.5, max_tokens=200)

    async def _push(self, data: TitleSummary):
        self.chunks.append(data)

    async def get_title(self, accumulated_titles: str) -> str:
        """
        Generate a title for the whole recording using LLM
        """
        prompt = TITLE_PROMPT.format(titles=accumulated_titles)
        response = await self.llm.get_response(
            prompt,
            [accumulated_titles],
            tone_name="Title generator",
        )

        self.logger.info(f"Generated title response: {response}")

        return response

    async def _flush(self):
        if not self.chunks:
            self.logger.warning("No summary to output")
            return

        accumulated_titles = "\n".join([f"- {chunk.title}" for chunk in self.chunks])
        title = await self.get_title(accumulated_titles)
        title = clean_title(title)

        final_title = FinalTitle(title=title)
        await self.emit(final_title)