Update ui_utils.py

ui_utils.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import weaviate_utils
 
 def display_initial_buttons():
     if "upload_flow" not in st.session_state:
@@ -18,11 +19,11 @@ def display_class_dropdown(client):
     if st.session_state.upload_flow:
         existing_classes = [cls["class"] for cls in client.schema.get()["classes"]]
         class_options = existing_classes + ["New Class"]
-        return st.selectbox("Select a class or create a new one:", class_options)
+        return st.selectbox("Select a class or create a new one:", class_options, key="class_selector_upload")
     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)
+        return st.selectbox("Select a class or query all data:", class_options, key="class_selector_query")
 
 def handle_new_class_selection(client, selected_class):
     if selected_class == "New Class":
@@ -31,18 +32,19 @@ def handle_new_class_selection(client, selected_class):
         if class_name and class_description:
             if st.button("Create Vector DB Class"):
                 # Call function to create new class schema in Weaviate
-                create_new_class_schema(class_name, class_description)
+                weaviate_utils.create_new_class_schema(client, class_name, class_description)
 
 def csv_upload_and_ingestion(client, selected_class):
-    csv_file = st.file_uploader("Upload a CSV file", type=["csv"])
+    csv_file = st.file_uploader("Upload a CSV file", type=["csv"], key="csv_uploader")
     if csv_file:
         if st.button("Confirm CSV upload"):
             # Call function to ingest CSV data into Weaviate
-            ingest_csv_to_weaviate(csv_file, selected_class)
+            weaviate_utils.ingest_data_to_weaviate(client, csv_file, selected_class)
 
 def display_query_input():
     question = st.text_input("Enter your question:")
     if question:
         if st.button("Submit Query"):
             # Call function to query TAPAS with selected data and entered question
+            # (This function needs to be implemented)
             query_tapas_with_weaviate_data(st.session_state.data_source, question)