Update ui_utils.py

ui_utils.py

@@ -49,14 +49,15 @@ def csv_upload_and_ingestion(client, selected_class):
 def display_query_input():
     question = st.text_input("Enter your question:", key="query_input")
     
-    # Check if the question input is not empty
-    if question:
+    # Display the "Submit Query" button if CSV has been uploaded
+    if st.session_state.get("csv_uploaded", False):
         if st.button("Submit Query"):
-            # Call function to query TAPAS with selected data and entered question
-            query_tapas_with_weaviate_data(client, selected_class, question, tokenizer, model)
-    elif st.session_state.get("csv_uploaded", False):  # Check if CSV has been uploaded
-        if st.button("Submit Query"):
-            st.warning("Please provide a text query in the 'Enter your question:' input box to proceed.")
+            if question:  # Check if the question input is not empty
+                # Call function to query TAPAS with selected data and entered question
+                query_tapas_with_weaviate_data(client, selected_class, question, tokenizer, model)
+            else:
+                st.warning("Please provide a text query in the 'Enter your question:' input box to proceed.")
+
 
 
 def query_tapas_with_weaviate_data(client, selected_class, question, tokenizer, model):