PrivacyFilterAPI.py

import time
import uvicorn
from fastapi import FastAPI
from PrivacyFilter import PrivacyFilter
from pydantic import BaseModel

privacyFilterApp = FastAPI()

pfilter = PrivacyFilter()
pfilter.initialize(clean_accents=True, nlp_filter=True, wordlist_filter=False, regular_expressions=True)


class Parameter(BaseModel):
    text: str
    use_nlp: bool


@privacyFilterApp.get("/")
async def root():
    return {"message": "Welcome, please navigate to /filter"}


@privacyFilterApp.post("/filter")
async def filtertext(item: Parameter):
    starttime = time.time()
    filtered_text = pfilter.filter(item.text, nlp_filter=item.use_nlp)
    return {"text": filtered_text,
            "time": time.time() - starttime}


if __name__ == '__main__':
    uvicorn.run("PrivacyFilterAPI:privacyFilterApp",
                host="0.0.0.0",
                port=8000,
                reload=True,
                ssl_keyfile="./key.pem",
                ssl_certfile="./cert.pem"
                )