Update app.py

app.py

@@ -282,15 +282,25 @@ if prompt := st.chat_input("How can I help you today?"):
     sound_output = st.empty()
     
     # Create a button to display the image
-    if st.button(":Reference:"):
-        image_output.clear()
-        generate_pdf()
+    # if st.button("Reference"):
+    #     image_output.clear()
+    #     generate_pdf()
     
-    # Create a button to play the sound
-    if st.button(":speaker:"):
-        sound_output.clear()
+    # # Create a button to play the sound
+    # if st.button(":speaker:"):
+    #     sound_output.clear()
+    #     generate_audio()
+
+
+    on_audio = st.toggle(':speaker:')
+    on_ref = st.toggle('Reference')
+    if on_audio:
         generate_audio()
 
+    if on_ref:
+        generate_pdf()
+
+        
     st.session_state.messages.append({"role": "assistant", "content": full_response})