Updated

Dockerfile

@@ -1,8 +1,8 @@
 # ==============================================================
-# Tech Disciples AI Backend — Optimized Dockerfile
+# Tech Disciples AI Backend — Dockerfile (Optimized)
 # ==============================================================
 
-# Use official slim Python 3.10 image
+# Use official Python 3.10 slim image
 FROM python:3.10-slim
 
 # Set environment variables
@@ -13,30 +13,46 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
 # Set working directory
 WORKDIR $APP_HOME
 
-# Install system dependencies
+# -------------------------
+# System dependencies
+# -------------------------
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     curl \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# Upgrade pip
+# -------------------------
+# Upgrade pip & essential build tools
+# -------------------------
 RUN pip install --upgrade pip setuptools wheel
 
+# -------------------------
+# Install PyTorch (CPU-only)
+# -------------------------
+RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
+# -------------------------
 # Copy requirements file first (for caching)
+# -------------------------
 COPY requirements.txt .
 
-# Install heavy packages separately first (torch) for reliability
-RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-
-# Install remaining Python dependencies
+# -------------------------
+# Install Python dependencies
+# -------------------------
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
+# -------------------------
+# Copy app code
+# -------------------------
 COPY . .
 
-# Expose port for FastAPI
+# -------------------------
+# Expose FastAPI port
+# -------------------------
 EXPOSE 7860
 
-# Start FastAPI with Uvicorn
+# -------------------------
+# Run the FastAPI app with Uvicorn
+# -------------------------
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]