remove data.zip

app.py

@@ -1,5 +1,6 @@
 """Main entry point to the app."""
 
+import os
 from pathlib import Path
 import streamlit as st
 from src.utils.logger_config import setup_logging
@@ -51,5 +52,11 @@
     else:
         st.warning("No files found. Please download the necessary data.")
         if st.button("Download and Extract Data"):
+            if os.path.exists("data.zip"):
+                try:
+                    os.remove("data.zip")
+                    print("Deleted data.zip after extraction.")
+                except OSError as e:
+                    print(f"Error deleting data.zip: {e}")
             download_and_extract_zip()
             st.success("Download and extraction complete!")

src/utils/helper.py

@@ -882,5 +882,12 @@ def download_and_extract_zip() -> None:
             if os.path.exists(extracted_data_folder):
                 shutil.rmtree(extracted_data_folder)
             return
+
+        if os.path.exists(zip_local_path):
+            try:
+                os.remove(zip_local_path)
+                logging.info(f"Deleted {zip_local_path} after extraction.")
+            except OSError as e:
+                logging.error(f"Error deleting {zip_local_path}: {e}")
     else:
         logging.info(f"{extracted_data_folder = } exists. Not downloading")