Update app.py

app.py

@@ -6,9 +6,8 @@ client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 def analyze_prompt(user_prompt):
     if not user_prompt.strip():
-        return "⚠️ Please enter a prompt to analyze.", "", "", ""
+        return "Please enter a prompt to analyze.", "", "", ""
 
-    # Step 1: Detect vulnerability type
     detection_prompt = f"""
     You are an AI security analyst. 
     Analyze the following prompt for potential vulnerabilities against large language models. 
@@ -23,7 +22,7 @@ def analyze_prompt(user_prompt):
 
     - Detected Vulnerability: <list types or "None">
     - Risk Level: <Low / Medium / High>
-    - Short Explanation (1–2 sentences)
+    - Short Explanation (1 sentence)
 
     Prompt to analyze:
     {user_prompt}
@@ -36,11 +35,10 @@ def analyze_prompt(user_prompt):
 
     analysis = detection_response.choices[0].message.content.strip()
 
-    # Step 2: Suggest a safer reformulation of the prompt
     rewrite_prompt = f"""
-    Rewrite the following prompt in a safer and responsible way 
-    that removes any security risks but keeps the learning intent.
-
+    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.
+    
     Prompt:
     {user_prompt}
     """