update app

app.py

@@ -16,23 +16,25 @@ ner_pipeline = pipeline(
 )
 
 
+# Helper function to print entities nicely
+def print_nicely(entities):
+    entity_details = []
+    for entity in entities:
+        entity_info = f"Entity: {entity['entity']} | Confidence: {entity['score']:.2f}% | Text: {entity['word'].strip()} | Start: {entity['start']} | End: {entity['end']}"
+        entity_details.append(entity_info)
+    return "\n".join(entity_details)
+
+
 # Function to process the sentence and extract entities
 def extract_entities(sentence):
     results = ner_pipeline(sentence)
-    entities_with_confidences = []
+    entity_results = []
 
-    # Extract and format the entities for highlighting
-    for entity in results:
-        entities_with_confidences.append(
-            (
-                entity["word"],
-                entity["start"],
-                entity["end"],
-                f"{entity['entity']} ({entity['score']:.2f}%)",
-            )
-        )
+    # Extract and format the entities
+    for key in results.keys():
+        entity_results.append(print_nicely(results[key]))
 
-    return {"text": sentence, "entities": entities_with_confidences}
+    return "\n".join(entity_results)
 
 
 # Create Gradio interface
@@ -40,7 +42,7 @@ def ner_app_interface():
     input_sentence = gr.Textbox(
         lines=5, label="Input Sentence", placeholder="Enter a sentence for NER..."
     )
-    output_entities = gr.HighlightedText(label="Extracted Entities")
+    output_entities = gr.Textbox(label="Extracted Entities")
 
     # Interface definition
     interface = gr.Interface(
@@ -49,11 +51,6 @@ def ner_app_interface():
         outputs=output_entities,
         title="Named Entity Recognition",
         description="Enter a sentence to extract named entities using the NER model from the Impresso project.",
-        examples=[
-            [
-                "In the year 1789, King Louis XVI, ruler of France, convened the Estates-General at the Palace of Versailles."
-            ]
-        ],
     )
 
     interface.launch()