From b06c4777a34616e1f69df1af0d80bc84ac2835d5 Mon Sep 17 00:00:00 2001
From: "Tuan Anh Nguyen Dang (Tadashi_Cin)" <tadashi@cinnamon.is>
Date: Sun, 8 Sep 2024 21:22:01 +0700
Subject: [PATCH] fix: add PDFJS download to Windows setup (#249)

---
 libs/ktem/ktem/app.py   |  5 ++++-
 scripts/run_windows.bat | 49 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+), 1 deletion(-)

diff --git a/libs/ktem/ktem/app.py b/libs/ktem/ktem/app.py
index c8f7ea4..4c6e06a 100644
--- a/libs/ktem/ktem/app.py
+++ b/libs/ktem/ktem/app.py
@@ -49,8 +49,11 @@ class BaseApp:
             self._js = self._js.replace("KH_APP_VERSION", self.app_version)
         with (dir_assets / "js" / "pdf_viewer.js").open() as fi:
             self._pdf_view_js = fi.read()
+            # workaround for Windows path
+            pdf_js_dist_dir = str(PDFJS_PREBUILT_DIR).replace("\\", "\\\\")
             self._pdf_view_js = self._pdf_view_js.replace(
-                "PDFJS_PREBUILT_DIR", str(PDFJS_PREBUILT_DIR)
+                "PDFJS_PREBUILT_DIR",
+                pdf_js_dist_dir,
             )
 
         self._favicon = str(dir_assets / "img" / "favicon.svg")
diff --git a/scripts/run_windows.bat b/scripts/run_windows.bat
index 7aa2d02..dbde1b5 100644
--- a/scripts/run_windows.bat
+++ b/scripts/run_windows.bat
@@ -55,6 +55,10 @@ CALL :print_highlight "Setting up a local model"
 CALL :setup_local_model
 IF ERRORLEVEL 1 GOTO :end
 
+CALL :print_highlight "Downloading and extracting PDF.js"
+CALL :download_and_extract_pdf_js
+IF ERRORLEVEL 1 GOTO :end
+
 CALL :print_highlight "Launching Kotaemon in your browser, please wait..."
 CALL :launch_ui
 
@@ -227,11 +231,56 @@ IF %ERRORLEVEL% == 0  (
 )
 GOTO :eof
 
+:download_and_extract_pdf_js
+:: Download and extract a ZIP file from a URL to a destination directory
+
+REM Check if the destination directory exists
+if exist "%dest_dir%" (
+    echo Destination directory %dest_dir% already exists. Skipping download.
+    goto :eof
+)
+
+REM Create the destination directory
+mkdir "%dest_dir%"
+
+REM Define variables
+set "pdf_js_version=4.0.379"
+set "pdf_js_dist_name=pdfjs-%pdf_js_version%-dist"
+set "pdf_js_dist_url=https://github.com/mozilla/pdf.js/releases/download/v%pdf_js_version%/%pdf_js_dist_name%.zip"
+for /f "delims=" %%i in ('cd') do set "current_dir=%%i"
+set "target_pdf_js_dir=%current_dir%\libs\ktem\ktem\assets\prebuilt\%pdf_js_dist_name%"
+
+REM Create the target directory if it does not exist (including parent folders)
+if not exist "%target_pdf_js_dir%" (
+    echo Creating directory %target_pdf_js_dir%
+    mkdir "%target_pdf_js_dir%"
+)
+
+REM Download the ZIP file using PowerShell
+set "zip_file=%temp%\downloaded.zip"
+echo Downloading %url% to %zip_file%
+powershell -Command "Invoke-WebRequest -Uri '%pdf_js_dist_url%' -OutFile '%zip_file%'"
+
+
+REM Extract the ZIP file using PowerShell
+echo Extracting %zip_file% to %dest_dir%
+powershell -Command "Expand-Archive -Path '%zip_file%' -DestinationPath '%target_pdf_js_dir%'"
+
+REM Clean up the downloaded ZIP file
+del "%zip_file%"
+echo Download and extraction completed successfully.
+
+goto :eof
+
 :setup_local_model
 python "%CD%\scripts\serve_local.py"
 GOTO :eof
 
 :launch_ui
+:: Workaround for diskcache path with folder start with .
+SET THEFLOW_TEMP_PATH=flow_tmp
+SET PDFJS_PREBUILT_DIR=%target_pdf_js_dir%
+ECHO Starting Kotaemon UI... (prebuilt PDF.js is at %PDFJS_PREBUILT_DIR%)
 CALL python "%CD%\app.py" || ( ECHO. && ECHO Will exit now... && GOTO :exit_func_with_error )
 GOTO :eof