button added

app.py

@@ -66,20 +66,21 @@ if clear_button:
 
 
 # Load the dataset from a provided source.
-if option == "YSA":
-    begin_t = time.time()
-    dataset = load_dataset(
-        "eagle0504/youthless-homeless-shelter-web-scrape-dataset-qa-formatted"
-    )
-    end_t = time.time()
-    st.success(f"{option} Database loaded. | Time: {end_t - begin_t} sec")
-    initial_input = "Tell me about YSA"
-else:
-    begin_t = time.time()
-    dataset = load_dataset("eagle0504/larkin-web-scrape-dataset-qa-formatted")
-    end_t = time.time()
-    st.success(f"{option} Database loaded. | Time: {end_t - begin_t} sec")
-    initial_input = "Tell me about Larkin"
+if submit_button:
+    if option == "YSA":
+        begin_t = time.time()
+        dataset = load_dataset(
+            "eagle0504/youthless-homeless-shelter-web-scrape-dataset-qa-formatted"
+        )
+        end_t = time.time()
+        st.success(f"{option} Database loaded. | Time: {end_t - begin_t} sec")
+        initial_input = "Tell me about YSA"
+    else:
+        begin_t = time.time()
+        dataset = load_dataset("eagle0504/larkin-web-scrape-dataset-qa-formatted")
+        end_t = time.time()
+        st.success(f"{option} Database loaded. | Time: {end_t - begin_t} sec")
+        initial_input = "Tell me about Larkin"
 
 
 # Initialize a new client for ChromeDB.