Update app.py

app.py

@@ -37,9 +37,8 @@ def analyze_prompt(user_prompt):
 
     # Step 2: Suggest a safer reformulation of the prompt (conditional)
     rewrite_prompt = f"""
-    You are an AI security assistant. 
-    Based on the following analysis of a user prompt, output a safer version of the prompt ONLY if the risk level is concerning.
-    If the analysis indicates no risk, simply return the original prompt as-is. 
+    You are an AI security assistant.
+    Here is the risk analysis of a user prompt and the prompt itself.
     
     Analysis Result:
     {analysis}
@@ -47,7 +46,10 @@ def analyze_prompt(user_prompt):
     Original Prompt:
     {user_prompt}
     
-    Output ONLY the rewritten prompt text (or the original prompt if safe), without explanations, notes, or extra words.
+    Your task:
+    - If the analysis indicates risk (Low or Medium or High), rewrite the prompt so it becomes a safe, educational question about the same topic.
+    - If the analysis indicates no risk, return the original prompt unchanged.
+    - Output ONLY the final safe prompt text, with no explanations, notes, or extra words.
     """
 
     rewrite_response = client.chat.completions.create(