Update app.py

app.py

@@ -272,9 +272,25 @@ if prompt := st.chat_input("How can I help you today?"):
         tts = gTTS(result['result'], lang='en')
         tts.write_to_fp(sound_file)
         st.audio(sound_file)
+
+
+    #st.button(':speaker:', type='primary',on_click=generate_audio)
+    #st.button('Reference',type='primary',on_click=generate_pdf)
+
+    # Create placeholders for output
+    image_output = st.empty()
+    sound_output = st.empty()
     
-    st.button(':speaker:', type='primary',on_click=generate_audio)
-    st.button('Reference',type='primary',on_click=generate_pdf)
+    # Create a button to display the image
+    if st.button(":Reference:"):
+        image_output.clear()
+        generate_pdf()
+    
+    # Create a button to play the sound
+    if st.button(":speaker:"):
+        sound_output.clear()
+        generate_audio()
+
     st.session_state.messages.append({"role": "assistant", "content": full_response})