Update app.py

app.py

@@ -46,14 +46,27 @@ def zip_subdirs(start_dir):
             yield zip_filename
 
 def get_zip_download_link(zip_file):
+    # Count number of files in the zip
+    with zipfile.ZipFile(zip_file, 'r') as zipf:
+        num_files = len(zipf.namelist())
+    # Get the size of the zip file
+    zip_size = os.path.getsize(zip_file)
+    # Convert size to a more readable format (e.g., KB, MB)
+    if zip_size < 1024:
+        size_str = f"{zip_size} B"
+    elif zip_size < 1024**2:
+        size_str = f"{zip_size / 1024:.2f} KB"
+    else:
+        size_str = f"{zip_size / 1024**2:.2f} MB"
+
+    # Create the download link
     with open(zip_file, 'rb') as f:
         bytes = f.read()
     b64 = base64.b64encode(bytes).decode()
     link_name = os.path.basename(zip_file)
-    href = f'<a href="data:file/zip;base64,{b64}" download="{link_name}">Download: {link_name}</a>'
+    href = f'<a href="data:file/zip;base64,{b64}" download="{link_name}">{link_name} - {num_files} files, {size_str}</a>'
     return href
-
-  
+    
 @st.cache_resource
 def create_zip_of_files(files):
     zip_name = "all_files.zip"
@@ -241,6 +254,7 @@ def main():
         if st.sidebar.button("⬇️ Download All"):
             start_directory = '.'  # Current directory
             for zip_file in zip_subdirs(start_directory):
+                # Display enhanced download link with file count and size
                 st.sidebar.markdown(get_zip_download_link(zip_file), unsafe_allow_html=True)
             
         # Expander for showing URL history and download links