Fin app file

app.py

@@ -17,7 +17,7 @@ example_list = [
     "Apple Inc. was created by Steve Jobs in Cupertino.",
 ]
 
-# Correct mapping from LABEL_X to entity names
+# Correct mapping from LABEL_X to real entity names
 ENTITY_MAP = {
     "LABEL_0": "O",
     "LABEL_1": "PER",
@@ -30,14 +30,7 @@ ENTITY_MAP = {
     "LABEL_8": "MISC",
 }
 
-# Colors for HTML rendering
-COLOR_MAP = {
-    "PER": "#ff6666",   # Red
-    "ORG": "#66b3ff",   # Blue
-    "LOC": "#99ff99",   # Green
-    "MISC": "#ffcc66",  # Orange
-}
-
+# Model info for README block
 model_info = {
     model_name: {
         "link": f"https://huggingface.co/{model_name}",
@@ -49,6 +42,7 @@ print(ner(text))"""
     }
 }
 
+# Load NER pipeline
 ner_pipeline = pipeline("ner", model=model_name, aggregation_strategy="simple")
 
 
@@ -59,7 +53,7 @@ ner_pipeline = pipeline("ner", model=model_name, aggregation_strategy="simple")
 def display_model_info(model_name):
     info = model_info[model_name]
     usage_code = info["usage"]
-    model_link = f'[Open model page: {model_name}]({info["link"]})'
+    model_link = f"[Open model page: {model_name}]({info['link']})"
     return usage_code, model_link
 
 
@@ -89,38 +83,6 @@ def analyze_highlight(text):
     return highlighted
 
 
-# --------------------------------------
-# HTML COLOR MODE FUNCTION
-# --------------------------------------
-
-def colorize_text(text):
-    results = ner_pipeline(text)
-    html = ""
-    last = 0
-
-    for ent in results:
-        start, end = ent["start"], ent["end"]
-        raw_label = ent["entity_group"]
-        label = ENTITY_MAP.get(raw_label, raw_label)
-
-        if start > last:
-            html += text[last:start]
-
-        color = COLOR_MAP.get(label, "#dddddd")
-        html += (
-            f"<span style='background-color:{color}; padding:2px; border-radius:3px;'>"
-            f"{text[start:end]} ({label})"
-            f"</span>"
-        )
-
-        last = end
-
-    if last < len(text):
-        html += text[last:]
-
-    return html
-
-
 # --------------------------------------
 # GRADIO UI
 # --------------------------------------
@@ -136,47 +98,36 @@ with gr.Blocks() as demo:
         placeholder="Type a sentence to analyze."
     )
 
+    # Example buttons
     with gr.Row():
         for ex in example_list:
             gr.Button(ex).click(lambda x=ex: x, outputs=text_input)
 
-    mode_selector = gr.Dropdown(
-        choices=["Highlighted Mode", "Colorized HTML Mode"],
-        value="Highlighted Mode",
-        label="Output Display Mode"
-    )
-
+    # Run button
     analyze_button = gr.Button("Run NER Analysis")
 
+    # Only one output: HighlightedText
     output_highlight = gr.HighlightedText(
-        label="NER Result (Highlighted Mode)",
-        combine_adjacent=True,
-        visible=True
-    )
-    output_html = gr.HTML(
-        label="NER Result (Colorized HTML Mode)",
-        visible=False
+        label="NER Result",
+        combine_adjacent=True
     )
 
+    # Code + link
     code_output = gr.Code(label="How to use this model")
     link_output = gr.Markdown()
 
+    # Load model info on startup
     demo.load(
         fn=display_model_info,
         inputs=[gr.Textbox(value=model_name, visible=False)],
         outputs=[code_output, link_output]
     )
 
-    def run_analysis(text, mode):
-        if mode == "Highlighted Mode":
-            return analyze_highlight(text), gr.update(visible=True), gr.update(visible=False)
-        else:
-            return gr.update(), gr.update(visible=False), colorize_text(text)
-
+    # Run analysis
     analyze_button.click(
-        run_analysis,
-        inputs=[text_input, mode_selector],
-        outputs=[output_highlight, output_highlight, output_html]
+        analyze_highlight,
+        inputs=text_input,
+        outputs=output_highlight
     )
 
-demo.launch()
+demo.launch(ssr_mode=False)