Update app.py

app.py

@@ -42,8 +42,15 @@ 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)
-ui_utils.display_query_input()
+
+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)