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,6 +31,7 @@ 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"])
@@ -56,15 +57,27 @@ 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)
-            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})
+        
+        # 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:
             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='base_model'
+    main(answer_mode='base_model')  # Or: answer_mode='pdf_chat'