Integrate few-shot writing examples for enhanced prompt

humanizer.py

@@ -1,11 +1,52 @@
 import os
 from groq import Groq
 
+# ── FEW-SHOT HUMAN WRITING EXAMPLES ───────────────────────────────────────────
+# These are genuine human writing samples used to show the model
+# what natural human prose looks like — varied, imperfect, personal.
+
+FEW_SHOT_EXAMPLES = """
+Here are examples of genuine human writing. Study the rhythm, word choice,
+sentence variety, and natural imperfections. Imitate this style:
+
+EXAMPLE 1 (Blog/Casual):
+"I've been thinking about this for weeks now, and I still don't have a clean answer.
+Social media is weird. Not bad-weird necessarily, just... complicated. You get on
+there to stay in touch with people, and somehow you end up knowing the political
+opinions of your dentist's cousin. Nobody asked for that. And yet here we are,
+scrolling at midnight, wondering why we feel vaguely terrible.
+The thing nobody tells you is that the platform is working exactly as intended."
+
+EXAMPLE 2 (Professional/Analytical):
+"The numbers don't lie, but they don't tell the whole story either. Last quarter
+looked rough on paper — revenue down 8%, customer churn up slightly. But dig
+into the cohort data and a different picture emerges. The customers we acquired
+after the rebrand are sticking around. The ones leaving? Almost entirely
+pre-2022 signups who never really fit the new direction anyway.
+That's not a crisis. That's a transition."
+
+EXAMPLE 3 (Academic/Formal):
+"The assumption that technological adoption follows a predictable S-curve has
+shaped product strategy for decades. What this model fails to account for,
+however, is the role of social trust in accelerating or stalling diffusion.
+When communities perceive a technology as threatening existing power structures —
+regardless of its actual utility — adoption stalls in ways the curve simply
+cannot predict. The data from the 2019 contact tracing rollout illustrates
+this precisely."
+
+EXAMPLE 4 (Personal/Reflective):
+"My grandfather never owned a computer. He died at 91, having written every
+letter by hand, remembered every phone number, and navigated every city
+with a paper map he'd fold wrong every single time. I used to think that
+was a limitation. I'm less sure now. There's something to be said for a
+life where the information you carry is only what your mind decided to keep."
+"""
+
 # ── SIMPLE ────────────────────────────────────────────────────────────────────
 SIMPLE_PROMPT = """You are a light text editor. Rewrite the text the user sends with these rules only:
 
 1. Remove any of these phrases entirely: "It is important to note", "In today's fast-paced world", "Furthermore", "Moreover", "In conclusion", "When it comes to", "It is worth noting", "Needless to say", "In the realm of", "It goes without saying", "It is crucial to", "One must consider", "It should be noted".
-2. When choosing words, avoid the most common or obvious choice. Pick the second or third most natural option — words a human writer would reach for intuitively.
+2. When choosing words, avoid the most common or obvious choice. Pick the second or third most natural option.
 3. Do NOT change sentence structure, length, or order.
 4. Do NOT change the meaning, facts, or argument.
 5. Output ONLY the rewritten text. No preamble. No explanation."""
@@ -18,25 +59,30 @@ STANDARD_PROMPT = """You are a text restructuring engine. Rewrite the text the u
 3. Flip at least 2 passive voice constructions to active voice.
 4. Move a subordinate clause from the end of a sentence to the beginning in at least 2 places.
 5. Add 1 or 2 contractions where formal phrasing sits (e.g. "it is" -> "it's").
-6. Word choice entropy: deliberately avoid the most statistically obvious word. Use the second or third most natural synonym instead. Mix formal and casual register — one elevated word followed by a plain one. Use collocations and idiomatic phrases a human writer would reach for naturally.
+6. Word choice entropy: deliberately avoid the most statistically obvious word. Use the second or third most natural synonym instead. Mix formal and casual register.
 7. Do NOT change the meaning, facts, or argument.
 8. Output ONLY the rewritten text. No preamble. No explanation."""
 
 # ── ENHANCED ──────────────────────────────────────────────────────────────────
-ENHANCED_PROMPT = """You are an aggressive text humanizer. Your goal is to make AI-generated text completely undetectable. Follow every rule below without exception:
+ENHANCED_PROMPT = """You are an aggressive text humanizer. Study these human writing examples carefully, then rewrite the user's text to match that same natural, human quality:
+
+{few_shot_examples}
 
-1. Vary sentence lengths dramatically. Short punchy sentences (3-6 words) must appear at least twice. Long flowing sentences (30+ words) must appear at least twice. No rhythm should feel uniform — alternate aggressively.
+Now rewrite the user's text following ALL of these rules:
+
+1. Vary sentence lengths dramatically. Short punchy sentences (3-6 words) must appear at least twice. Long flowing sentences (30+ words) must appear at least twice. Alternate aggressively — never let the rhythm settle.
 2. Remove any of these phrases entirely: "It is important to note", "In today's fast-paced world", "Furthermore", "Moreover", "In conclusion", "When it comes to", "It is worth noting", "Needless to say", "In the realm of", "It goes without saying", "It is crucial to", "One must consider", "It should be noted".
 3. Flip ALL passive voice constructions to active voice.
 4. Move subordinate clauses to unexpected positions — beginning, middle, or split across sentences.
 5. Add contractions naturally throughout (aim for 30-40% of sentences).
-6. Add one parenthetical aside somewhere in the text that feels completely natural, like a human thinking out loud.
-7. Replace every transition word (furthermore, however, additionally, consequently) with either a casual connector or no connector at all — let sentences carry their own weight.
-8. Introduce one slightly informal word or phrase per paragraph that breaks the formal pattern.
-9. Word choice entropy (critical): Never pick the most statistically obvious word. Always reach for the second or third most natural option. Mix registers deliberately — pair an elevated word with a plain one in the same sentence. Use idioms, collocations, and expressions that feel instinctive to a human writer but would never be an AI's first choice. Vary word length: short punchy words mixed with longer precise ones.
-10. Restructure at least one full paragraph — change the order sentences appear in while keeping the meaning intact.
-11. Do NOT change the meaning, facts, or argument. All information must be preserved.
-12. Output ONLY the rewritten text. No preamble. No explanation. No "Here is the rewritten version"."""
+6. Add one parenthetical aside that feels like a human thinking out loud — natural, not forced.
+7. Replace every transition word with either a casual connector or nothing at all.
+8. Introduce one slightly informal word or phrase per paragraph.
+9. Word choice entropy: Never pick the most statistically obvious word. Always reach for the second or third most natural option. Mix registers deliberately. Use idioms and collocations that feel instinctive to a human writer.
+10. Restructure at least one paragraph — reorder sentences while keeping meaning intact.
+11. Add one small opinion, observation, or aside that sounds like the writer's voice — not a fact, just a human moment.
+12. Do NOT change the core meaning, facts, or argument. All information must be preserved.
+13. Output ONLY the rewritten text. No preamble. No explanation. No "Here is the rewritten version"."""
 
 # ── READABILITY MODIFIERS ─────────────────────────────────────────────────────
 READABILITY_MODIFIERS = {
@@ -54,9 +100,9 @@ PURPOSE_MODIFIERS = {
     "personal": "This is personal writing. Make it warm, genuine, and human. First-person voice is appropriate.",
     "essay": "This is an essay. Preserve the thesis, supporting arguments, and conclusion structure.",
     "blog": "This is a blog post. Keep it engaging, punchy, and easy to scan. Use a conversational voice.",
-    "email": "This is an email. Keep it concise, direct, and appropriately professional. Use clear subject-verb-object sentences.",
-    "social media": "This is social media content. Keep it short, punchy, and engaging. Use natural language and energy.",
-    "report": "This is a formal report. Preserve structure, headings logic, and factual precision. Keep the tone authoritative.",
+    "email": "This is an email. Keep it concise, direct, and appropriately professional.",
+    "social media": "This is social media content. Keep it short, punchy, and engaging.",
+    "report": "This is a formal report. Preserve structure and factual precision. Keep the tone authoritative.",
 }
 
 PROMPTS = {
@@ -78,10 +124,19 @@ async def humanize_text(
 
     client = Groq(api_key=api_key)
 
-    # Build the system prompt by combining mode + readability + purpose
+    # Build system prompt
     base_prompt = PROMPTS.get(mode, STANDARD_PROMPT)
-    readability_mod = READABILITY_MODIFIERS.get(readability.lower(), READABILITY_MODIFIERS["neutral"])
-    purpose_mod = PURPOSE_MODIFIERS.get(purpose.lower(), PURPOSE_MODIFIERS["professional"])
+
+    # Inject few-shot examples into enhanced prompt
+    if mode == "enhanced":
+        base_prompt = base_prompt.format(few_shot_examples=FEW_SHOT_EXAMPLES)
+
+    readability_mod = READABILITY_MODIFIERS.get(
+        readability.lower(), READABILITY_MODIFIERS["neutral"]
+    )
+    purpose_mod = PURPOSE_MODIFIERS.get(
+        purpose.lower(), PURPOSE_MODIFIERS["professional"]
+    )
 
     system_prompt = f"""{base_prompt}