update Dockerfile

Dockerfile

@@ -1,19 +1,26 @@
 FROM python:3.12
 
-COPY . .
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-WORKDIR /
+# Sử dụng thư mục trong /app thay vì /root
+ENV OLLAMA_MODELS=/app/.ollama/models
 
-RUN pip install --no-cache-dir --upgrade -r /requirements.txt
+# Tạo thư mục cho Ollama
+RUN mkdir -p /app/.ollama && chmod 777 /app/.ollama
 
-RUN curl -fsSL https://ollama.com/install.sh | sh 
+# Cài đặt Ollama
+RUN curl -fsSL https://ollama.com/install.sh | sh
 
-COPY start.sh /start.sh
-RUN chmod +x /start.sh
+# Copy application files
+COPY . .
 
-COPY pull_models.sh /pull_models.sh
-RUN chmod +x /pull_models.sh
+# Make scripts executable
+RUN chmod +x /app/start.sh
+RUN chmod +x /app/pull_models.sh
 
 EXPOSE 7860
 
-CMD ["/start.sh"]
+CMD ["/app/start.sh"]

start.sh

@@ -1,14 +1,14 @@
 #!/bin/bash
 
-# Thiết lập home directory
-export HOME=/root
+# Sử dụng thư mục trong /app thay vì /root
+export OLLAMA_MODELS=/app/.ollama/models
 
-# Kiểm tra xem Ollama đã chạy chưa
 echo "Checking if Ollama is already running..."
 if pgrep -x "ollama" > /dev/null; then
     echo "Ollama is already running."
 else
     echo "Starting Ollama server..."
+    # Chạy Ollama với biến môi trường đúng
     ollama serve &
     
     # Đợi Ollama khởi động
@@ -18,21 +18,9 @@ fi
 
 # Kiểm tra lại xem Ollama đã chạy chưa
 if ! pgrep -x "ollama" > /dev/null; then
-    echo "ERROR: Failed to start Ollama!"
-    exit 1
+    echo "ERROR: Failed to start Ollama! Continuing anyway..."
 fi
 
-# Tải mô hình nếu cần
-echo "Running pull_models.sh"
-/app/pull_models.sh
-
-# Kiểm tra kết nối đến Ollama server
-echo "Testing connection to Ollama server..."
-curl -s http://localhost:11434/api/version || {
-    echo "ERROR: Cannot connect to Ollama server!"
-    exit 1
-}
-
 # Bắt đầu ứng dụng FastAPI
 echo "Starting FastAPI application..."
 uvicorn app:app --host 0.0.0.0 --port 7860