Updated FastAPI code

Dockerfile

@@ -13,19 +13,19 @@ WORKDIR /app
 # Create a non-root user with a dedicated home directory
 RUN addgroup --system app && adduser --system --group --home /home/app app
 
-# Create cache directories and set ownership
+# Create cache directories and set ownership to the 'app' user
 RUN mkdir -p /tmp/huggingface /tmp/matplotlib
 RUN chown -R app:app /tmp/huggingface /tmp/matplotlib
 
-# Copy source code and set ownership
+# Copy source code and set its ownership
 COPY . .
 RUN chown -R app:app /app
 
 # Switch to the non-root user
 USER app
 
-# Set environment variables
-# 1. Add user's local bin to the PATH for executables (like uvicorn)
+# Set environment variables so the system can find executables and modules
+# 1. Add user's local bin to PATH for executables (like uvicorn)
 # 2. Add user's site-packages to PYTHONPATH for module imports
 ENV PATH="/home/app/.local/bin:${PATH}"
 ENV PYTHONPATH="/home/app/.local/lib/python3.10/site-packages"
@@ -34,14 +34,14 @@ ENV HF_HUB_CACHE=/tmp/huggingface
 ENV TRANSFORMERS_CACHE=/tmp/huggingface
 ENV MPLCONFIGDIR=/tmp/matplotlib
 
-# Now, run pip install as the 'app' user.
+# Run pip install as the 'app' user.
 # Pip will correctly install packages to /home/app/.local
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir torch torchvision torchaudio
 RUN pip install --no-cache-dir git+https://github.com/pyannote/pyannote-audio.git
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Expose FastAPI port
+# Expose the application port
 EXPOSE 7860
 
 # Run the FastAPI app

app.py

@@ -16,14 +16,6 @@ from google.api_core import exceptions as api_core_exceptions
 
 logger = logging.getLogger(__name__)
 
-os.environ["HF_HOME"] = "/tmp/huggingface"
-os.environ["HF_HUB_CACHE"] = "/tmp/huggingface"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
-os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
-
-os.makedirs("/tmp/huggingface", exist_ok=True)
-os.makedirs("/tmp/matplotlib", exist_ok=True)
-
 app = FastAPI(
     title="Transcription and Summarization API",
     description="API using Faster-Whisper, spaCy, and Hugging Face Transformers",