module import solutions

piilo/main.py → main.py

@@ -1,12 +1,14 @@
 '''API for PIILO'''
 
-from analyzer import prepare_analyzer
-from anonymizer import surrogate_anonymizer
-from models.anonymize import AnonymizeRequest, AnonymizeResponse
-from fastapi import FastAPI
 import logging
+
+from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
+from piilo.analyzer import prepare_analyzer
+from piilo.anonymizer import surrogate_anonymizer
+from piilo.models.anonymize import AnonymizeRequest, AnonymizeResponse
+
 # Define Student Name Detection Model
 configuration = {
     'nlp_engine_name': 'spacy',
@@ -59,8 +61,9 @@ def anonymize(anon_req: AnonymizeRequest) -> AnonymizeResponse:
     return anonymize_response
 
 if __name__ == "__main__":
-    import uvicorn
     import os
+
+    import uvicorn
     
     uvicorn.run(
         "main:app", host="0.0.0.0",

setup.py

@@ -1,11 +0,0 @@
-from setuptools import setup, find_packages
-
-setup(
-    name='piilo',
-    version='0.1',
-    packages=find_packages(),
-    install_requires=[
-        nameparser,
-        presidio_analyzer
-    ],
-)

piilo/test_main.py → test_main.py

@@ -1,7 +1,8 @@
+import logging
+
 from fastapi.testclient import TestClient
 
 from main import app
-import logging
 
 logger = logging.getLogger('api')
 logging.basicConfig(level=logging.INFO)