Update app.py

app.py

@@ -1,18 +1,24 @@
+import streamlit as st
 import zipfile
 import os
 
-# Specify the path to the .zip file
-zip_file_path = "cspc_db.zip"
+# Function to unzip the 'cspc_db.zip' file and save it in the 'cspc_db' folder
+def unzip_file(zip_file_path, extract_to_path):
+    # Check if the destination directory exists, and if not, create it
+    if not os.path.exists(extract_to_path):
+        os.makedirs(extract_to_path)
 
-# Specify the directory where you want to extract the files
-extract_to_path = "cspc_db"
+    # Unzip the file
+    with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
+        zip_ref.extractall(extract_to_path)
 
-# Check if the destination directory exists, and if not, create it
-if not os.path.exists(extract_to_path):
-    os.makedirs(extract_to_path)
+    st.success(f"Files have been extracted to: {extract_to_path}")
 
-# Unzip the file
-with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
-    zip_ref.extractall(extract_to_path)
+# Define the path for the zip file and the extraction folder
+zip_file_path = "cspc_db.zip"
+extract_to_path = "cspc_db"
 
-print(f"Files have been extracted to: {extract_to_path}")
+# Add a button to trigger the unzip process
+if st.button('Unzip CSPC Database'):
+    unzip_file(zip_file_path, extract_to_path)
+    st.write(f"Unzipped files are available in the '{extract_to_path}' folder.")