Update app.py

app.py

@@ -87,31 +87,47 @@ class HumanizeRequest(BaseModel):
 # ==================== STYLE PROMPTS ====================
 STYLE_PROMPTS = {
     "professional": """<|system|>
-You are a professional writing assistant. Rewrite the following text in formal, corporate business language.
-IMPORTANT: Output ONLY the rewritten text, nothing else.<|end|>
+You are a text rewriter. Your ONLY task is to rewrite the user's text in professional business language.
+RULES: 
+1. Output ONLY the rewritten version
+2. No explanations, no examples, no notes
+3. Keep it brief - same length as input
+4. Just rewrite, don't add content<|end|>
 <|user|>
-Text to rewrite: {text}<|end|>
+Rewrite this professionally: {text}<|end|>
 <|assistant|>""",
 
     "casual": """<|system|>
-You are a casual writing assistant. Rewrite the following text in friendly, natural, conversational English.
-IMPORTANT: Output ONLY the rewritten text, nothing else.<|end|>
+You are a text rewriter. Your ONLY task is to rewrite the user's text in casual, conversational English.
+RULES: 
+1. Output ONLY the rewritten version
+2. No explanations, no examples, no notes
+3. Keep it brief - same length as input
+4. Just rewrite, don't add content<|end|>
 <|user|>
-Text to rewrite: {text}<|end|>
+Rewrite this casually: {text}<|end|>
 <|assistant|>""",
 
     "academic": """<|system|>
-You are an academic writing assistant. Rewrite the following text in formal scholarly language.
-IMPORTANT: Output ONLY the rewritten text, nothing else.<|end|>
+You are a text rewriter. Your ONLY task is to rewrite the user's text in academic language.
+RULES: 
+1. Output ONLY the rewritten version
+2. No explanations, no examples, no notes
+3. Keep it brief - same length as input
+4. Just rewrite, don't add content<|end|>
 <|user|>
-Text to rewrite: {text}<|end|>
+Rewrite this academically: {text}<|end|>
 <|assistant|>""",
 
     "marketing": """<|system|>
-You are a marketing copywriter. Rewrite the following text into persuasive marketing language.
-IMPORTANT: Output ONLY the rewritten text, nothing else.<|end|>
+You are a text rewriter. Your ONLY task is to rewrite the user's text in persuasive marketing language.
+RULES: 
+1. Output ONLY the rewritten version
+2. No explanations, no examples, no notes
+3. Keep it brief - same length as input
+4. Just rewrite, don't add content<|end|>
 <|user|>
-Text to rewrite: {text}<|end|>
+Rewrite this for marketing: {text}<|end|>
 <|assistant|>"""
 }
 
@@ -190,11 +206,13 @@ async def transform_with_model(text: str, style: str) -> str:
         
         output = llm(
             prompt,
-            max_tokens=min(200, len(text) + 50),
-            temperature=temperature,
-            top_p=0.95,
-            repeat_penalty=1.1,
-            stop=["<|end|>", "\n"],
+            max_tokens=150,           # REDUCE further - limit output length
+            temperature=0.4,          # Balanced creativity
+            top_p=0.9,
+            repeat_penalty=1.2,       # Higher penalty for repetition
+            frequency_penalty=0.2,    # NEW: Penalize common words
+            presence_penalty=0.1,     # NEW: Penalize already mentioned topics
+            stop=["<|end|>", "\n\n", "###", "Explanation:", "For example:", "Here's how:"],
             echo=False,
         )