@echo off
chcp 65001 >nul
echo ========================================
echo   RAG知识库 - 处理上传文件
echo ========================================
echo.

REM 检查应用是否运行
echo [1/3] 检查应用状态...
curl -s http://localhost:30081/api/knowledge/status >nul 2>&1
if errorlevel 1 (
    echo [错误] 应用未启动！
    echo.
    echo 请先启动Spring Boot应用，然后再运行此脚本。
    echo.
    pause
    exit /b 1
)
echo [✓] 应用正在运行

echo.
echo [2/3] 检查上传目录...
if not exist "D:\wwwroot\RAG\uploads" (
    echo [错误] 上传目录不存在！
    pause
    exit /b 1
)

dir /b "D:\wwwroot\RAG\uploads\*.txt" "D:\wwwroot\RAG\uploads\*.pdf" "D:\wwwroot\RAG\uploads\*.docx" "D:\wwwroot\RAG\uploads\*.md" 2>nul | find /c /v "" > temp_count.txt
set /p FILE_COUNT=<temp_count.txt
del temp_count.txt

echo [✓] 找到 %FILE_COUNT% 个文档文件

if "%FILE_COUNT%"=="0" (
    echo.
    echo [提示] 上传目录为空，请先将文档文件放入：
    echo D:\wwwroot\RAG\uploads
    echo.
    echo 支持的格式：.txt .pdf .docx .md
    echo.
    pause
    exit /b 0
)

echo.
echo [3/3] 开始处理文件并生成向量...
echo 这可能需要几分钟，请耐心等待...
echo.

curl -X POST http://localhost:30081/api/knowledge/rebuild -H "Content-Type: application/json"

echo.
echo.
echo ========================================
echo   处理完成！
echo ========================================
echo.
echo 现在可以：
echo 1. 运行 "检查知识库状态.bat" 查看加载情况
echo 2. 在系统中使用AI分析功能
echo.
pause