# ROCm-совместимый образ с PyTorch
FROM rocm/pytorch:rocm5.7.3_ubuntu22.04_py3.10_pytorch2.1.2

# Установка системных зависимостей
RUN apt-get update && apt-get install -y \
    ffmpeg \
    curl \
    git \
    && rm -rf /var/lib/apt/lists/*

# Создание рабочей директории
WORKDIR /app

# Установка переменных окружения для Python
ENV PYTHONUNBUFFERED=1
ENV HSA_OVERRIDE_GFX_VERSION=10.3.0

# Установка Python зависимостей (без torch/torchaudio - уже в базовом образе)
RUN pip install --no-cache-dir \
    fastapi==0.109.0 \
    uvicorn[standard]==0.27.0 \
    python-multipart==0.0.6 \
    openai-whisper==20231117

# Копирование исходного кода
COPY main.py .

# Экспорт порта
EXPOSE 8080

# Запуск приложения
CMD ["python", "main.py"]
