Update paraphraser.py

paraphraser.py

@@ -2,13 +2,28 @@
 import torch
 from model_loader import paraphrase_model, paraphrase_tokenizer
 
-def paraphrase_comment(comment, prompt_template="Paraphrase the following comment to make it non-toxic while preserving its meaning: \"{comment}\""):
+def paraphrase_comment(comment):
     """
     Paraphrase a toxic comment using the Granite 3.2-2B-Instruct model.
     Returns the paraphrased comment.
     """
+    # Define the paraphrasing prompt with system instruction, guidelines, examples, and the task
+    prompt = (
+        "You are a content moderator tasked with rewriting toxic comments into neutral and constructive ones while maintaining the original meaning.\n"
+        "Guidelines:\n"
+        "- Remove explicit hate speech, personal attacks, or offensive language.\n"
+        "- Keep the response neutral and professional.\n"
+        "- Ensure the rewritten comment retains the original intent but in a constructive tone.\n"
+        "Examples:\n"
+        "Toxic: \"You're so dumb! You never understand anything!\"\n"
+        "Neutral: \"I think there's some misunderstanding. Let's clarify things.\"\n"
+        "Toxic: \"This is the worst idea ever. Only an idiot would suggest this.\"\n"
+        "Neutral: \"I don't think this idea works well. Maybe we can explore other options.\"\n"
+        "Now, rewrite this comment: \"{comment}\""
+    )
+
     # Format the prompt with the input comment
-    prompt = prompt_template.format(comment=comment)
+    prompt = prompt.format(comment=comment)
 
     # Tokenize the prompt
     inputs = paraphrase_tokenizer(prompt, return_tensors="pt", truncation=True, padding=True, max_length=512)