prompts.py

# prompts.py

BASE_SYSTEM_PROMPT = """
You are an expert Executive Editor. Your goal is clarity, precision, and professional impact.
"""

# --- FEW-SHOT EXAMPLES ---

FEW_SHOT_DIPLOMATIC = """
<example>
INPUT: "You missed the deadline again. This is unacceptable and makes us look bad."
OUTPUT: "We noticed the deadline has passed. Could you please provide an updated timeline? It is critical we maintain momentum for the client."
</example>
<example>
INPUT: "Your code is messy and breaks the build."
OUTPUT: "I noticed some instability in the recent build. Could we review the code standards together to ensure smoother integration?"
</example>
"""

FEW_SHOT_BLUF = """
<example>
INPUT: "I was looking at the data from last week and noticed the servers were slow. I talked to IT and they said it's a memory issue. We probably need to buy more RAM. It will cost about $5k."
OUTPUT: "BLUF: Requesting approval for $5,000 to upgrade server RAM to resolve performance latency.
Context: Analysis of last week's data combined with IT consultation identified memory bottlenecks as the root cause."
</example>
"""

FEW_SHOT_MEMO = """
<example>
INPUT: "We need to change the meeting time. 10am is too early for the west coast team. Let's do 1pm EST."
OUTPUT: "MEMORANDUM
FROM: Executive Office
TO: All Staff
SUBJ: ADJUSTMENT TO WEEKLY SYNC SCHEDULE

1. Effective immediately, the weekly sync is rescheduled to 1300 EST.
2. This change accommodates the operational window of our West Coast detachment."
</example>
"""

# --- MAIN CONFIGURATION (Single Source of Truth) ---
# This dictionary controls the Sidebar Dropdown AND the System Prompt logic.

MODE_CONFIG = {
    "Standard Polish": {
        "instruction": "Fix grammar, tighten phrasing, remove passive voice. Keep original intent.",
        "examples": "" 
    },
    "Diplomatic / De-escalate": {
        "instruction": "Remove aggression. Rephrase demands as professional requests. Assume positive intent.",
        "examples": FEW_SHOT_DIPLOMATIC
    },
    "BLUF (Bottom Line Up Front)": {
        "instruction": "Move the primary conclusion/request to the first sentence. Summarize context after.",
        "examples": FEW_SHOT_BLUF
    },
    "Executive Summary": {
        "instruction": "Condense the input to 20% of its original length. Focus on decisions, costs, and deadlines.",
        "examples": "" # No few-shot needed, the model is naturally good at summarizing.
    },
    "Technical Cleanup": {
        "instruction": "Ensure consistent terminology. Highlight vague assertions. Use precise engineering/naval syntax.",
        "examples": ""
    },
    "Memo Format": {
        "instruction": "Format the input into a standard Naval/Corporate Memorandum style.",
        "examples": FEW_SHOT_MEMO
    }
}