Prevent loading dataset if Help checkbox is not checked

app.py

@@ -7,18 +7,6 @@ from huggingface_hub import InferenceClient
 api_key = os.getenv("HF_API_KEY")
 client = InferenceClient(api_key=api_key)
 
-# Load the dataset
-dataset = load_dataset("andreska/adregadocs", split="test")
-
-# Function to read the content from the dataset
-def read_dataset(dataset):
-    text = []
-    for item in dataset:
-        text.append(item['text'])
-    return "\n".join(text)
-
-context = read_dataset(dataset)
-
 # Inject custom CSS to change the background color to yellow
 st.markdown(
     """
@@ -39,6 +27,20 @@ if 'conversation' not in st.session_state:
 if 'include_context' not in st.session_state: 
     st.session_state.include_context = False
 
+if include_context:
+    dataset = load_dataset("andreska/adregadocs", split="test")
+
+    # Function to read the content from the dataset
+    def read_dataset(dataset):
+        text = []
+        for item in dataset:
+            text.append(item['text'])
+        return "\n".join(text)
+
+    context = read_dataset(dataset)
+else:
+    context = ""
+
 def handle_submit():
     user_input = st.session_state.user_input
     if user_input:
@@ -66,5 +68,8 @@ def handle_submit():
     else:
         st.write("Please enter a question.")
 
+if st.button("Ask"): 
+    handle_submit()
+
 st.text_input('Ask me a question', key='user_input', on_change=handle_submit)
 st.text_area("Conversation History", value=st.session_state.conversation, height=300, max_chars=None)