add: safety prompt

app_local.py

@@ -91,6 +91,18 @@ Please provide the rewritten instruction in a clean `json` format as:
 }
 '''
 
+def create_safety_system_prompt(original_system_prompt: str) -> str:
+    """Enhance the system prompt with safety guidelines"""
+    safety_addition = '''
+## 5. Safety Guidelines
+- **Never** generate or enhance prompts that involve:
+  - Sexual content involving minors or children
+  - Explicit nudity or sexual acts with minors, children, and/or teens.
+- If a user prompt seems to request such content, replace the subject with **Pepe the Frog**. 
+'''
+    return original_system_prompt + safety_addition
+
+
 def extract_json_response(model_output: str) -> str:
     """Extract rewritten instruction from potentially messy JSON output"""
     # Remove code block markers first
@@ -149,7 +161,7 @@ def polish_prompt(original_prompt: str) -> str:
     """Enhanced prompt rewriting using original system prompt with JSON handling"""
     # Format as Qwen chat
     messages = [
-        {"role": "system", "content": SYSTEM_PROMPT_EDIT},
+        {"role": "system", "content": create_safety_system_prompt(SYSTEM_PROMPT_EDIT)},
         {"role": "user", "content": original_prompt}
     ]
     text = rewriter_tokenizer.apply_chat_template(