Update ui_utils.py

ui_utils.py

@@ -37,12 +37,16 @@ def handle_new_class_selection(client, selected_class):
 
 def csv_upload_and_ingestion(client, selected_class):
     csv_file = st.file_uploader("Upload a CSV file", type=["csv"], key="csv_uploader")
-    if csv_file and st.button("Confirm CSV upload"):
-        dataframe = weaviate_utils.ingest_data_to_weaviate(client, csv_file, selected_class)  # Updated this line
-        if dataframe is not None:  # Check if dataframe is returned
-            st.write(f"Your CSV was successfully integrated into the vector database under the class '{selected_class}'")
-            st.write(dataframe.head())  # Display the first few rows of the dataframe as a preview
-            st.session_state.csv_uploaded = True  # Set session state variable
+    if csv_file:
+        if st.button("Confirm CSV upload"):
+            # Call function to ingest CSV data into Weaviate
+            dataframe = weaviate_utils.ingest_data_to_weaviate(client, csv_file, selected_class)  # Updated this line
+            if dataframe is not None:  # Check if ingestion was successful
+                # Display a preview of the ingested data
+                st.write(f"Your CSV was successfully integrated into the vector database under the class '{selected_class}'")
+                st.write(dataframe.head())  # Display the first few rows of the dataframe as a preview
+                st.session_state.csv_uploaded = True  # Set session state variable
+
 
 def display_query_input(client, selected_class, tokenizer, model):  # Added parameters
     question = st.text_input("Enter your question:", key="query_input")