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,5 +66,17 @@ 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')  # Or: answer_mode='base_model'
+    main(answer_mode='base_model')