Update ui_utils.py

ui_utils.py

@@ -24,7 +24,7 @@ def display_class_dropdown(client):
         class_options = existing_classes + ["Query all data"]
         return st.selectbox("Select a class or query all data:", class_options)
 
-def handle_new_class_selection(selected_class):
+def handle_new_class_selection(client, selected_class):
     if selected_class == "New Class":
         class_name = st.text_input("Enter the new class name:")
         class_description = st.text_input("Enter a description for the class:")
@@ -33,7 +33,7 @@ def handle_new_class_selection(selected_class):
                 # Call function to create new class schema in Weaviate
                 create_new_class_schema(class_name, class_description)
 
-def csv_upload_and_ingestion(selected_class):
+def csv_upload_and_ingestion(client, selected_class):
     csv_file = st.file_uploader("Upload a CSV file", type=["csv"])
     if csv_file:
         if st.button("Confirm CSV upload"):