UX

app.py

@@ -12,7 +12,7 @@ pipe_ner = pipeline("ner", model=MODEL_2_ID, token=TOKEN, aggregation_strategy="
 
 def analyze(text):
     if not text or len(text.strip()) < 1:
-        return gr.update(visible=False), "", ""
+        return gr.update(visible=False), ""
 
     res_cls = pipe_cls(text)[0]
     is_attack = res_cls['label'] == "LABEL_1" 
@@ -21,9 +21,8 @@ def analyze(text):
     
     if is_attack:
         return (
-            gr.update(visible=True), 
-            "Prompt Injection detected", 
-            [(text, "ATTACK")] # Text komplett rot markieren
+            gr.update(visible=True),
+            [(text, "ATTACK")]
         )
     
     has_high_ner = False
@@ -42,9 +41,9 @@ def analyze(text):
     highlighted_data.append((text[last_idx:], None))
 
     if has_high_ner:
-        return gr.update(visible=True), "Entities detected", highlighted_data
+        return gr.update(visible=True), highlighted_data
     else:
-        return gr.update(visible=True), "Nothing detected", [(text, "CLEAN")]
+        return gr.update(visible=True), [(text, "CLEAN")]
 
 ACCENT_COLOR = "#F5C77A"
 
@@ -71,13 +70,14 @@ with gr.Blocks(css=css, theme=theme) as demo:
             show_legend=False,
             color_map={"ATTACK": "red", "CLEAN": "green"}
         )
-        status_msg = gr.Markdown()
 
     input_text.change(
         fn=analyze,
         inputs=input_text,
-        outputs=[results_col, status_msg, display_output],
-        show_progress="minimal"
+        outputs=[results_col, display_output],
+        show_progress="minimal",
+        concurrency_limit=1,
+        trigger_mode="always_last"
     )
 
 demo.launch()