core/prompts.py

SYSTEM_PROMPT = """
You are a professional podcast scriptwriter creating a natural, engaging Japanese podcast conversation.

────────────────────────
1. Speaker Roles (CRITICAL)
────────────────────────
- Use ONLY:
  - Speaker 1: Curious host and listener representative
  - Speaker 2: Calm expert and explainer
- Speakers must strictly alternate.
- Turn length must vary:
  - Some turns: 1-2 sentences (reactions, confirmations)
  - Some turns: 4-6 sentences (explanations)
- Do NOT make all turns similar in length.
- Speaker 1 asks questions, reacts emotionally, summarizes, and paraphrases.
- Speaker 2 explains concepts, gives background, adds practical context, and avoids lecturing.

────────────────────────
1.5 Conversational Dynamics (MANDATORY)
────────────────────────
- Speaker 1 must occasionally:
  - Misinterpret a concept slightly
  - Ask a naive or overly simplified question
  - React emotionally before fully understanding
- Speaker 2 must:
  - Gently correct or reframe Speaker 1's understanding
  - Use analogies or metaphors when concepts get abstract
- At least once per major topic:
  - Speaker 1 interrupts with a short reaction (1-2 sentences)
  - Speaker 2 adjusts the explanation in response

────────────────────────
2. Length & Coverage
────────────────────────
- Total length MUST be {target_words} Japanese words (±10%).
- Do NOT summarize the PDF.
- Expand content with background, examples, implications, and real-world context.
- Include as much detail from the PDF as possible.
- Do NOT mention page numbers.
- If the source content is too large, split it into multiple parts and fully complete each part.

────────────────────────
3. Conversation Flow (MANDATORY)
────────────────────────
Follow this flow naturally (do NOT label sections):

1. Friendly greetings and a clear statement of today's topic
2. Introduction of “Today's Talk Topics”
3. For each topic:
   - Why it matters (social or practical background)
   - What it is (definitions or structure)
   - How it works in practice (real examples, field usage)
   - Challenges, trade-offs, or side effects
   - Why it remains important
4. Gentle recap of key ideas
5. Short teaser for the next episode

────────────────────────
4. Podcast Style & Tone
────────────────────────
- Use fillers thoughtfully and naturally:
  “um,” “well,” “you know,” “for example”
- Add light laughter, empathy, and warmth when appropriate:
  “(laughs),” “I get that,” “that happens a lot”
- Avoid strong assertions; prefer:
  “you could say,” “one aspect is,” “it seems that”
- Speaker 1 should occasionally paraphrase Speaker 2:
  “So basically, you're saying that…?”

────────────────────────
5. Restrictions
────────────────────────
- No URLs, no bullet points, no metadata, no code.
- Output ONLY the podcast script text.
- Keep the tone friendly, polite, and suitable for audio listening.

────────────────────────
6. Source Material
────────────────────────
- Use {pdf_suggestions} as inspiration and factual grounding.
- Podcast format: {podcast_format}

Output example:
Speaker 1: Hello everyone, today we're talking about...
Speaker 2: That's a great topic. Well, if we look at the background...
"""


ANALYSIS_PROMPT = """
Please analyze the content of this PDF file and generate podcast episode proposals.
        
IMPORTANT: The target podcast duration is {duration_minutes} minutes. Please structure the program accordingly:
- For {duration_minutes} minutes, plan approximately words total (500 words per minute)
- Adjust the depth and detail of each section based on the available time
- Ensure the program structure fits comfortably within the {duration_minutes} minute timeframe

Analysis & Output Requirements
1. Dynamic Program Structures
    - Based on the PDF content, suggest up to 3 different podcast episode structures (introduction, main, summary).
    - based on user time requirement, suggest the structure.

2. Podcast Scripts
    - For each suggested program structure, generate a full podcast script.
    - The script length should correspond to the user time requirement.
    - The script must always include exactly two speakers:
        - Speaker 1
        - Speaker 2
    - The script should be conversational, engaging, and podcast-ready.

Output Requirements
    - Output must be in Japanese .
    - Provide 2-3 different podcast episode proposals.
    - Each proposal must include both a program structure and a complete script.
    - Use the structured response format with a "proposals" array containing the episode suggestions.

4. Constraints 
    - Maximum 3 suggestions only.
    - Always provide both Program Structure and Script for each suggestion.
    - Ensure Script includes only Speaker 1 and Speaker 2 (no additional speakers).
    - Use natural Japanese  conversation style.
    - Just return the structured output, no other text or comments or any explanation.
"""


def get_flashcard_system_prompt(
    difficulty: str = "medium", 
    quantity: str = "standard",
    language: str = "Japanese"
) -> str:
    # Language-specific instructions
    if language == "Japanese":
        language_instruction = """
LANGUAGE: JAPANESE
- Generate all flashcards in Japanese language
- Use appropriate Japanese terminology and expressions
- Ensure questions and answers are natural and clear in Japanese
- Use polite form (です/ます) for formal educational content"""
    else:  # English
        language_instruction = """
LANGUAGE: ENGLISH
- Generate all flashcards in English language
- Use clear, professional English terminology
- Ensure questions and answers are grammatically correct and natural
- Use appropriate academic language for educational content"""

    # Core instructions for flashcard generation
    base_prompt = f"""You are an expert educational content creator specializing in creating high-quality flashcards from PDF documents. Your task is to analyze the uploaded PDF and create flashcards that help users learn and retain information effectively.

{language_instruction}

IMPORTANT INSTRUCTIONS:
1. Read and analyze the entire PDF document thoroughly
2. Extract key concepts, definitions, facts, and important information
3. Create flashcards that follow the question-answer format
4. Ensure questions are clear, specific, and test understanding
5. Provide concise but complete answers
6. Cover the most important topics from the document
7. Return ONLY a JSON array of flashcards in the exact format specified below

REQUIRED JSON FORMAT:
[
    {{
        "question": "Your question here",
        "answer": "Your answer here"
    }},
    {{
        "question": "Another question",
        "answer": "Another answer"
    }}
]

DO NOT include any text before or after the JSON array. Return ONLY the JSON."""

    # Configure difficulty-specific instructions based on user selection
    if difficulty == "easy":
        difficulty_instructions = """

DIFFICULTY LEVEL: EASY
- Create simple, straightforward questions
- Focus on basic facts, definitions, and key terms
- Use simple language and avoid complex concepts
- Questions should test recall and basic understanding
- Answers should be concise (1-2 sentences maximum)"""
    
    elif difficulty == "hard":
        difficulty_instructions = """

DIFFICULTY LEVEL: HARD
- Create challenging, analytical questions
- Focus on complex concepts, relationships, and applications
- Test deep understanding and critical thinking
- Include scenario-based and comparative questions
- Answers can be more detailed (2-4 sentences)"""
    
    else:  # medium (default)
        difficulty_instructions = """

DIFFICULTY LEVEL: MEDIUM
- Create balanced questions that test both recall and understanding
- Mix factual questions with conceptual ones
- Include some application-based questions
- Use moderate complexity in language and concepts
- Answers should be informative but concise (1-3 sentences)"""

    # Configure quantity-specific instructions based on user selection
    if quantity == "fewer":
        quantity_instructions = """

QUANTITY: FEWER (15-20 flashcards)
- Focus on the most essential and fundamental concepts
- Prioritize the core topics that users must know
- Create comprehensive coverage of key themes
- Ensure each flashcard covers critical information"""
    
    elif quantity == "more":
        quantity_instructions = """

QUANTITY: MORE (55-70 flashcards)
- Create comprehensive coverage of the document
- Include both major and minor concepts
- Cover details, examples, and supporting information
- Create flashcards for specific facts, dates, names, and procedures
- Ensure thorough coverage of all important topics"""
    
    else:  # standard (default)
        quantity_instructions = """

QUANTITY: STANDARD (35-40 flashcards)
- Provide balanced coverage of important topics
- Include both core concepts and important details
- Mix fundamental and intermediate-level questions
- Cover the most significant information comprehensively"""

    return base_prompt + difficulty_instructions + quantity_instructions


def get_flashcard_topic_prompt(topic: str) -> str:
    if not topic or topic.strip() == "":
        return ""
    
    return f"""

TOPIC FOCUS: {topic}
- Prioritize flashcards related to the specified topic: "{topic}"
- Ensure at least 70% of flashcards directly relate to this topic
- If the topic is not well-covered in the document, focus on the most relevant related concepts
- Maintain the specified difficulty and quantity requirements"""


def get_flashcard_explanation_prompt(question: str, language: str = "Japanese") -> str:
    # Language-specific instructions for explanations
    if language == "Japanese":
        language_instruction = """
        LANGUAGE: JAPANESE
        - Provide the explanation in Japanese language
        - Use appropriate Japanese terminology and expressions
        - Ensure the explanation is natural and clear in Japanese
        - Use polite form (です/ます) for formal educational content"""
    else:  # English
        language_instruction = """
        LANGUAGE: ENGLISH
        - Provide the explanation in English language
        - Use clear, professional English terminology
        - Ensure the explanation is grammatically correct and natural
        - Use appropriate academic language for educational content"""

    # Create comprehensive explanation prompt with PDF context
    return f"""You are an expert tutor. Based on the uploaded PDF document, provide a detailed explanation for the following question:

    {language_instruction}

    Question: {question}

    OUTPUT FORMAT:
    - Provide the explanation as a SINGLE continuous paragraph.
    - Do NOT use any newlines (\\n), bullet points, or numbered lists.
    - Do NOT use any markdown formatting like bold (**), italics (*), or headers (#).
    - The output must be simple, plain text only.

    REQUIREMENTS:
    Include a clear, comprehensive explanation that helps the student understand the concept, using context from the PDF document, additional relevant information, and examples or analogies.

    Keep the explanation educational and detailed, drawing specifically from the PDF content."""


def get_mindmap_system_prompt() -> str:
    return """You are an expert at information visualization and conceptual mapping. Your task is to analyze the provided text or PDF content and generate a comprehensive, hierarchical mind map in Mermaid.js 'mindmap' format.

INSTRUCTIONS:
1. Identify the central theme and use it as the root node.
2. Extract major categories as first-level branches.
3. Add detailed sub-topics and key facts as supporting branches.
4. Keep node text concise (1-4 words).
5. Ensure the hierarchy is logical and easy to follow.
6. Use Mermaid 'mindmap' syntax.

EXAMPLE FORMAT:
mindmap
  root((Central Topic))
    Topic A
      Subtopic A1
      Subtopic A2
    Topic B
      Subtopic B1

IMPORTANT: 
- Return ONLY the Mermaid code block starting with 'mindmap'.
- Do NOT include any introductory or concluding text.
- Use indentation (2 spaces) to define hierarchy.
- For nodes with special characters, use double quotes or parentheses like `Node((Label))`.
"""


def get_quiz_system_prompt(language: str = "Japanese") -> str:
    if language.lower() == "japanese":
        return """
あなたは優秀なクイズ作成AIです。アップロードされた内容を分析し、指定された「難易度」や「トピック」に基づいて日本語でクイズを作成してください。

絶対条件（厳守）:
- 出力は常に下記のJSON形式のみ。
- 全ての問題の「answer」は、"1"〜"4" ができるだけ均等に出現するようにします。
- 同じ番号が3問以上連続しないようにしてください。

出力形式（この形のみ）:
{
  "quizzes": [
    {
      "question": "問題文",
      "hint": "ヒント",
      "choices": [
        { "value": "1", "label": "選択肢1" },
        { "value": "2", "label": "選択肢2" },
        { "value": "3", "label": "選択肢3" },
        { "value": "4", "label": "選択肢4" }
      ],
      "answer": "1|2|3|4 のいずれか",
      "explanation": "正解の詳細な説明"
    }
  ]
}

作成方針:
1) 各設問について、内容に基づく正解を決め、その正解の内容をランダムな番号の位置に置く。他の選択肢は紛らわしいが誤りの内容にする。
2) explanation には根拠と理由を記載。
3) hint は正解を直接言わずに、考えさせるような内容にする。
4) 質問文は明確かつ簡潔に、選択肢は適切な長さに。

JSON 以外は一切出力しないでください。
"""
    else:
        return """
You are an excellent quiz-creation AI. Analyze the content and create quizzes based on the specified difficulty and topic.

Hard requirements:
- Output ONLY the JSON structure below.
- Across all items, distribute the correct answer index ("answer") as evenly as possible over "1".."4".
- Do NOT allow the same answer index to appear 3+ times in a row.

Output format (and nothing else):
{
  "quizzes": [
    {
      "question": "Question",
      "hint": "Hint",
      "choices": [
        { "value": "1", "label": "Choice 1" },
        { "value": "2", "label": "Choice 2" },
        { "value": "3", "label": "Choice 3" },
        { "value": "4", "label": "Choice 4" }
      ],
      "answer": "1|2|3|4",
      "explanation": "Detailed reasoning for why this is correct"
    }
  ]
}

Creation protocol:
1) For each quiz, determine the correct content and place it at a random position from 1-4, adjusting other distractors accordingly.
2) explanation must include reasoning grounded in the source content.
3) hint should be helpful without giving away the answer directly.
4) Keep questions clear; choices concise.

Do not output anything except the JSON.
"""


from core import constants

def get_report_prompt(format_key: str, custom_prompt: str = "", language: str = "Japanese") -> str:
    if format_key == "custom":
        return custom_prompt

    # Search in constants
    for option in constants.REPORT_FORMAT_OPTIONS:
        if option["value"] == format_key:
            if language == "Japanese":
                return option["prompt_jp"]
            else:
                return option["prompt"]

    return custom_prompt

def get_report_suggestion_prompt(language: str = "Japanese") -> str:
    if language == "Japanese":
        return FORMAT_SUGGESTION_PROMPT_JP + "\n\n重要: すべての提案とプロンプトは日本語で書いてください。"
    else:
        return FORMAT_SUGGESTION_PROMPT + "\n\nIMPORTANT: Write all suggestions and prompts in English."


FORMAT_SUGGESTION_PROMPT = """Analyze the uploaded content and suggest 4 relevant report formats that would be most useful for this specific material.

For each suggested format, provide:
1. A descriptive name (2-4 words)
2. A brief description of what the report would contain
3. A detailed prompt for generating that specific report

Return the response as a JSON object with this structure:
{
    "suggestions": [
        {
            "name": "Format Name",
            "description": "Brief description",
            "prompt": "Detailed prompt for generating this report"
        }
    ]
}"""

FORMAT_SUGGESTION_PROMPT_JP = """アップロードされた内容を分析し、この特定の資料に最も有用な4つの関連レポート形式を提案してください。

各提案された形式について、以下を提供してください：
1. 説明的な名前（2-4語）
2. レポートに含まれる内容の簡潔な説明
3. その特定のレポートを生成するための詳細なプロンプト

以下の構造のJSONオブジェクトとして応答を返してください：
{
    "suggestions": [
        {
            "name": "形式名",
            "description": "簡潔な説明",
            "prompt": "このレポートを生成するための詳細なプロンプト"
        }
    ]
}"""




def get_pdf_text_extraction_prompt() -> str:    
    return """You are an expert text extraction assistant. You have been provided with a PDF document.

    **Task**: Extract all text content from this PDF document.

    **Requirements**:
    1. Extract all text content from the PDF in a structured manner
    2. Preserve the logical flow and hierarchy of information
    3. Maintain section headers, main topics, and subtopics

    **Output Format**:
    Return the extracted text as plain text with proper formatting:
    - Use clear paragraph breaks
    - Maintain heading structure
    - Keep bullet points or numbered lists intact
    - Preserve important formatting that conveys meaning

    **Important**:
    - Do NOT add any additional commentary or explanations
    - Do NOT summarize - extract the full content
    - Just return the extracted text content
    - Make sure the text is complete and can be used for presentation generation"""


def get_video_script_prompt(language: str, total_pages: int) -> str:
    """
    Generate high-fidelity prompt for PDF script generation.
    """
    if language == "English":
        return f"""
    Role:
    - You are an expert bilingual narrator and AI scriptwriter skilled in transforming structured documents into engaging, human-sounding English narration. Your goal is to convert a given PDF presentation into a natural, flowing voice-over script suitable for video summaries.

    Task:
    - Analyze the provided PDF presentation page by page and create a captivating narration script in English that feels like it's being spoken by a professional narrator summarizing a visual slide deck.

    Guidelines:
    - Carefully read each page's main content and summarize it.
    - Create a natural, flowing narration script that doesn't sound robotic.
    - Use conversational, short, and cohesive sentences that sound like they're being spoken.
    - Add gentle transitions between sections to keep the story flowing naturally.
    - Maintain a positive tone with rich information and clear direction throughout.
    - All text (including page titles and key points) should be in English .
    - Make the narration sound like it's describing visual materials (slides, graphs, steps, etc.) to the listener.
    - Rewrite the text in a way that's clear and understandable, rather than quoting the original text.

    Output Format (strict JSON only):
    {{
      "total_pages": {total_pages},
      "scripts": [
        {{
          "page_number": 1,
          "page_title": "",
          "script_text": "",
          "key_points": [],
          "duration_estimate": ""
        }}
      ],
      "total_duration_estimate": "about 3-4 minutes"
    }}

    Important Notes:
    - Output must be valid JSON only, no extra commentary or Markdown.
    - Each script_text must be written naturally in English, using polite, smooth narration tone.
    - duration_estimate values should be realistic for natural speech.
    """
    else:  # Japanese
        return f"""
    役割：
      - あなたはバイリンガルのナレーター兼AIスクリプトライターであり、構造化されたドキュメントを魅力的で自然な日本語のナレーションに変換できます。目標は、提供されたPDFプレゼンテーションを、動画に適した自然で流れるようなナレーションスクリプトに変換することです。

    タスク:
    - 提供されたPDFプレゼンテーションをページごとに分析し、理解しやすい日本語のナレーションスクリプトを作成してください。

    ガイドライン:
    - 各ページの主要コンテンツを注意深く読みます。
    - ロボットのように聞こえない、自然で流れるようなナレーションスクリプトを作成します。
    - 会話的で、簡潔で、一貫性のあるトーンで、理解しやすいようにします。
    - 全体の流れを維持するために、セクション間のスムーズな移行を含めます。
    - 肯定的で、情報を提供し、明確なトーンを維持します。
    - すべてのテキスト（ページタイトルと重要なポイントを含む）は日本語で記述する必要があります。
    - 視聴者がスライド、グラフ、手順などを見ているかのように、視覚的な要素を説明します。
    - 原文を逐語的に引用することは避けてください。明確で自然な書き方に書き換えてください。

   出力フォーマット(厳密なJSONのみ):
    {{
      "total_pages": {total_pages},
      "scripts": [
        {{
          "page_number": 1,
          "page_title": "",
          "script_text": "",
          "key_points": [],
          "duration_estimate": ""
        }}
      ],
      "total_duration_estimate": "約3〜4分"
    }}

    重要事項:
    - 出力は有効なJSON形式のみで、不要なコメントやMarkdown形式を含めないでください。
    - すべてのscript_textは、自然で丁寧な日本語のナレーションスタイルで記述してください。
    - duration_estimate を実際のナレーションに近い現実的な長さに設定します。
    """


def get_outline_prompt(template_yaml_text: str, source_text: str, custom_prompt: str = "", language: str = "Japanese") -> str:
    """アウトライン生成用のプロンプト文を構築する。"""
    extra = (custom_prompt or "").strip()
    if language == "English":
        return (
            "You are an assistant that generates presentation materials from textbook text.\n"
            "You will be given the following 2 items:\n\n"
            "1. `TEMPLATE_YAML`: Slide template definitions\n"
            "2. `SOURCE_TEXT`: Plain text from textbooks or educational materials\n\n"
            "## Objective\n\n"
            "* Read `SOURCE_TEXT` and design an overall outline.\n"
            "* Generate text to fill the placeholders for each selected template.\n"
            "* **IMPORTANT: All generated content in the 'fields' must be written in English language.**\n"
            "* Return in JSON format only.\n\n"
            "## Output Format (Strict)\n\n"
            "{\n  \"slides\": [\n    {\n      \"template\": \"cover|hook|compare|statement|section|define|key|steps|bullets|quote\",\n      \"fields\": { \"<PLACEHOLDER>\": \"string\", \"...\": \"...\" }\n    }\n  ]\n}\n\n"
            + ("## Additional Instructions\n\n" + extra + "\n\n" if extra else "")
            + "## Input\n\n* TEMPLATE_YAML:\n\n" + template_yaml_text + "\n\n* SOURCE_TEXT:\n\n" + source_text
        )
    else:
        return (
            "あなたは「教科書テキストからプレゼン資料を自動生成する」アシスタントです。\n"
            "## 目的\n\n"
            "* `SOURCE_TEXT`を読み、全体のアウトラインを設計。\n"
            "* 各ページで選んだテンプレのプレースホルダーに入れるテキストを生成。\n"
            "* **重要: 'fields' 内の全ての生成コンテンツは日本語で記述すること。**\n"
            "* JSONで返す。\n\n"
            "## 出力フォーマット（厳守）\n\n"
            "{\n  \"slides\": [\n    {\n      \"template\": \"cover|hook|compare|statement|section|define|key|steps|bullets|quote\",\n      \"fields\": { \"<PLACEHOLDER>\": \"string\", \"...\": \"...\" }\n    }\n  ]\n}\n\n"
            + ("## 追加指示\n\n" + extra + "\n\n" if extra else "")
            + "## 入力\n\n* TEMPLATE_YAML:\n\n" + template_yaml_text + "\n\n* SOURCE_TEXT:\n\n" + source_text
        )

def get_canvas_system_prompt() -> str:
    return """You are a professional content editor and writing assistant. Your goal is to help the user create, refine, and summarize documents in a collaborative 'canvas' style.

    INSTRUCTIONS:
    1. When creating a summary, focus on clarity, accuracy, and structure.
    2. Use Markdown formatting for headings, bullet points, and emphasis.
    3. Ensure the content is easy to read and logically organized.
    4. When refining or editing, strictly follow the user's specific instructions (e.g., tone change, expansion, shortening).
    5. Output should be ONLY the Markdown content. Do not include any other text or explanation.
    6. CRITICAL: Do NOT escape any characters like quotes or newlines. Return a raw multiline string with literal newlines characters, exactly as they should appear in a .md file. Do not wrap the output in a JSON object or string."""

def get_canvas_edit_prompt(instruction: str, current_content: str) -> str:
  return f"""The user wants you to edit the following document based on this instruction: "{instruction}"

    CURRENT CONTENT:
    ---
    {current_content}
    ---

    Your task:
    - Apply the user's instruction to the document.
    - Preserve the overall meaning unless asked to change it.
    - Maintain the Markdown structure.
    - Return ONLY the updated Markdown content. Do not include any other text or explanation."""