Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
+import hmac
 import os
-import textwrap
 from helpers import text_to_speech, autoplay_audio, speech_to_text, get_api_key
 from generate_answer import base_model_chatbot, with_pdf_chatbot
 from audio_recorder_streamlit import audio_recorder
@@ -31,7 +31,6 @@ def main(answer_mode: str):
     with footer_container:
         audio_bytes = audio_recorder()
 
-    # Display previous messages
     for message in st.session_state.messages:
         with st.chat_message(message["role"]):
             st.write(message["content"])
@@ -57,27 +56,15 @@ def main(answer_mode: str):
                     final_response = base_model_chatbot(st.session_state.messages)
                 elif answer_mode == 'pdf_chat':
                     final_response = with_pdf_chatbot(st.session_state.messages)
-        
-        # Display response in chunks if it's too long
-        with st.spinner("Generating audio response..."):
-            audio_files = text_to_speech(final_response)
-            autoplay_audio(audio_files)
-
-        # Use the display_text_in_chunks function to avoid truncation
-        display_text_in_chunks(final_response)
-        
-        st.session_state.messages.append({"role": "assistant", "content": final_response})
-        for audio_file in audio_files:
+            with st.spinner("Generating audio response..."):
+                audio_file = text_to_speech(final_response)
+                autoplay_audio(audio_file)
+            st.write(final_response)
+            st.session_state.messages.append({"role": "assistant", "content": final_response})
             os.remove(audio_file)
 
     # Float the footer container and provide CSS to target it with
     footer_container.float("bottom: 0rem;")
 
-def display_text_in_chunks(text, chunk_size=500):
-    """Display long text in manageable chunks."""
-    chunks = textwrap.wrap(text, chunk_size)
-    for chunk in chunks:
-        st.write(chunk)
-
 if __name__ == "__main__":
-    main(answer_mode='base_model')  # Or: answer_mode='pdf_chat'
+    main(answer_mode='base_model')  # Or: answer_mode='base_model'