added more text for less errors

app.py

@@ -75,6 +75,7 @@ def main():
     df, total_rows = load_dataset_with_limit(dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects, limit=1)
     dataset_limit = st.slider("Size of Dataset to be searched from", min_value=0, max_value=min(total_rows, 80000), value=int(min(total_rows, 80000)/2))
     st.text(f'The smaller the dataset the faster the search will work.')
+    st.text('Please click Load Dataset to finalise selection for search')
     
 
     # Load dataset with limit only if not already loaded
@@ -112,7 +113,8 @@ def main():
 
     # Search button
     if st.button("Search"):
-        if not df:
+        #Ensure Dataset is selected
+        if st.session_state.df = None:
             st.warning("You have not selected a Dataset. Please click on Load Dataset to load your respective Dataset")
         # Validate input
         if not query:
@@ -130,14 +132,16 @@ def main():
                     results = search(query, df, limit)
                     top_k_paths = get_file_paths(df, results)
                     top_k_cordinates = get_cordinates(df, results)
+                    search_type = 'Splits'
                 else:
                     # Normal Search
                     results = search(query, df, limit)
                     top_k_paths = get_file_paths(df, results)
+                    search_type = 'Main'
 
                 # Complete the search progress
                 search_progress_bar.progress(100)
-                search_loading_text.text("Search completed!")
+                search_loading_text.text(f"Search completed among {dataset_limit} for {dataset_name} in {search_type} {st.session_state.dataset_number}")
 
                 # Load Images with Bounding Boxes if applicable
                 if st.session_state.search_in_small_objects and top_k_paths and top_k_cordinates: