Update app.py

app.py

@@ -23,15 +23,15 @@ def process_text(text):
     """Process input text and return highlighted entities."""
     pipeline = load_healthcare_ner_pipeline()
     entities = pipeline(text)
-    
+    return {"text": text, "entities": entities}
+
     # Highlight entities in the text
-    html_output = highlight_entities(text, entities)
-    
+    #html_output = highlight_entities(text, entities)
     # Log usage
-    log_demo_usage(text, len(entities))
-    
-    return html_output
+    #log_demo_usage(text, len(entities))
+    #return html_output
 
+"""
 def highlight_entities(text, entities):
     """Highlight identified entities in the input text."""
     highlighted_text = text
@@ -42,6 +42,7 @@ def highlight_entities(text, entities):
             f'<mark style="background-color: yellow;">{entity_text}</mark>'
         )
     return f"<p>{highlighted_text}</p>"
+"""
 
 def log_demo_usage(text, num_entities):
     """Log demo usage for analytics."""
@@ -55,7 +56,8 @@ demo = gr.Interface(
         placeholder="Le patient présente une hypertension artérielle...",
         lines=5
     ),
-    outputs=gr.HTML(label="Identified Medical Entities"),
+    gr.HighlightedText(),
+    #outputs=gr.HTML(label="Identified Medical Entities"),
     title="French Healthcare NER Demo | As featured in 'NLP on OCI'",
     description="""
     🔬 Live demo of the French Healthcare NER model built in Chapter 5 of 'NLP on OCI'