Update app.py

app.py

@@ -42,18 +42,7 @@ ui_utils.display_initial_buttons()
 selected_class = ui_utils.display_class_dropdown(client)
 ui_utils.handle_new_class_selection(client, selected_class)
 ui_utils.csv_upload_and_ingestion(client, selected_class)
-
-def display_query_input():
-    if st.session_state.upload_flow:
-        question = st.text_input("Enter your question:", key="upload_query_input")
-    elif st.session_state.query_flow:
-        question = st.text_input("Enter your question:", key="query_flow_input")
-
-question = st.text_input("Enter your question:", key="tapas_query_input")
-
-if question:
-    if st.button("Submit Query"):
-        ui_utils.query_tapas_with_weaviate_data(client, selected_class, question, tokenizer, model)
+ui_utils.display_query_input(client, selected_class, tokenizer, model)  # Updated this line
 
 # Initialize session state attributes
 if "debug" not in st.session_state: