Update app.py

app.py

@@ -5,7 +5,7 @@ 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
 from streamlit_float import *
-import io 
+
 def main(answer_mode: str):
     # Float feature initialization
     float_init()
@@ -66,17 +66,5 @@ def main(answer_mode: str):
     # Float the footer container and provide CSS to target it with
     footer_container.float("bottom: 0rem;")
 
-    # --- Download Button ---
-    conversation_text = "\n".join(
-        f"{msg['role'].capitalize()}: {msg['content']}" for msg in st.session_state.messages
-    )
-    conversation_bytes = io.BytesIO(conversation_text.encode("utf-8"))
-    st.download_button(
-        label="Download Conversation",
-        data=conversation_bytes,
-        file_name="conversation.txt",
-        mime="text/plain",
-    )
-
 if __name__ == "__main__":
-    main(answer_mode='base_model')
+    main(answer_mode='base_model')  # Or: answer_mode='base_model'