button added

app.py

@@ -59,7 +59,7 @@ special_threshold = st.sidebar.number_input(
 st.sidebar.success(
     "The 'distances' score indicates the proximity of your question to our database questions (lower is better). The 'ai_judge' ranks the similarity between user's question and database answers independently (higher is better)."
 )
-submit_button = st.sidebar.button("Submit", type="primary")
+submit_button = st.button("Submit", type="primary")
 clear_button = st.sidebar.button("Clear Conversation", key="clear")
 if clear_button:
     st.session_state.messages = []
@@ -103,11 +103,11 @@ collection = client.create_collection(combined_string)
 # Embed and store the first N supports for this demo
 if submit_button:
     with st.spinner("Loading, please be patient with us ... 🙏"):
-        L = len(dataset["train"]["questions"][0:20])
+        L = len(dataset["train"]["questions"])
         begin_t = time.time()
         collection.add(
             ids=[str(i) for i in range(0, L)],  # IDs are just strings
-            documents=dataset["train"]["questions"][0:20],  # Enter questions here
+            documents=dataset["train"]["questions"],  # Enter questions here
             metadatas=[{"type": "support"} for _ in range(0, L)],
         )
         end_t = time.time()
@@ -132,8 +132,8 @@ if prompt := st.chat_input(initial_input):
         ref = pd.DataFrame(
             {
                 "idx": idx,
-                "questions": [dataset["train"]["questions"][0:20][i] for i in idx],
-                "answers": [dataset["train"]["answers"][0:20][i] for i in idx],
+                "questions": [dataset["train"]["questions"][i] for i in idx],
+                "answers": [dataset["train"]["answers"][i] for i in idx],
                 "distances": results["distances"][0],
             }
         )