full text ner output

interfaces/ner.py

@@ -33,10 +33,11 @@ def named_entity_recognition(text, language):
     model_id = build_spacy_path(language)
     pipeline = huspacy.load() if model_id.startswith("hu") else spacy.load(model_id)
     doc = pipeline(text)
-    entities = [(ent.text, ent.label_) for ent in doc.ents]
+    entities = [{"entity":ent.label_, "start":ent.start_char, "end":ent.end_char} for ent in doc.ents]
+    output = {"text":text, "entities":entities}
     model_id_hf = f"huspacy/{model_id}" if model_id.startswith("hu") else f"spacy/{model_id}"
     output_info = f'<p style="text-align: center; display: block">Prediction was made using the <a href="https://huggingface.co/{model_id_hf}">{model_id_hf}</a> model.</p>'
-    return entities, output_info
+    return output, output_info
 
 demo = gr.Interface(
     fn=named_entity_recognition,