From 5002da7bf416aaa0230def8fe01c4ef7880e460f Mon Sep 17 00:00:00 2001
From: Jeffreytsai1004 <jeffreytsai1004@gmail.com>
Date: Sat, 15 Feb 2025 16:36:21 +0800
Subject: [PATCH] Update download_resource.bat

---
 download_resource.bat | 30 +++++++++++++++++++++---------
 1 file changed, 21 insertions(+), 9 deletions(-)

diff --git a/download_resource.bat b/download_resource.bat
index 5e0c434..07cc4db 100644
--- a/download_resource.bat
+++ b/download_resource.bat
@@ -24,7 +24,8 @@ IF %ERRORLEVEL% NEQ 0 (
 
 REM 安装gdown
 echo Installing gdown...
-python -m pip install --user gdown requests --quiet --upgrade
+python -m pip install --user gdown --upgrade
+python -m pip install --user requests --upgrade
 IF %ERRORLEVEL% NEQ 0 (
     echo ERROR: Failed to install gdown
     echo Please check your internet connection and try again
@@ -38,12 +39,15 @@ set PYTHON_USER_SCRIPTS=%USERPROFILE%\AppData\Roaming\Python\Python3*\Scripts
 set PATH=%PYTHON_SCRIPTS_PATH%;%PYTHON_USER_SCRIPTS%;%PATH%
 
 REM 验证gdown是否可用
-python -m gdown --version >nul 2>nul
+echo Testing gdown installation...
+python -c "import gdown" >nul 2>nul
 IF %ERRORLEVEL% NEQ 0 (
-    echo ERROR: gdown installation found but not accessible
-    echo Please try reinstalling gdown: python -m pip install --user --upgrade gdown
+    echo ERROR: gdown installation failed
+    echo Please try running the script again
     exit /b 1
 )
+echo gdown installed successfully!
+echo.
 
 REM 创建resource目录结构
 mkdir resource 2>nul
@@ -62,13 +66,17 @@ echo.
 REM 下载 Neural Haircut 文件
 cd NeuralHaircut
 echo Downloading Neural Haircut files...
-python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8" --retry-times 5
+python -m gdown --folder "https://drive.google.com/drive/folders/1TCdJ0CKR3Q6LviovndOkJaKm8S1T9F_8"
 cd diffusion_prior
-python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f" --retry-times 5
+python -m gdown "1_9EOUXHayKiGH5nkrayncln3d6m1uV7f"
 cd ..\PIXIE
-python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf" --retry-times 5
+python -m gdown "1mPcGu62YPc4MdkT8FFiOCP629xsENHZf"
 
 REM 使用7-Zip解压tar.gz文件
+IF NOT EXIST pixie_data.tar.gz (
+    echo ERROR: Failed to download pixie_data.tar.gz
+    exit /b 1
+)
 7z x pixie_data.tar.gz -y
 7z x pixie_data.tar -y
 del pixie_data.tar.gz
@@ -84,7 +92,11 @@ python -m gdown "1d97oKuITCeWgai2Tf3iNilt6rMSSYzkW" -O model.pth --retry-times 5
 REM 下载 OpenPose 模型
 cd ..\openpose
 echo Downloading OpenPose models...
-python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz --retry-times 5
+python -m gdown "1Yn03cKKfVOq4qXmgBMQD20UMRRRkd_tV" -O models.tar.gz
+IF NOT EXIST models.tar.gz (
+    echo ERROR: Failed to download models.tar.gz
+    exit /b 1
+)
 7z x models.tar.gz -y
 7z x models.tar -y -omodels
 del models.tar.gz
@@ -93,7 +105,7 @@ del models.tar
 REM 下载 hyperIQA 模型
 cd ..\hyperIQA\pretrained
 echo Downloading hyperIQA models...
-python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE" --retry-times 5
+python -m gdown "1OOUmnbvpGea0LIGpIWEbOyxfWx6UCiiE"
 
 cd ..\..
 echo.