Update app.py

app.py

@@ -292,14 +292,26 @@ if prompt := st.chat_input("How can I help you today?"):
     #     generate_audio()
 
 
-    on_audio = st.checkbox(':speaker:', key="speaker")
-    on_ref = st.checkbox('Reference', key="reference")
-    if on_audio:
-        generate_audio()
+    # on_audio = st.checkbox(':speaker:', key="speaker")
+    # on_ref = st.checkbox('Reference', key="reference")
+    # if on_audio:
+    #     generate_audio()
+
+    # if on_ref:
+    #     generate_pdf()
 
-    if on_ref:
-        generate_pdf()
 
+    if 'clicked' not in st.session_state:
+        st.session_state.clicked = False
+
+    def click_button():
+        st.session_state.clicked = True
+
+    st.button(':speaker:', on_click=click_button)
+
+    if st.session_state.clicked:
+        # The message and nested widget will remain on the page
+        generate_audio()
         
     st.session_state.messages.append({"role": "assistant", "content": full_response})