Upload app.py

app.py

@@ -53,6 +53,7 @@ def verify_token(x_api_key: str = Depends(token_header)):
         raise HTTPException(status_code=403, detail="Unauthorized")
     return x_api_key
 
+"""
 @app.post("/ner")
 @limiter.limit("10/minute")
 def ner_predict(request: TextRequest, api_key: str = Depends(verify_token), req: Request = None):
@@ -64,6 +65,27 @@ def ner_predict(request: TextRequest, api_key: str = Depends(verify_token), req:
         "entities": predictions,
         "usage": API_KEYS[api_key]["usage_count"]
     }
+"""
+
+from fastapi import Request
+
+@app.post("/ner")
+@limiter.limit("10/minute")
+def ner_predict(
+    body: TextRequest, 
+    req: Request, 
+    api_key: str = Depends(verify_token)
+):
+    logger.info("NER request from IP: %s", get_remote_address(req))
+    predictions = ner_pipeline(body.text)
+    API_KEYS[api_key]["usage_count"] = API_KEYS[api_key].get("usage_count", 0) + 1
+    save_keys()
+    return {
+        "entities": predictions,
+        "usage": API_KEYS[api_key]["usage_count"]
+    }
+
+
 
 @app.post("/register_user")
 def register_user(request: RegisterRequest, x_api_key: str = Depends(token_header)):