#!/bin/bash

# Quick fix script for Ollama service
echo "🚀 Быстрое исправление проблемы Ollama..."

# Colors
GREEN='\033[0;32m'
RED='\033[0;31m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

# Check if we're in the right directory
if [ ! -f "ollama.service" ]; then
    echo -e "${RED}❌ Файл ollama.service не найден в текущей директории${NC}"
    echo "Убедитесь, что вы находитесь в /home/user/ai-server"
    exit 1
fi

# Stop the service first
echo "1. Остановка сервиса Ollama..."
sudo systemctl stop ollama

# Copy the updated service file
echo "2. Копирование обновленного файла сервиса..."
sudo cp /home/user/ai-server/ollama.service /etc/systemd/system/ollama.service

# Reload systemd configuration
echo "3. Перезагрузка конфигурации systemd..."
sudo systemctl daemon-reload

# Create Ollama directory
echo "4. Создание директории .ollama..."
mkdir -p /home/user/ai-server/.ollama
chmod 755 /home/user/ai-server/.ollama

# Start the service
echo "5. Запуск сервиса Ollama..."
sudo systemctl start ollama

# Wait a moment
sleep 3

# Check status
echo "6. Проверка статуса..."
sleep 2
if sudo systemctl is-active --quiet ollama; then
    echo -e "${GREEN}✅ Ollama сервис успешно запущен!${NC}"
    echo -e "${GREEN}✅ Проблема с /nonexistent исправлена${NC}"
    
    # Test API
    echo "7. Тестирование API..."
    if curl -s -m 10 http://localhost:11434 >/dev/null 2>&1; then
        echo -e "${GREEN}✅ Ollama API работает${NC}"
    else
        echo -e "${YELLOW}⚠️  API пока недоступен, подождите...${NC}"
    fi
    
    # Show available models
    echo "8. Доступные модели:"
    /usr/local/bin/ollama list 2>/dev/null || echo "Модели загружаются..."
    
else
    echo -e "${RED}❌ Проблемы все еще есть${NC}"
    echo "Последние логи:"
    sudo journalctl -u ollama --no-pager -l | tail -10
    echo ""
    echo -e "${YELLOW}💡 Попробуйте запустить диагностику: ./diagnose.sh${NC}"
fi

echo ""
echo -e "${GREEN}🎉 Исправление завершено!${NC}"
echo -e "${BLUE}📋 Что дальше:${NC}"
echo "  - Откройте веб-интерфейс: http://your-server-ip:8080"
echo "  - Или используйте: curl http://localhost:11434"
echo "  - Для диагностики: ./diagnose.sh"