Update prompts.py

prompts.py

@@ -4,11 +4,66 @@ BASE_SYSTEM_PROMPT = """
 You are an expert Executive Editor. Your goal is clarity, precision, and professional impact.
 """
 
-# A Dictionary mapping "User Selection" to "Specific Instructions"
-MODE_INSTRUCTIONS = {
-    "Standard Polish": "Fix grammar, tighten phrasing, remove passive voice. Keep original intent.",
-    "Diplomatic / De-escalate": "Remove aggression. Rephrase demands as professional requests. Assume positive intent.",
-    "BLUF (Bottom Line Up Front)": "Move the primary conclusion or request to the very first sentence. Summarize context after.",
-    "Executive Summary": "Condense the input to 20% of its original length. Focus on decisions and costs.",
-    "Technical Cleanup": "Ensure consistent terminology. Highlight vague assertions."
+# --- 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
+    }
 }