Update Dockerfile

Dockerfile

@@ -1,43 +1,46 @@
-# Use NVIDIA CUDA 11.8 base (Matches our Torch 2.1.2 requirement)
+# Use NVIDIA CUDA 11.8 base
 FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
-# Setup basic environment to avoid interactive prompts
+# Setup environment
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 \
     PIP_CACHE_DIR=/var/cache/pip
 
-# 1. Install FFmpeg and Python 3.10
+# 1. Install System Dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.10 \
     python3-pip \
     python3-dev \
     ffmpeg \
     git \
+    wget \
     pkg-config \
-    libavformat-dev libavcodec-dev libavdevice-dev libavutil-dev libswscale-dev \
+    # These libraries allow 'av' and other audio tools to compile
+    libavformat-dev libavcodec-dev libavdevice-dev libavutil-dev libswscale-dev libswresample-dev libavfilter-dev \
     && rm -rf /var/lib/apt/lists/*
 
 # Set python3.10 as default
 RUN ln -s /usr/bin/python3.10 /usr/bin/python
 
-# 2. Setup User (Hugging Face requirement)
+# 2. Setup User
 WORKDIR /app
 RUN useradd -m -u 1000 user
 RUN chown -R user:user /app
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
-# 3. Install Dependencies
+# 3. Install Python Dependencies
 COPY --chown=user requirements.txt requirements.txt
+
+# WE SPLIT THE INSTALL to handle the index-url correctly
 RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir \
-    "torch==2.1.2" \
-    "torchaudio==2.1.2" \
-    --index-url https://download.pytorch.org/whl/cu118 && \
+    # Step A: Install Torch from PyTorch Index
+    pip install --no-cache-dir "torch==2.1.2" "torchaudio==2.1.2" --index-url https://download.pytorch.org/whl/cu118 && \
+    # Step B: Install the rest from standard PyPI
     pip install --no-cache-dir -r requirements.txt
 
 # 4. Copy Code
 COPY --chown=user . .
 
-# 5. Launch FastAPI with Uvicorn
+# 5. Launch
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]