Update app.py

app.py

@@ -116,46 +116,47 @@ def main():
         #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")
-        # Validate input
-        if not query:
-            st.warning("Please enter a search query.")
         else:
-            try:
-                # Progress bar for search
-                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:
-                    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(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:
-                    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)
-                elif not st.session_state.search_in_small_objects and top_k_paths:
-                    st.write(f"Displaying top {len(top_k_paths)} results for query '{query}':")
-                    get_images_from_s3_to_display(bucket_name, top_k_paths, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, folder_path)
-                    
-                else:
-                    st.write("No results found.")
-
-
-            except Exception as e:
-                st.error(f"Search failed: {e}")
+            # Validate input
+            if not query:
+                st.warning("Please enter a search query.")
+            else:
+                try:
+                    # Progress bar for search
+                    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:
+                        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(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)
+                    elif not st.session_state.search_in_small_objects and top_k_paths:
+                        st.write(f"Displaying top {len(top_k_paths)} results for query '{query}':")
+                        get_images_from_s3_to_display(bucket_name, top_k_paths, AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, folder_path)
+                        
+                    else:
+                        st.write("No results found.")
+
+
+                except Exception as e:
+                    st.error(f"Search failed: {e}")
 
 if __name__ == "__main__":
     main()