@echo off
chcp 65001 >nul
title ComfyUI CUDA Fix

echo.
echo ================================================================
echo   ComfyUI CUDA Onarim Scripti
echo ================================================================
echo.

set "COMFY_DIR=D:\__sistem__\ProgramFiles\ComfyUi\ComfyUI"
cd /d "%COMFY_DIR%"
call venv\Scripts\activate.bat

echo [1/4] NVIDIA Driver kontrolu...
nvidia-smi
if %errorlevel% neq 0 (
    echo.
    echo [HATA] nvidia-smi calismadi! NVIDIA driver kurulu degil veya bozuk.
    echo Lutfen https://www.nvidia.com/drivers adresinden driver indirin.
    pause
    exit /b 1
)

echo.
echo [2/4] Mevcut PyTorch kaldiriliyor...
pip uninstall torch torchvision torchaudio xformers -y

echo.
echo [3/4] PyTorch + CUDA 12.6 yukleniyor (bu biraz surecek)...
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

echo.
echo [4/4] CUDA testi yapiliyor...
python -c "import torch; print('CUDA Kullanilabilir:', torch.cuda.is_available()); print('GPU:', torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'YOK')"

echo.
echo ================================================================
if %errorlevel% equ 0 (
    echo   [TAMAM] Onarim tamamlandi!
    echo   Simdi 3_COMFYUI_BASLAT.bat ile ComfyUI'yi baslatabilirsiniz.
) else (
    echo   [HATA] Bir sorun olustu. Ciktiyi kontrol edin.
)
echo ================================================================
echo.
pause