Update Dockerfile

Dockerfile

@@ -13,28 +13,22 @@ RUN apt-get update && apt-get install -y \
 # Ollama 설치
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# 환경변수 설정
+# ✅ 핵심 변경: /tmp 디렉토리 사용
 ENV OLLAMA_MODEL=gemma2:9b
-ENV OLLAMA_HOME=/app/.ollama
-ENV OLLAMA_MODELS=/app/.ollama
-
-# Ollama 디렉토리 구조를 완전히 생성하고 권한 설정
-RUN mkdir -p /app/.ollama/models \
-    && mkdir -p /app/.ollama/manifests \
-    && mkdir -p /app/.ollama/manifests/registry.ollama.ai/library \
-    && mkdir -p /app/.ollama/blobs \
-    && chmod -R 777 /app/.ollama
+ENV OLLAMA_HOME=/tmp/ollama-home
+ENV OLLAMA_MODELS=/tmp/ollama-home
 
-# 빌드 중 잠깐 서버를 띄워서 모델만 미리 받아두고 종료
+# 빌드 중 모델 다운로드 (/tmp 사용)
 RUN set -eux; \
-    ollama serve & pid=$!; \
+    mkdir -p /tmp/ollama-home; \
+    OLLAMA_HOME=/tmp/ollama-home OLLAMA_MODELS=/tmp/ollama-home ollama serve & pid=$!; \
     i=0; \
     until curl -fsS http://127.0.0.1:11434/api/tags >/dev/null 2>&1; do \
       i=$((i+1)); \
       [ "$i" -lt 60 ] || { echo "Ollama not ready"; kill "$pid"; exit 1; }; \
       sleep 1; \
     done; \
-    ollama pull gemma2:9b; \
+    OLLAMA_HOME=/tmp/ollama-home OLLAMA_MODELS=/tmp/ollama-home ollama pull gemma2:9b; \
     kill "$pid" || true
 
 # 3. 작업 디렉터리 설정
@@ -55,26 +49,25 @@ ENV HOME=/app
 ENV STREAMLIT_HOME=/app/.streamlit
 RUN mkdir -p /app/.streamlit && chmod -R 777 /app/.streamlit
 
+# 캐시 경로도 /tmp로 통일
 ENV HF_HOME=/tmp/hf-home \
     TRANSFORMERS_CACHE=/tmp/hf-cache \
     HUGGINGFACE_HUB_CACHE=/tmp/hf-cache \
     TORCH_HOME=/tmp/torch-cache \
     XDG_CACHE_HOME=/tmp/xdg-cache
 
-RUN mkdir -p /tmp/hf-home /tmp/hf-cache /tmp/torch-cache /tmp/xdg-cache
-
 ENV OLLAMA_HOST=http://127.0.0.1:11434
 EXPOSE 8501
 EXPOSE 11434
 
-# root 권한으로 실행 (권한 문제 완전 해결)
-USER root
-
-# 최종 실행 커맨드
+# ✅ 간단하고 안전한 CMD
 CMD bash -lc '\
   export OLLAMA_MODEL=${OLLAMA_MODEL:-gemma2:9b}; \
+  export OLLAMA_HOME=/tmp/ollama-home; \
+  export OLLAMA_MODELS=/tmp/ollama-home; \
   echo "Starting Ollama with model: $OLLAMA_MODEL"; \
-  chmod -R 777 /app/.ollama; \
+  echo "Using Ollama home: $OLLAMA_HOME"; \
+  mkdir -p /tmp/ollama-home; \
   ollama serve & \
   for i in {1..120}; do \
     if curl -fsS http://127.0.0.1:11434/api/version >/dev/null 2>&1; then \
@@ -84,7 +77,12 @@ CMD bash -lc '\
     echo "Waiting for Ollama... ($i/120)"; \
     sleep 1; \
   done; \
-  echo "Ensuring model is available: $OLLAMA_MODEL"; \
-  ollama pull "$OLLAMA_MODEL" || { echo "Failed to pull model"; exit 1; }; \
+  echo "Checking if model exists..."; \
+  if ! ollama list | grep -q "$OLLAMA_MODEL"; then \
+    echo "Model not found, pulling: $OLLAMA_MODEL"; \
+    ollama pull "$OLLAMA_MODEL" || { echo "Failed to pull model"; exit 1; }; \
+  else \
+    echo "Model already available: $OLLAMA_MODEL"; \
+  fi; \
   echo "Starting Streamlit app"; \
   streamlit run app.py --server.address=0.0.0.0 --server.port=${PORT:-8501}'