Update app.py

app.py

@@ -25,6 +25,7 @@ from langchain.vectorstores import FAISS
 import transformers
 from pydub import AudioSegment
 #from streamlit_extras.stateful_button import button
+from streamlit_card import card
 
 user_session_id = uuid.uuid4()
 
@@ -174,40 +175,49 @@ if prompt := st.chat_input("How can I help you today?"):
             pdf_document.close()
         pdf_page_to_image('/home/user/app/pdf2image/output.pdf', page_number, '/home/user/app/pdf2image/output.png')
         #image = Image.open('/home/user/app/pdf2image/output.png')
+        st.image('/home/user/app/pdf2image/output.png')
         #message_placeholder.image(image)
         #st.session_state.reference = True
-    
-    
-    
+
+
     def generate_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")
-    
-    if "reference" not in st.session_state:
-        st.session_state.reference = '/home/user/app/pdf2image/default_output.png'
-    if "audio" not in st.session_state:
-        st.session_state.audio = "/home/user/app/audio/beep-06.mp3"
-
-
-    with st.sidebar:
-        choice = st.radio("References and TTS",["Reference" , 'TTS'], index=None,)
-
-        if choice == 'Reference':
-            generate_pdf()
-            st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
-            st.image(st.session_state['reference'])
-
-        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.audio("/home/user/app/audio/audio.wav")
+
+    card(title="Reference",
+         url=None,
+         on_click=generate_pdf)
+
+    card(title='Audio',
+        url = None,
+        on_click = generate_audio)
+
+    # if "reference" not in st.session_state:
+    #     st.session_state.reference = '/home/user/app/pdf2image/default_output.png'
+    # if "audio" not in st.session_state:
+    #     st.session_state.audio = "/home/user/app/audio/beep-06.mp3"
+
+
+    # with st.sidebar:
+    #     choice = st.radio("References and TTS",["Reference" , 'TTS'], index=None,)
+
+    #     if choice == 'Reference':
+    #         generate_pdf()
+    #         st.session_state['reference'] = '/home/user/app/pdf2image/output.png'
+    #         st.image(st.session_state['reference'])
+
+    #     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})