more experiemtns

app.py

@@ -56,6 +56,7 @@ def main():
 
     # Select dataset from dropdown
     dataset_name = st.selectbox("Select Dataset", datasets)
+    st.session_state.df = None
     #For Loading from Box
     folder_path = folder_path_dict[dataset_name]
 
@@ -65,11 +66,13 @@ def main():
         st.session_state.search_in_small_objects = True
         st.text("Small Object Search Enabled")
         st.session_state.dataset_number = st.selectbox("Select Subset of Data", list(range(1, selection[dataset_name][1] + 1)))
+        st.session_state.df = None
         st.text(f"You have selected Split Dataset {st.session_state.dataset_number}")
     else:
         st.session_state.search_in_small_objects = False
         st.text("Small Object Search Disabled")
         st.session_state.dataset_number = st.selectbox("Select Subset of Data", list(range(1, selection[dataset_name][0] + 1)))
+        st.session_state.df = None
         st.text(f"You have selected Main Dataset {st.session_state.dataset_number}")
     
     df, total_rows = load_dataset_with_limit(dataset_name, st.session_state.dataset_number, st.session_state.search_in_small_objects, limit=1)
@@ -110,13 +113,11 @@ def main():
 
     # Number of results to display
     limit = st.number_input("Number of results to display", min_value=1, max_value=10, value=10)
-
-    # Search button
-    if st.button("Search"):
-        #Ensure Dataset is selected
-        if 'df' not in st.session_state:
-            st.warning("You have not selected a Dataset. Please click on Load Dataset to load your respective Dataset")
-        else:
+    if st.session_state.df == None and st.button("Search"):
+        st.warning("Please Load Datset")
+    else:
+        # Search button
+        if st.button("Search"):
             # Validate input
             if not query:
                 st.warning("Please enter a search query.")
@@ -126,7 +127,7 @@ def main():
                     search_loading_text = st.empty()
                     search_loading_text.text("Searching...")
                     search_progress_bar = st.progress(0)
-    
+
                     # Perform search on the loaded dataset from session state
                     df = st.session_state.df
                     if st.session_state.search_in_small_objects:
@@ -139,11 +140,11 @@ def main():
                         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(f"Search completed among {dataset_limit} rows 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:
                         get_images_with_bounding_boxes_from_s3(bucket_name, top_k_paths, top_k_cordinates, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, folder_path)