Update app.py

app.py

@@ -197,32 +197,23 @@ if prompt := st.chat_input("How can I help you today?"):
         st.session_state.audio = False
     
     
-
-    #choice = st.radio("References and TTS",["Reference" , 'TTS'])
-    on_ref = st.toggle('Reference', key='onref')
-    st.session_state.onref = False
-
-    if on_ref:
-        st.session_state.onref = True
-        with st.sidebar:
+    with st.sidebar:
+        col1, col2 = st.columns(2)
+        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'])
-    
-    on_TTS = st.toggle('TTS', key='ontts')
-    st.session_state.ontts = False
+            col1.image(st.session_state['reference'])
 
-    if on_TTS:
-        st.session_state.ontts = True
-        with st.sidebar:
+        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'])
-    
+            col2.audio(st.session_state['audio'])
+        
     st.session_state.messages.append({"role": "assistant", "content": full_response})