Update ui_utils.py

ui_utils.py

@@ -24,7 +24,14 @@ def display_class_dropdown(client):
     elif st.session_state.query_flow:
         existing_classes = [cls["class"] for cls in client.schema.get()["classes"]]
         class_options = existing_classes + ["Query all data"]
-        return st.selectbox("Select a class or query all data:", class_options, key="class_selector_query")
+        selected_option = st.selectbox("Select a class or query all data:", class_options, key="class_selector_query")
+        if selected_option == "Query all data":
+            # If "Query all data" is selected, return the first class as a default for now
+            # You can modify this behavior as needed
+            return existing_classes[0]
+        else:
+            return selected_option
+
 
 def handle_new_class_selection(client, selected_class):
     if selected_class == "New Class":