clarified buttons

app.py

@@ -77,25 +77,16 @@ rag_chain = (
 
 st.title("🐔 Volker-Chat")
 
-def click_button(btn):
+def click_button(prompt):
     st.session_state.clicked = True
-    st.session_state['btn'] = btn
-
-def predefined_question(btn):
-    if btn == "btn_1":
-        prompt = "Erläutere die Säule 'Lernen' aus der Volker-App."
-    elif btn == "btn_2":
-        prompt = "Erläutere die Säule 'Tracken' aus der Volker-App."
-    elif btn == "btn_3":
-        prompt = "Erläutere die Säule 'Handeln' aus der Volker-App."
-    return prompt
+    st.session_state['prompt'] = prompt
 
 c = st.container()
 c.write("Beispielfragen")
 col1, col2, col3 = c.columns(3)
-col1.button("Was ist 'Lernen'?", on_click=click_button, args=['btn_1'])
-col2.button("Was ist 'Tracken'?", on_click=click_button, args=['btn_2'])
-col3.button("Was ist 'Handeln'?", on_click=click_button, args=['btn_3'])
+col1.button("Was ist 'Lernen'?", on_click=click_button, args=["Erläutere die Säule 'Lernen' der Volker-App."])
+col2.button("Was ist 'Tracken'?", on_click=click_button, args=["Erläutere die Säule 'Tracken' der Volker-App."])
+col3.button("Was ist 'Handeln'?", on_click=click_button, args=["Erläutere die Säule 'Handeln' der Volker-App."])
 
 if 'clicked' not in st.session_state:
     st.session_state.clicked = False
@@ -107,7 +98,7 @@ for msg in st.session_state.messages:
     st.chat_message(msg["role"]).write(msg["content"])
 
 if st.session_state.clicked:
-    prompt = predefined_question(st.session_state['btn'])
+    prompt = st.session_state['prompt']
     st.chat_message("user").write(prompt)
     with get_openai_callback() as cb:
         response = rag_chain.invoke(prompt)