Update app.py

app.py

@@ -186,91 +186,30 @@ if prompt := st.chat_input("How can I help you today?"):
         sound = AudioSegment.from_mp3("/home/user/app/audio/audio.mp3")
         sound.export("/home/user/app/audio/audio.wav", format="wav")
     
-    # st.session_state['reference'] = '/home/user/app/pdf2image/default_output.png'
-    # st.session_state['audio'] = ''
-    
-    
-    
-    
-    # if "reference" not in st.session_state:
-    #     st.session_state.reference = False
-    # if "audio" not in st.session_state:
-    #     st.session_state.audio = False
-    
-    # #choice = st.radio(label ="References and TTS", options =["Reference" , 'TTS'])
-
-    # if 'onr' not in st.session_state:
-    #     st.session_state.onr = False
-    
-    # def click_button_onr():
-    #     st.session_state.onr = True
-    #     if st.session_state.onr:
-    #         with st.sidebar:
-    #             st.write('ref')
-    #             generate_pdf()
-    #             st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
-    #             st.image(st.session_state['reference'])
-        
-    # st.button('Ref', on_click=click_button_onr)
-    
-
-    
-    #onr = st.toggle('Reference',key='onr')
-    #if choice == 'Reference':
-    #if onr:
-    #    st.session_state['onr'] = True
-    #    with st.sidebar:
-    #        generate_pdf()
-    #        st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
-    #        st.image(st.session_state['reference'])
-
-
-    # if 'ontts' not in st.session_state:
-    #     st.session_state.ontts = False
-    
-    # def click_button_ontts():
-    #     st.session_state.ontts = True
-    #     if st.session_state.ontts:
-    #         with st.sidebar:
-    #             st.write('audio')
-    #             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.button('Audio', on_click=click_button_ontts)
-
-
-    if button('Reference', key = 'button2'):
-        generate_pdf()
-        #st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
-        st.image('/home/user/app/pdf2image/output.png')
-    if button('TTS', key = 'button3'):
-        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('/home/user/app/audio/audio.wav')
-            
-
-        
-        
-    # ontts = st.toggle('Audio')
-    # #if choice == 'TTS':
-    # if ontts:
-    #     with st.sidebar:
-    #         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'])
+    if "reference" not in st.session_state:
+        st.session_state.reference = False
+    if "audio" not in st.session_state:
+        st.session_state.audio = False
+
+
+    with st.sidebar:
+        choice = st.radio("References and TTS",["Reference" , 'TTS'])
+
+        if choice == 'Reference':
+            st.session_state['audio'] = True
+            generate_pdf()
+            st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+            st.image(st.session_state['reference'])
+
+        if choice == 'TTS':
+            st.session_state.reference = True    
+            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})