Update app.py

app.py

@@ -196,35 +196,22 @@ if prompt := st.chat_input("How can I help you today?"):
         st.session_state.audio = False
 
 
-    if st.button('Reference'):
-        with st.sidebar:
+    with st.sidebar:
+        choice = st.radio("References and TTS",["Reference" , 'TTS'])
+
+        if choice == 'Reference':
             generate_pdf()
             st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
             st.image(st.session_state['reference'])
-        
-        #choice = st.radio("References and TTS",["Reference" , 'TTS'])
-        #choice = st.radio("References and TTS",['TTS'])
-
-        # if choice == 'Reference':
-        #     generate_pdf()
-        #     st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
-        #     st.image(st.session_state['reference'])
-
-        # if choice == 'TTS':
-        #     generate_audio()
-        #     st.session_state['audio'] = '/home/user/app/audio/audio.wav'
-        #     st.audio(st.session_state['audio'])
-
-        if st.button('TTS'):
-            with st.sidebar:
-                with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
-                    #sound_file = BytesIO()
-                    tts = gTTS(result['result'], lang='en', tld = 'co.in')
-                    tts.write_to_fp(sound_file)
-                sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
-                sound.export("/home/user/app/audio/audio.wav", format="wav")
-                st.session_state['audio'] = '/home/user/app/audio/audio.wav'
-                st.audio(st.session_state['audio'])
+
+        if choice == 'TTS':
+            with open('/home/user/app/audio/audio.mp3','wb') as sound_file:
+                tts = gTTS(result['result'], lang='en', tld = 'co.in')
+                tts.write_to_fp(sound_file)
+            sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
+            sound.export("/home/user/app/audio/audio.wav", format="wav")
+            st.session_state['audio'] = '/home/user/app/audio/audio.wav'
+            st.audio(st.session_state['audio'])
 
     st.session_state.messages.append({"role": "assistant", "content": full_response})