@echo off
chcp 65001 >nul
setlocal enabledelayedexpansion

echo ========================================
echo        AI Crawler Launcher
echo ========================================
echo.
echo Input instance count for each crawler
echo Press Enter for default (1), input 0 to skip
echo.

set /p "ds_count=DeepSeek [default 2]: "
if "!ds_count!"=="" set "ds_count=2"

set /p "kimi_count=Kimi [default 3]: "
if "!kimi_count!"=="" set "kimi_count=3"

set /p "mita_count=Mita【密塔】 [default 2]: "
if "!mita_count!"=="" set "mita_count=2"

set /p "tongyi_count=Tongyi【通义】 [default 1]: "
if "!tongyi_count!"=="" set "tongyi_count=1"

set /p "yuanbao_count=Yuanbao【元宝】 [default 1]: "
if "!yuanbao_count!"=="" set "yuanbao_count=1"

set /p "wxyy_count=Wenxin【文心一言】 [default 1]: "
if "!wxyy_count!"=="" set "wxyy_count=1"

echo.
echo ========================================
echo Config:
echo ========================================
echo DeepSeek: !ds_count!
echo Kimi: !kimi_count!
echo Mita: !mita_count!
echo Tongyi: !tongyi_count!
echo Yuanbao: !yuanbao_count!
echo Wenxin: !wxyy_count!
echo ========================================
echo.
pause

if !ds_count! gtr 0 (
    echo [1/6] Starting DeepSeek...
    for /l %%i in (1,1,!ds_count!) do (
        start cmd /k "title DeepSeek-%%i && python deepseekstart.py"
        timeout /t 2 /nobreak >nul
    )
)

if !kimi_count! gtr 0 (
    echo [2/6] Starting Kimi...
    for /l %%i in (1,1,!kimi_count!) do (
        start cmd /k "title Kimi-%%i && python kimistart.py"
        timeout /t 2 /nobreak >nul
    )
)

if !mita_count! gtr 0 (
    echo [3/6] Starting Mita...
    for /l %%i in (1,1,!mita_count!) do (
        start cmd /k "title Mita-%%i && python mitastart.py"
        timeout /t 2 /nobreak >nul
    )
)

if !tongyi_count! gtr 0 (
    echo [4/6] Starting Tongyi...
    for /l %%i in (1,1,!tongyi_count!) do (
        start cmd /k "title Tongyi-%%i && python tongyistart.py"
        timeout /t 2 /nobreak >nul
    )
)

if !yuanbao_count! gtr 0 (
    echo [5/6] Starting Yuanbao...
    for /l %%i in (1,1,!yuanbao_count!) do (
        start cmd /k "title Yuanbao-%%i && python yuanbao.py"
        timeout /t 2 /nobreak >nul
    )
)

if !wxyy_count! gtr 0 (
    echo [6/6] Starting Wenxin...
    for /l %%i in (1,1,!wxyy_count!) do (
        start cmd /k "title Wenxin-%%i && python wxyystart.py"
        timeout /t 2 /nobreak >nul
    )
)

echo.
echo ========================================
set /a "total=!ds_count!+!kimi_count!+!mita_count!+!tongyi_count!+!yuanbao_count!+!wxyy_count!"
echo All started! Total: !total! windows
echo ========================================
pause