Updated FastAPI code

Dockerfile

@@ -5,7 +5,7 @@ RUN apt-get update && apt-get install -y \
     git \
     ffmpeg \
     libsndfile1 \
- && rm -rf /var/lib/apt/lists/*
+    && rm -rf /var/lib/apt/lists/*
 
 # Set working directory
 WORKDIR /app
@@ -13,33 +13,39 @@ 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 to the 'app' user
-RUN mkdir -p /tmp/huggingface /tmp/matplotlib
-RUN chown -R app:app /tmp/huggingface /tmp/matplotlib
+# Create cache directories with proper permissions
+RUN mkdir -p /home/app/.cache/huggingface \
+    /home/app/.cache/matplotlib \
+    /home/app/.cache/torch \
+    && chown -R app:app /home/app
 
-# Copy source code and set its ownership
-COPY . .
-RUN chown -R app:app /app
+# Copy requirements first for better caching
+COPY requirements.txt .
+RUN chown app:app requirements.txt
 
-# Switch to the non-root user
+# Switch to the non-root user early
 USER app
 
-# 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
+# Set environment variables to use user's home directory for caches
+ENV HF_HOME=/home/app/.cache/huggingface
+ENV HF_HUB_CACHE=/home/app/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/home/app/.cache/huggingface
+ENV MPLCONFIGDIR=/home/app/.cache/matplotlib
+ENV TORCH_HOME=/home/app/.cache/torch
 ENV PATH="/home/app/.local/bin:${PATH}"
-ENV PYTHONPATH="/home/app/.local/lib/python3.10/site-packages"
-ENV HF_HOME=/tmp/huggingface
-ENV HF_HUB_CACHE=/tmp/huggingface
-ENV TRANSFORMERS_CACHE=/tmp/huggingface
-ENV MPLCONFIGDIR=/tmp/matplotlib
-
-# 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
+
+# Install Python packages as the 'app' user
+RUN pip install --user --upgrade pip
+RUN pip install --user --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+RUN pip install --user --no-cache-dir git+https://github.com/pyannote/pyannote-audio.git
+RUN pip install --user --no-cache-dir -r requirements.txt
+
+# Copy source code and set ownership
+COPY --chown=app:app . .
+
+# Pre-download models to avoid runtime permission issues (optional)
+# RUN python -c "from transformers import pipeline; pipeline('summarization', model='facebook/bart-large-cnn')"
+# RUN python -c "from faster_whisper import WhisperModel; WhisperModel('tiny')"
 
 # Expose the application port
 EXPOSE 7860