Update app.py

app.py

@@ -44,7 +44,27 @@ if st.button('Record audio'):
     else:
         st.error("API key is required to interact with GPT.")
 
+# # Download conversation button
+# st.download_button(label="Download conversation", 
+#                    data=pd.DataFrame(st.session_state['messages']).to_csv(index=False).encode('utf-8'), 
+#                    file_name=output_conversation_filename)
+
+# Function to generate conversation as plain text
+def generate_conversation_text(messages):
+    conversation_text = ""
+    for message in messages:
+        if message["role"] == "user":
+            conversation_text += f"Me: {message['content']}\n"
+        elif message["role"] == "assistant":
+            conversation_text += f"GPT: {message['content']}\n"
+        elif message["role"] == "system":
+            conversation_text += f"System: {message['content']}\n"
+    return conversation_text
+
 # Download conversation button
-st.download_button(label="Download conversation", 
-                   data=pd.DataFrame(st.session_state['messages']).to_csv(index=False).encode('utf-8'), 
-                   file_name=output_conversation_filename)
+st.download_button(
+    label="Download conversation", 
+    data=generate_conversation_text(st.session_state['messages']).encode('utf-8'), 
+    file_name=output_conversation_filename,
+    mime="text/plain"
+)