added loading dataset progress bar

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from helper import (
+from helper3 import (
     load_dataset, search, get_file_paths,
     get_cordinates, get_images_from_s3_to_display,
     get_images_with_bounding_boxes_from_s3, load_dataset_with_limit
@@ -62,15 +62,24 @@ def main():
         st.text(f"You have selected Main Dataset {st.session_state.dataset_number}")
 
     
-    dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=10000, value=5000)
+    dataset_limit = st.slider("Size of Dataset to be searched from", min_value=1000, max_value=20000, value=10000)
     st.text(f'The smaller the dataset the faster the search will work.')
     
     # Load dataset with limit only if not already loaded
     if st.button("Load Dataset"):
         try:
+            loading_dataset_text = st.empty()
+            loading_dataset_text.text("Loading Dataset...")
+            loading_dataset_bar = st.progress(0)
+            # Simulate dataset loading progress
+            for i in range(0, 100, 25):
+                time.sleep(0.2)  # Simulate work being done
+                loading_dataset_bar.progress(i + 25)
             df, total_rows = load_dataset_with_limit(dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects, limit=dataset_limit)
             # Store loaded dataset in session state
             st.session_state.df = df
+            loading_dataset_bar.progress(100)
+            loading_dataset_text.text("Dataset loaded successfully!")
             st.success(f"Dataset loaded successfully with {len(df)} rows.")
             
         except Exception as e: