#!/bin/bash

# 启动 Ollama 服务
echo "Starting Ollama service..."
ollama serve > /app/ollama.log 2>&1 &

# 等待 Ollama 服务完全启动
echo "Waiting for Ollama service to start..."
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"
        cat /app/ollama.log
        exit 1
    fi
    echo "Waiting... ($i/30)"
    sleep 2
done

# 检查模型文件
echo "Checking model file..."
if [ ! -f /app/model.gguf ]; then
    echo "Error: Model file not found!"
    exit 1
fi

# 检查并创建模型
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 "Failed to create model"
        cat /app/ollama.log
        exit 1
    fi
    echo "Model created successfully!"
fi

# 启动 Python 应用
echo "Starting Gradio application..."
exec python app.py