wqewqewqe

app.py

@@ -1,54 +1,51 @@
 import re
 import gradio as gr
 
-# Simulated GLiNER output
-SIMULATED_ENTITIES = {
+# Simulated GLiNER‐style output
+SIM_ENTITIES = {
     "Household Survey Data": "named dataset",
     "Ethiopia":             "data geography",
     "2020":                 "reference year",
     "World Bank":           "publisher"
 }
 
-def highlight_text(text):
-    """
-    Return a dict with the original text and a list of start/end/label spans.
-    """
+def annotate_text(text):
     entities = []
-    for phrase, label in SIMULATED_ENTITIES.items():
-        # find all occurrences of phrase
+    for phrase, label in SIM_ENTITIES.items():
         for m in re.finditer(re.escape(phrase), text):
             entities.append({
-                "start": m.start(),
-                "end":   m.end(),
-                "label": label
+                "entity": label,    # note: 'entity' key
+                "start":  m.start(),
+                "end":    m.end(),
             })
     return {"text": text, "entities": entities}
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Simulated GLiNER Entity Highlighter\n"
-                "Type a sentence (or use the default) then hit **Highlight Entities** or press Enter.")
-    
-    text_input = gr.Textbox(
-        label="Input Text",
+    gr.Markdown("## Simulated GLiNER Highlighter\n"
+                "Type or paste a sentence and click **Highlight**.")
+
+    txt = gr.Textbox(
         lines=3,
-        value="The Household Survey Data for Ethiopia in 2020 was published by the World Bank."
+        value="The Household Survey Data for Ethiopia in 2020 was published by the World Bank.",
+        label="Input Text",
+        placeholder="Enter any sentence here…"
     )
-    highlight_btn = gr.Button("Highlight Entities")
-    highlighted   = gr.HighlightedText(label="Highlighted Entities")
-    
-    # Wire up both the button and pressing Enter to our highlighter
-    highlight_btn.click(fn=highlight_text, inputs=text_input, outputs=highlighted)
-    text_input.submit(fn=highlight_text, inputs=text_input, outputs=highlighted)
-    # Also do an initial run on load so you immediately see highlights
-    demo.load(fn=highlight_text, inputs=text_input, outputs=highlighted)
-    
+    btn = gr.Button("Highlight")
+    out = gr.HighlightedText(label="Annotated Entities")
+
+    # fire on button click or Enter
+    btn.click(fn=annotate_text, inputs=txt, outputs=out)
+    txt.submit(fn=annotate_text, inputs=txt, outputs=out)
+    # initial render
+    demo.load(fn=annotate_text, inputs=txt, outputs=out)
+
     gr.Markdown("""
-**Entity Legend**  
-- **named dataset**: Household Survey Data  
-- **data geography**: Ethiopia  
-- **reference year**: 2020  
-- **publisher**: World Bank
+**Legend**  
+- **named dataset** → Household Survey Data  
+- **data geography** → Ethiopia  
+- **reference year** → 2020  
+- **publisher** → World Bank
 """)
 
 if __name__ == "__main__":
-    demo.launch(inline=True)
+    demo.launch(debug=True)