Update app.py

app.py

@@ -3,7 +3,17 @@ import fitz  # PyMuPDF
 import io
 import zipfile
 
+def format_data_size(size_in_bytes):
+    units = ["B", "KB", "MB", "GB", "TB"]
+    size = size_in_bytes
+    unit_index = 0
 
+    while size >= 1024 and unit_index < len(units) - 1:
+        size /= 1024
+        unit_index += 1
+
+    return f"{size:.2f} {units[unit_index]}"
+    
 def split_pdf_by_pages_and_zip_in_memory(doc, dpi):
     """
     Splits a PDF document into individual pages, saves each page as an image in memory,
@@ -75,7 +85,7 @@ def main():
             del st.session_state["progress_bar"]
             st.subheader("Download ZIP Archive")
             zip_size = zip_data.getbuffer().nbytes
-            data_size = f"{zip_size / (1024 * 1024) :.2f} MB"
+            data_size = format_data_size(zip_size)
             create_download_link(zip_data, data_size, f"pages_1_to_{total_pages}.zip")
 
         except Exception as e: