minor

app.py

@@ -58,13 +58,13 @@ with st.sidebar:
         chunk_overlap = st.number_input("Chunk overlap", value=def_chunk_overlap, step=10, placeholder=def_chunk_overlap, disabled=disabled)
         text_splitter = RecursiveCharacterTextSplitter(chunk_size=chunk_size, chunk_overlap=chunk_overlap)
         splits = text_splitter.split_documents(docs)
-        vectorstore = Chroma.from_documents(documents=splits, embedding=OpenAIEmbeddings(), persist_directory=directory)
+        vectorstore = Chroma.from_documents(documents=splits, embedding=OpenAIEmbeddings())
         if chunk_size != def_chunk_size | chunk_overlap != def_chunk_overlap:
             splits = text_splitter.split_documents(docs)
             vectorstore = Chroma.from_documents(documents=splits, embedding=OpenAIEmbeddings())
             print("Created new vectordb for this session.")
     else:
-        vectorstore = Chroma.from_documents(documents=docs, embedding=OpenAIEmbeddings(), persist_directory=f"{directory}/complete")
+        vectorstore = Chroma.from_documents(documents=docs, embedding=OpenAIEmbeddings())
         print("Used vectordb with all blog articles.")
 
 
@@ -132,16 +132,15 @@ def click_button(prompt):
 
 c = st.container()
 c.write("Beispielfragen")
-col1, col2, col3 = c.columns(3)
 if data_source == 'Blog articles':
-    col1.button("Mehr zu 'Lernen'", on_click=click_button, args=["Was macht die Säule 'Lernen' aus?"])
-    col1.button("Was macht die Fimo Health App?", on_click=click_button, args=["Was macht die Fimo Health App?"])
-    col2.button("Mehr zu 'Tracken'", on_click=click_button, args=["Was macht die Säule 'Tracken' aus?"])
-    col2.button("Was ist Pacing?", on_click=click_button, args=["Was ist Pacing?"])
-    col3.button("Mehr zu 'Handeln'", on_click=click_button, args=["Was macht die Säule 'Handeln' aus?"])
+    examples = ['Was ist Pacing?', 'Wie funktioniert die Wiedereingliederung?', 'Sollte ich eine Reha machen?']
+    for i, col in enumerate(c.columns(len(examples))):
+        question = examples[i]
+        col.button(question, on_click=click_button, args=[question])
+
 elif data_source == 'FAQ':
     examples = ['Wie komme ich an meinen PDF-Report?', 'Wer steckt hinter den Kurs-Inhalten?', 'Wozu dient der Check-Out?']
-    for i, col in enumerate(c.columns(3)):
+    for i, col in enumerate(c.columns(len(examples))):
         question = examples[i]
         col.button(question, on_click=click_button, args=[question])