# Базовий образ з Python
FROM python:3.10-slim

# Создаем пользователя и директорию
RUN useradd -m -u 1000 user

# Создаем директорию для кеша и устанавливаем права
RUN mkdir -p /cache && chown -R user:user /cache
RUN mkdir -p /results && chown -R user:user /results
RUN mkdir -p /fine_tuned_model && chown -R user:user /fine_tuned_model

# Устанавливаем переменную окружения для кеша
ENV TRANSFORMERS_CACHE=/cache

# Переключаемся на созданного пользователя
USER user
ENV PATH="/home/user/.local/bin:$PATH"

# Оновлення pip та встановлення потрібних бібліотек
RUN pip install --upgrade pip

# Встановлення необхідних пакетів
RUN pip install torch transformers datasets tqdm
RUN pip install transformers[torch]

# Скопіювати ваш Python-код у контейнер
WORKDIR /app
COPY --chown=user . /app

# Запуск скрипта донавчання
CMD ["python3", "finetunning.py"]