Update app.py

app.py

@@ -266,12 +266,14 @@ if prompt := st.chat_input("How can I help you today?"):
     
         image = Image.open('/home/user/app/pdf2image/output.png')
         st.image(image)
+        st.session_state.image_displayed = True
 
     def generate_audio():
         sound_file = BytesIO()
         tts = gTTS(result['result'], lang='en')
         tts.write_to_fp(sound_file)
         st.audio(sound_file)
+        st.session_state.sound_played = True
 
 
     #st.button(':speaker:', type='primary',on_click=generate_audio)
@@ -300,22 +302,32 @@ if prompt := st.chat_input("How can I help you today?"):
     # 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()
-        sound_file = BytesIO()
-        tts = gTTS(result['result'], lang='en')
-        tts.write_to_fp(sound_file)
-        st.audio(sound_file)
+        # Initialize session state variables
+    if "image_displayed" not in st.session_state:
+        st.session_state.image_displayed = False
+    if "sound_played" not in st.session_state:
+        st.session_state.sound_played = False
+    
+    # Define functions to display the image and play the sound
+    # def display_image():
+    #     st.image("image.png")
+    #     st.session_state.image_displayed = True
+    
+    # def play_sound():
+    #     st.audio("sound.mp3")
+    #     st.session_state.sound_played = True
+    
+    # Create the two buttons
+    st.button("Display Image", on_click=generate_pdf)
+    st.button("Play Sound", on_click=generate_audio)
+    
+    # Check if the image has been displayed and display it if it has not
+    if not st.session_state.image_displayed:
+        generate_pdf()
+    
+    # Check if the sound has been played and play it if it has not
+    if not st.session_state.sound_played:
+        generate_audio()
         
     st.session_state.messages.append({"role": "assistant", "content": full_response})