FROM python:3.11

WORKDIR /app

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git git-lfs ffmpeg libsm6 libxext6 cmake libgl1 \
    && git lfs install

# Copy requirements and install
COPY requirements.txt .

RUN pip install --no-cache-dir torch==2.2.2+cu121 torchaudio==2.2.2+cu121 \
    --index-url https://download.pytorch.org/whl/cu121

RUN pip install --no-cache-dir -r requirements.txt

# # Install huggingface hub CLI
# RUN pip install huggingface_hub

# # Download model files
# RUN python -c "from huggingface_hub import hf_hub_download; \
# import os; \
# MODEL_DIR = 'my_model'; \
# os.makedirs(MODEL_DIR, exist_ok=True); \
# hf_hub_download('MariaKaiser/EGTTS_finetuned', 'config.json', cache_dir=MODEL_DIR); \
# hf_hub_download('MariaKaiser/EGTTS_finetuned', 'vocab.json', cache_dir=MODEL_DIR); \
# hf_hub_download('MariaKaiser/EGTTS_finetuned', 'model.pth', cache_dir=MODEL_DIR)"



# Copy the app code
COPY . .

# Run the app
CMD ["python", "app.py"]