Update Dockerfile

Dockerfile

@@ -1,34 +1,33 @@
-# ---- BASE PYTHON IMAGE ----
 FROM python:3.10-slim
 
-# ---- ENV & WORKDIR ----
-ENV PYTHONDONTWRITEBYTECODE=1
-ENV PYTHONUNBUFFERED=1
 WORKDIR /code
 
-# ---- SYSTEM DEPENDENCIES ----
+# System dependencies
 RUN apt-get update && apt-get install -y \
     libsndfile1 ffmpeg git \
     && rm -rf /var/lib/apt/lists/*
 
-# ---- COPY PROJECT FILES ----
+# Copy files
 COPY . /code
 
-# ---- INSTALL DEPENDENCIES ----
+# Set environment for Streamlit and Transformers
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONUNBUFFERED=1
+ENV HF_HOME=/tmp/hf-home
+ENV TRANSFORMERS_CACHE=/tmp/hf-cache
+ENV NLTK_DATA=/tmp/nltk_data
+ENV XDG_CONFIG_HOME=/tmp  # 👈 Streamlit will store .streamlit here
+
+# Install Python dependencies
 RUN pip install --upgrade pip
 RUN pip install -r requirements.txt
 
-# Avoid permission issues with Streamlit
-ENV HOME=/code
-ENV STREAMLIT_HOME=/code
-
-ENV TRANSFORMERS_CACHE=/tmp/hf-cache
-ENV HF_HOME=/tmp/hf-home
-ENV NLTK_DATA=/tmp/nltk_data
-ENV STREAMLIT_ROOT=/tmp/.streamlit
+# Make .streamlit dir in /tmp
+RUN mkdir -p /tmp/.streamlit
 
-# ---- EXPOSE PORTS ----
+# Expose ports
 EXPOSE 7860
 EXPOSE 8000
 
+# Run both frontend and backend
 CMD ["bash", "-c", "streamlit run frontend.py --server.port 7860 & uvicorn main:app --host 0.0.0.0 --port 8000"]