Update app.py

app.py

@@ -5,7 +5,19 @@ pipe = JailbreakPipeline(consider_output=False)
 
 def analyze(prompt: str):
     r = pipe.process(prompt)
-    return r["risk_score"], ", ".join(r["fired_rules"]), r["safe_output"]
+    fired = r.get("fired_rules", [])
+
+    # fired_rules can be ["rule1", "rule2"] OR [{"name": "rule1"}, ...]
+    if fired and isinstance(fired[0], dict):
+        fired_text = ", ".join(
+            str(item.get("name") or item.get("rule") or item.get("id") or item)
+            for item in fired
+        )
+    else:
+        fired_text = ", ".join(str(x) for x in fired)
+
+    return r["risk_score"], fired_text, r["safe_output"]
+
 
 with gr.Blocks() as demo:
     gr.Markdown("# JailBreakDefense – Prompt Jailbreak Detector")