move main execution to __main__

Signed-off-by: Michele Dolfi <[email protected]>

Containerfile

@@ -58,4 +58,4 @@ COPY --chown=1001:0 --chmod=664 ./docling_serve ./docling_serve
 
 EXPOSE 5001
 
-CMD ["python", "docling_serve/app.py"]
+CMD ["python", "-m", "docling_serve"]

docling_serve/__main__.py

@@ -0,0 +1,20 @@
+import os
+
+from docling_serve.app import app
+from docling_serve.helper_functions import _str_to_bool
+
+# Launch the FastAPI server
+if __name__ == "__main__":
+    from uvicorn import run
+
+    port = int(os.getenv("PORT", "5001"))
+    workers = int(os.getenv("UVICORN_WORKERS", "1"))
+    reload = _str_to_bool(os.getenv("RELOAD", "False"))
+    run(
+        app,
+        host="0.0.0.0",
+        port=port,
+        workers=workers,
+        timeout_keep_alive=600,
+        reload=reload,
+    )

docling_serve/app.py

@@ -222,20 +222,3 @@ async def process_file(
     )
 
     return response
-
-
-# Launch the FastAPI server
-if __name__ == "__main__":
-    from uvicorn import run
-
-    port = int(os.getenv("PORT", "5001"))
-    workers = int(os.getenv("UVICORN_WORKERS", "1"))
-    reload = _str_to_bool(os.getenv("RELOAD", "False"))
-    run(
-        "app:app",
-        host="0.0.0.0",
-        port=port,
-        workers=workers,
-        timeout_keep_alive=600,
-        reload=reload,
-    )