Update Dockerfile

Dockerfile

@@ -1,6 +1,78 @@
-# Dockerfile para IA Akira
-FROM python:3.11-slim
-WORKDIR /app
-COPY . .
-RUN pip install --upgrade pip && pip install -r requirements.txt
+# =====================================================
+# Dockerfile - Akira IA (YAYA-23-8B + GPU + Cache)
+# =====================================================
+FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
+
+# Evita interações
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+
+# =====================================================
+# 1. SISTEMA + DEPENDÊNCIAS
+# =====================================================
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    python3.11 \
+    python3.11-dev \
+    python3-pip \
+    python3-venv \
+    git \
+    git-lfs \
+    curl \
+    wget \
+    build-essential \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    ffmpeg \
+    && rm -rf /var/lib/apt/lists/*
+
+# Link python
+RUN ln -sf /usr/bin/python3.11 /usr/bin/python
+
+# =====================================================
+# 2. DIRETÓRIO + CÓDIGO
+# =====================================================
+WORKDIR /app
+
+# Copia apenas o necessário primeiro (cache)
+COPY requirements.txt .
+COPY modules/ modules/
+COPY main.py .
+
+# =====================================================
+# 3. PYTHON + DEPENDÊNCIAS
+# =====================================================
+RUN pip install --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt && \
+    pip cache purge
+
+# =====================================================
+# 4. BAIXA YAYA-23-8B (OU FALLBACK)
+# =====================================================
+RUN mkdir -p models && \
+    cd models && \
+    git lfs install && \
+    echo "Baixando YAYA-23-8B..." && \
+    (git clone https://huggingface.co/angola-ai/yaya-23-8b || \
+     echo "YAYA não encontrado. Baixando Llama 3.1 8B como fallback..." && \
+     git clone https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct yaya-23-8b) && \
+    echo "Modelo pronto em ./models/yaya-23-8b"
+
+# =====================================================
+# 5. PERMISSÕES + LIMPEZA
+# =====================================================
+RUN chmod -R 755 /app
+
+# =====================================================
+# 6. PORTA + COMANDO
+# =====================================================
+EXPOSE 7860
+
+# Variável de ambiente para GPU
+ENV NVIDIA_VISIBLE_DEVICES=all
+ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
+
+# Inicia a API
 CMD ["python", "main.py"]