Update app.py

app.py

@@ -35,12 +35,19 @@ def analyze_prompt(user_prompt):
 
     analysis = detection_response.choices[0].message.content.strip()
 
+    # Step 2: Suggest a safer reformulation of the prompt (conditional)
     rewrite_prompt = f"""
-    Rewrite the following prompt into a safer version that removes any security risks. 
-    Output ONLY the rewritten prompt text, without any explanations, notes, or extra words.
+    You are an AI security assistant. 
+    Based on the following analysis of a user prompt, output a safer version of the prompt ONLY if it is risky.
+    If the analysis indicates no risk, simply return the original prompt as-is. 
     
-    Prompt:
+    Analysis Result:
+    {analysis}
+    
+    Original Prompt:
     {user_prompt}
+    
+    Output ONLY the rewritten prompt text (or the original prompt if safe), without explanations, notes, or extra words.
     """
 
     rewrite_response = client.chat.completions.create(