Update Dockerfile

Dockerfile

@@ -18,6 +18,7 @@ COPY pyproject.toml poetry.lock ./
 RUN poetry config installer.max-workers 10
 RUN poetry install --with ui
 RUN poetry install --extras chroma
+RUN pip install fastapi uvicorn
 
 FROM base as app
 
@@ -30,18 +31,17 @@ RUN adduser --system worker
 WORKDIR /home/worker/app
 
 # Ensure the worker user has the necessary permissions
-RUN mkdir -p local_data/private_gpt/chroma_db && chown -R worker local_data/private_gpt
+RUN mkdir -p local_data/private_gpt/chromadb && chown -R worker local_data/private_gpt
+
+# Copy only the necessary files for the app
+COPY --chown=worker --from=dependencies /home/worker/app/.venv/ .venv
+COPY --chown=worker private_gpt/ private_gpt
+COPY --chown=worker docs/ docs
+COPY --chown=worker *.yaml *.md ./
 
-# Ensure the 'meta.json' file has the correct permissions
-RUN touch local_data/private_gpt/chroma_db/meta.json && chown worker local_data/private_gpt/chroma_db/meta.json
 
-# Allow the user to write to the chroma_db directory
-RUN chmod 777 local_data/private_gpt/chroma_db
-USER worker
 
-# Create and activate virtual environment
-RUN python -m venv .venv
-ENV PATH="/home/worker/app/.venv/bin:$PATH"
+USER worker
 
-# Print the contents of .venv/bin and run uvicorn
-CMD ["sh", "-c", "ls -l .venv/bin && .venv/bin/uvicorn private_gpt.main:app --host 0.0.0.0 --port 7860"]
+# Activate the virtual environment and run uvicorn
+CMD [".venv/bin/uvicorn", "private_gpt.main:app", "--host", "0.0.0.0", "--port", "7860"]