@echo off
REM Byte Dream - Quick Fix and Setup Script for Windows
REM Run this script to fix the model loading issue

echo ============================================================
echo Byte Dream - Quick Fix and Setup
echo ============================================================
echo.

REM Check if Python is installed
python --version >nul 2>&1
if %errorlevel% neq 0 (
    echo ERROR: Python not found! Please install Python 3.8+
    pause
    exit /b 1
)

echo Step 1: Checking for trained model...
echo.

if exist "models\bytedream" (
    echo Found model at: models\bytedream
) else if exist "models" (
    echo Found models directory
) else (
    echo WARNING: No trained model found!
    echo.
    echo To train the model, run:
    echo   python train.py --epochs 1000 --batch_size 4
    echo.
)

echo.
echo Step 2: Testing pipeline with random initialization...
echo.

python quick_fix.py

echo.
echo ============================================================
echo Next Steps:
echo ============================================================
echo.
echo 1. If you want to train the model:
echo    python train.py --epochs 1000 --batch_size 4
echo.
echo 2. If you want to upload to Hugging Face:
echo    a. Install huggingface_hub: pip install huggingface_hub
echo    b. Login: huggingface-cli login
echo    c. Upload: python upload_to_hf.py --repo_id "YourUsername/ByteDream" --create_space
echo.
echo 3. To use the web interface now:
echo    python app.py
echo.
echo For detailed instructions, see: UPLOAD_GUIDE_PT.md
echo ============================================================
echo.
pause