simple commit

app.py

@@ -17,6 +17,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from presidio_analyzer import AnalyzerEngine
 from presidio_anonymizer import AnonymizerEngine
 
+from pydantic import BaseModel
 from fastapi import FastAPI, Request, UploadFile, File
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
@@ -36,6 +37,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+class TextItem(BaseModel):
+    text: str
+
 def create_nlp_engine_with_spacy(
     model_path: str = "en_core_web_lg",
 ):
@@ -179,7 +183,8 @@ async def parse_doc(file: UploadFile):
 
 
 @app.post("/presidio_mask")
-async def presidio_mask(text):
+async def presidio_mask(text: TextItem):
+    text = text.text
     results = analyzer.analyze(text=text, language='en')
     # for rec in results:
     #     print(rec.start)