Upload app.py

app.py

@@ -73,14 +73,16 @@ from fastapi import Request
 @app.post("/ner")
 @limiter.limit("10/minute")
 def ner_predict(body: TextRequest, request: Request, api_key: str = Depends(verify_token)):
-    logger.info("Received NER request from IP: %s", get_remote_address(request))
+    logger.info("NER request from IP: %s", get_remote_address(request))
+
     raw_predictions = ner_pipeline(body.text)
 
-    # Convert all numpy.float32 scores to float
+    # Use original character spans to extract clean entity words
     predictions = [
         {
             **ent,
-            "score": float(ent["score"])
+            "score": float(ent["score"]),
+            "word": body.text[ent["start"]:ent["end"]]  # ← this fixes the issue
         }
         for ent in raw_predictions
     ]
@@ -94,7 +96,6 @@ def ner_predict(body: TextRequest, request: Request, api_key: str = Depends(veri
     }
 
 
-
 @app.post("/register_user")
 def register_user(request: RegisterRequest, x_api_key: str = Depends(token_header)):
     if x_api_key != ADMIN_KEY: