From ec46022fd8bd18cf02574c1fd8b609a85447799a Mon Sep 17 00:00:00 2001
From: TylerCG <117808427+TylerCG@users.noreply.github.com>
Date: Mon, 12 May 2025 10:40:14 -0400
Subject: [PATCH] Update download.py

---
 app/download.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/app/download.py b/app/download.py
index 806a880..009ccf4 100644
--- a/app/download.py
+++ b/app/download.py
@@ -6,8 +6,18 @@ logger = logging.getLogger(__name__)
 
 # Define the hook function
 def my_hook(d):
-    if d['status'] == 'finished':
-        logging.info(f"Download completed: {d['filename']}")
+    status = d.get('status')
+    
+    if status == 'downloading':
+        logger.info(f"Downloading: {d.get('_percent_str')} at {d.get('_speed_str')} for {d.get('filename')}")
+    elif status == 'finished':
+        logger.info(f"Download completed: {d.get('filename')}")
+    elif status == 'error':
+        logger.error(f"Error occurred: {d.get('error')}")
+    elif status == 'postprocessing':
+        logger.info(f"Post-processing: {d.get('filename')}")
+    elif status == 'processing':
+        logger.info(f"Processing: {d.get('filename')}")
 
 def ebook(url, author):
     destination = f"/ebooks/{author}"