Update start.sh

start.sh

@@ -1,29 +1,37 @@
 #!/bin/bash
 
 # 启动 Ollama 服务
-ollama serve &
+echo "Starting Ollama service..."
+ollama serve > /app/ollama.log 2>&1 &
 
 # 等待 Ollama 服务完全启动
 echo "Waiting for Ollama service to start..."
-sleep 10
-
-# 检查 Ollama 服务是否正在运行
-curl --retry 5 --retry-delay 2 --retry-connrefused http://127.0.0.1:11434/api/version
-if [ $? -ne 0 ]; then
-    echo "Error: Ollama service failed to start"
-    exit 1
-fi
+for i in {1..30}; do
+    if curl -s http://127.0.0.1:11434/api/version > /dev/null; then
+        echo "Ollama service is up!"
+        break
+    fi
+    if [ $i -eq 30 ]; then
+        echo "Timeout waiting for Ollama service"
+        exit 1
+    fi
+    echo "Waiting... ($i/30)"
+    sleep 2
+done
 
-# 检查模型是否存在，如果不存在则创建
+# 检查并创建模型
+echo "Checking for llama3-zh model..."
 if ! ollama list | grep -q "llama3-zh"; then
     echo "Creating llama3-zh model..."
     ollama create llama3-zh -f /app/Modelfile
     if [ $? -ne 0 ]; then
-        echo "Error: Failed to create llama3-zh model"
+        echo "Failed to create model"
+        cat /app/ollama.log
         exit 1
     fi
+    echo "Model created successfully!"
 fi
 
-echo "Starting Python application..."
 # 启动 Python 应用
-python app.py
+echo "Starting Gradio application..."
+exec python app.py