Update app.py

app.py

@@ -1,15 +1,30 @@
 from copy import deepcopy
 from langchain.callbacks import StreamlitCallbackHandler
 import streamlit as st
-import pandas as pd
-from io import StringIO
-from transformers import AutoTokenizer, AutoModelForTableQuestionAnswering
-import numpy as np
-import weaviate
-from weaviate.embedded import EmbeddedOptions
-from weaviate import Client
-from weaviate.util import generate_uuid5
 import logging
+from weaviate_utils import *
+from tapas_utils import *
+from ui_utils import *
+
+# ...
+selected_class = ui_utils.display_class_dropdown(client)
+ui_utils.handle_new_class_selection(selected_class)
+ui_utils.csv_upload_and_ingestion(selected_class)
+ui_utils.display_query_input()
+# ...
+
+# Initialize Weaviate client
+client = initialize_weaviate_client()
+
+# Initialize TAPAS
+tokenizer, model = initialize_tapas()
+
+# UI components
+display_initial_buttons()
+selected_class = display_class_dropdown(client)
+handle_new_class_selection()
+csv_upload_and_ingestion()
+display_query_input()
 
 # Initialize session state attributes
 if "debug" not in st.session_state: