Add readability and purpose controls to humanizer

humanizer.py

@@ -1,33 +1,63 @@
 import os
 from groq import Groq
 
+# ── 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".
-2. Replace obvious synonym swaps where words feel robotic or overly formal.
+
+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.
 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."""
 
+# ── STANDARD ──────────────────────────────────────────────────────────────────
 STANDARD_PROMPT = """You are a text restructuring engine. Rewrite the text the user sends with these rules:
+
 1. Vary sentence lengths. Mix short sentences (under 8 words) with longer ones (over 25 words). Never have 3 sentences of similar length in a row.
-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".
+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 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. Do NOT change the meaning, facts, or argument.
-7. Output ONLY the rewritten text. No preamble. No explanation."""
-
-ENHANCED_PROMPT = """You are an aggressive text humanizer. Rewrite the text the user sends to sound completely natural and human. Follow all of these rules:
-1. Vary sentence lengths dramatically. Short punchy sentences (3-6 words) must appear. Long flowing sentences (30+ words) must appear. No rhythm should feel uniform.
-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".
-3. Flip all passive voice constructions to active voice.
-4. Move subordinate clauses to unexpected positions.
-5. Add contractions naturally throughout (aim for 30% of sentences).
-6. Add one parenthetical aside somewhere in the text that feels natural.
-7. Replace at least one transition word per paragraph with a casual connector or none at all.
-8. Introduce one slightly informal word choice per paragraph.
-9. Do NOT change the meaning, facts, or argument. All information must be preserved.
-10. Output ONLY the rewritten text. No preamble. No explanation."""
+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.
+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:
+
+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.
+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"."""
+
+# ── READABILITY MODIFIERS ─────────────────────────────────────────────────────
+READABILITY_MODIFIERS = {
+    "casual": "Write in a relaxed, conversational tone. Use everyday language, short sentences, and contractions freely. Write like you're talking to a friend.",
+    "neutral": "Write in a balanced, natural tone. Neither too formal nor too casual. Aim for clear, readable prose.",
+    "formal": "Write in a polished, professional tone. Use precise language and complete sentences. Avoid contractions and slang.",
+    "academic": "Write in a scholarly tone appropriate for academic papers. Use discipline-specific vocabulary where natural, complex sentence structures, and avoid contractions entirely.",
+}
+
+# ── PURPOSE MODIFIERS ─────────────────────────────────────────────────────────
+PURPOSE_MODIFIERS = {
+    "academic": "This is academic writing. Preserve citations, arguments, and scholarly structure. Prioritize clarity and precision.",
+    "persuasive": "This is persuasive writing. Keep the argument sharp, the tone confident, and the call to action clear.",
+    "professional": "This is professional business writing. Keep it clear, concise, and action-oriented.",
+    "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.",
+}
 
 PROMPTS = {
     "simple": SIMPLE_PROMPT,
@@ -36,13 +66,28 @@ PROMPTS = {
 }
 
 
-async def humanize_text(text: str, mode: str = "standard") -> str:
+async def humanize_text(
+    text: str,
+    mode: str = "standard",
+    readability: str = "neutral",
+    purpose: str = "professional",
+) -> str:
     api_key = os.getenv("GROQ_API_KEY")
     if not api_key:
         raise ValueError("GROQ_API_KEY not set in environment variables.")
 
     client = Groq(api_key=api_key)
-    system_prompt = PROMPTS.get(mode, STANDARD_PROMPT)
+
+    # Build the system prompt by combining mode + readability + purpose
+    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"])
+
+    system_prompt = f"""{base_prompt}
+
+Tone guidance: {readability_mod}
+
+Context: {purpose_mod}"""
 
     response = client.chat.completions.create(
         model="llama-3.1-8b-instant",
@@ -50,7 +95,7 @@ async def humanize_text(text: str, mode: str = "standard") -> str:
             {"role": "system", "content": system_prompt},
             {"role": "user", "content": text}
         ],
-        temperature=0.7,
+        temperature=0.85,
         max_tokens=2000,
     )
 

main.py

@@ -27,10 +27,14 @@ app.add_middleware(
 class HumanizeRequest(BaseModel):
     text: str
     mode: str = "standard"
+    readability: str = "neutral"
+    purpose: str = "professional"
 
 class HumanizeResponse(BaseModel):
     humanized: str
     mode: str
+    readability: str
+    purpose: str
 
 @app.post("/humanize", response_model=HumanizeResponse)
 async def humanize(request: HumanizeRequest):
@@ -40,8 +44,18 @@ async def humanize(request: HumanizeRequest):
             detail="mode must be one of: simple, standard, enhanced"
         )
     try:
-        humanized_text = await humanize_text(request.text, mode=request.mode)
-        return HumanizeResponse(humanized=humanized_text, mode=request.mode)
+        humanized_text = await humanize_text(
+            request.text,
+            mode=request.mode,
+            readability=request.readability,
+            purpose=request.purpose,
+        )
+        return HumanizeResponse(
+            humanized=humanized_text,
+            mode=request.mode,
+            readability=request.readability,
+            purpose=request.purpose,
+        )
     except Exception as e:
         import traceback
         raise HTTPException(status_code=500, detail=traceback.format_exc())