Update ui_utils.py

ui_utils.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import weaviate_utils
+import tapas_utils
 
 def display_initial_buttons():
     if "upload_flow" not in st.session_state:
@@ -48,3 +49,17 @@ def display_query_input():
             # 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)
+
+def query_tapas_with_weaviate_data(client, selected_class, question, tokenizer, model):
+    # 1. Perform hybrid search
+    data = weaviate_utils.hybrid_search_weaviate(client, selected_class, question)
+
+    # 2. Convert the data to TAPAS format
+    table = weaviate_utils.convert_to_tapas_format(data)
+
+    # 3. Call TAPAS with the table and the question
+    answers = tapas_utils.ask_llm_chunk(tokenizer, model, table, [question])
+
+    # Display the answers
+    for answer in answers:
+        st.write(f"Answer: {answer}")