Fix the dockerfile for now?

Breaks cache layers, after all using setup.py not the best idea?
We need the whole project to pip install .
properly, but then we copy it in Dockerfile too early

Dockerfile

@@ -13,17 +13,16 @@ ENV PYTHONUNBUFFERED=1
 RUN python -m venv venv
 ENV PATH="$HOME/comma-fixer/venv/bin:$PATH"
 
-COPY --chown=user setup.py .
+# TODO use requirements after all, since for setup.py to work properly we need the whole source code which breaks cache
+COPY --chown=user . .
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir --upgrade .
 
 # This pre-downloads models and tokenizers
-COPY --chown=user commafixer/src/ commafixer/src/
+# TODO should we give user an option to provide local models so that they don't donwload each time?
 RUN python commafixer/src/baseline.py
 RUN python commafixer/src/fixer.py
 
-COPY --chown=user . .
-
 FROM base as test
 
 RUN pip install .[test]

app.py

@@ -1,8 +1,8 @@
-from os.path import realpath
 import uvicorn
 from fastapi import FastAPI
 from fastapi.responses import FileResponse
 from fastapi.staticfiles import StaticFiles
+from os.path import realpath
 
 from commafixer.routers import baseline, fixer
 

commafixer/routers/baseline.py

@@ -2,7 +2,7 @@ from fastapi import APIRouter
 import logging
 
 from commafixer.src.baseline import BaselineCommaFixer
-from common import fix_commas_request_handler
+from commafixer.routers.common import fix_commas_request_handler
 
 logger = logging.Logger(__name__)
 logging.basicConfig(level=logging.INFO)

commafixer/routers/common.py

@@ -1,7 +1,7 @@
 from fastapi import HTTPException
 from logging import Logger
 
-from comma_fixer_interface import CommaFixerInterface
+from commafixer.src.comma_fixer_interface import CommaFixerInterface
 
 
 def fix_commas_request_handler(

commafixer/routers/fixer.py

@@ -1,4 +1,4 @@
-from fastapi import APIRouter, HTTPException
+from fastapi import APIRouter
 import logging
 
 from commafixer.src.fixer import CommaFixer

commafixer/src/baseline.py

@@ -1,7 +1,7 @@
 from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline, NerPipeline
 import re
 
-from comma_fixer_interface import CommaFixerInterface
+from commafixer.src.comma_fixer_interface import CommaFixerInterface
 
 
 class BaselineCommaFixer(CommaFixerInterface):

commafixer/src/fixer.py

@@ -3,7 +3,7 @@ from transformers import AutoTokenizer, AutoModelForTokenClassification, Roberta
 import nltk
 import re
 
-from comma_fixer_interface import CommaFixerInterface
+from commafixer.src.comma_fixer_interface import CommaFixerInterface
 
 
 class CommaFixer(CommaFixerInterface):