Design

app.py

@@ -2,13 +2,11 @@ import gradio as gr
 import os
 from transformers import pipeline
 
-# Konfiguration
 MODEL_1_ID = "patronus-protect/wolf-guard"
 MODEL_2_ID = "HuggingLil/pii-sensitive-ner-german"
 TOKEN = os.getenv("HF_TOKEN")
 CLS_MAP = {"LABEL_0": "Benign", "LABEL_1": "Attack"}
 
-# Pipelines laden (einfachster Weg für verschiedene Modelltypen)
 pipe_cls = pipeline("text-classification", model=MODEL_1_ID, token=TOKEN)
 pipe_ner = pipeline("ner", model=MODEL_2_ID, token=TOKEN, aggregation_strategy="simple")
 
@@ -25,7 +23,7 @@ def analyze(text):
         return (
             gr.update(visible=True), 
             "Prompt Injection detected", 
-            [(text, "ATTACK")]
+            [(text, "ATTACK")] # Text komplett rot markieren
         )
     
     has_high_ner = False
@@ -38,7 +36,7 @@ def analyze(text):
             has_high_ner = True
 
             highlighted_data.append((text[last_idx:ent['start']], None))
-            highlighted_data.append((text[ent['start']:ent['end']], f"NER ({ent['score']:.2%})"))
+            highlighted_data.append((text[ent['start']:ent['end']], f"{ent['entity_group']} ({ent['score']:.2%})"))
             last_idx = ent['end']
     
     highlighted_data.append((text[last_idx:], None))
@@ -48,31 +46,38 @@ def analyze(text):
     else:
         return gr.update(visible=True), "Nothing detected", [(text, "CLEAN")]
 
-# Custom CSS für die Farben (Rot, Orange, Grün)
+ACCENT_COLOR = "#F5C77A"
+
+theme = gr.themes.Soft()
+theme.font = [gr.themes.GoogleFont("Inter"), "sans-serif"]
+
 css = """
 .color-attack { background-color: #ffcccb !important; }
 .color-ner { background-color: #ffe5b4 !important; }
 .color-clean { background-color: #d1ffbd !important; }
 footer {display: none !important;}
+
+.gradio-container {border: none !important;}
+.generating { border-color: """ + ACCENT_COLOR + """ !important; }
 """
 
-with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
+with gr.Blocks(css=css, theme=theme) as demo:
     input_text = gr.Textbox(label="Input Scan", placeholder="Text eingeben...", lines=3)
     
     with gr.Column(visible=False) as results_col:
-        status_msg = gr.Markdown()
         display_output = gr.HighlightedText(
-            label="Analyse Ergebnis",
+            label="Result",
             combine_adjacent=True,
             show_legend=False,
-            color_map={"ATTACK": "red", "CLEAN": "green", "NER": "orange"}
+            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="hidden"
+        show_progress="minimal"
     )
 
 demo.launch()