#!/bin/sh
# Полное исправление поиска в Open WebUI
# Исправляет конфигурацию SearXNG и перезапускает контейнеры

echo "=== ПОЛНОЕ ИСПРАВЛЕНИЕ ПОИСКА ==="
echo ""

# 1. Исправляем конфигурацию SearXNG
echo "1. Исправление конфигурации SearXNG..."
./scripts/fix_searxng_config.sh

# 2. Перезапускаем Open WebUI для применения патча User-Agent
echo ""
echo "2. Перезапуск Open WebUI для применения патча User-Agent..."
docker restart open-webui

echo ""
echo "Ожидание запуска Open WebUI (15 секунд)..."
sleep 15

# 3. Проверка
echo ""
echo "3. Финальная проверка..."
echo ""

# Проверка SearXNG
SEARXNG_STATUS=$(docker ps | grep searxng | awk '{print $7}')
if [ "$SEARXNG_STATUS" = "healthy" ] || [ "$SEARXNG_STATUS" = "Up" ]; then
    echo "✓ SearXNG работает"
else
    echo "✗ SearXNG не работает"
fi

# Проверка JSON
JSON_TEST=$(docker exec open-webui curl -s "http://searxng:8080/search?q=test&format=json" 2>&1 | head -c 200)
if echo "$JSON_TEST" | grep -q "results"; then
    echo "✓ JSON формат работает"
else
    echo "✗ JSON формат не работает"
fi

# Проверка патча
RAG_BOT_FOUND=$(docker exec open-webui grep -r "RAG Bot" /app/backend/open_webui/routers/retrieval.py /app/backend/open_webui/utils/middleware.py 2>/dev/null | wc -l)
if [ "$RAG_BOT_FOUND" -eq 0 ]; then
    echo "✓ Патч User-Agent применен"
else
    echo "⚠ Патч User-Agent может быть не применен (найдено: $RAG_BOT_FOUND)"
fi

echo ""
echo "=== ГОТОВО ==="
echo ""
echo "Проверьте поиск в Open WebUI:"
echo "1. Откройте Settings → Web Search"
echo "2. URL должен быть: http://searxng:8080/search?q=<query>&format=json"
echo "3. Попробуйте поиск в чате"