# 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 = """
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."
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?"
"""
FEW_SHOT_BLUF = """
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."
"""
FEW_SHOT_MEMO = """
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."
"""
# --- 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
}
}