Update app.py

app.py

@@ -151,24 +151,23 @@ if submit:
             # Display conversation history
             for message in st.session_state['flowmessages']:
                 if isinstance(message, AIMessage):
-                    st.header("Doctor AI", divider=True)
-                    st.write(message.content)
-                elif isinstance(message, HumanMessage):
-                    st.header(":blue[You]", divider=True)
-                    st.write(message.content)
-
-            # Text-to-speech
-            audio_response = openai.audio.speech.create(
-                model="tts-1",
-                voice="nova",
-                input=response,
-                response_format="mp3",
-                speed=1.0
-            )
-            
-            # Embed audio in the webpage without saving it
-            st.header(':blue[Listen or Download] :loud_sound:')
-            st.audio(audio_response.content,format="audio/wav",start_time=0)  
+                    # Display Doctor AI's response header and content
+                    doctor_ai_header = st.header("Doctor AI", divider=True)
+                    doctor_ai_content = st.write(message.content)
+
+                    # Add a button to play the audio
+                    if st.button("🔊 Play Audio", key="audio_button"):
+                        # Text-to-speech
+                        audio_response = openai.audio.speech.create(
+                            model="tts-1",
+                            voice="nova",
+                            input=response,
+                            response_format="mp3",
+                            speed=1.0
+                        )
+                        
+                        # Play the audio
+                        st.audio(audio_response.content, format="audio/wav", start_time=0)