FROM rocm/pytorch:latest

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

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

# Создание директории для базы данных
RUN mkdir -p /app/db

# Копирование зависимостей
COPY requirements.txt .

# Установка Python зависимостей
RUN pip install --no-cache-dir -r requirements.txt

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

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

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