Update app.py

app.py

@@ -193,4 +193,32 @@ for i in range(row_count):
             st.session_state[f"prompt_tokens{i}"] = prompt_tokens
             st.session_state[f"response_tokens{i}"] = response_tokens
             st.session_state[f"word_count{i}"] = word_count
-            st.experimental_rerun()
+            st.experimental_rerun()
+
+def create_download_link(data, filename):
+    csv_data = StringIO()
+    writer = csv.writer(csv_data, delimiter='\n', quotechar='"')
+    writer.writerow(data)
+    b64 = base64.b64encode(csv_data.getvalue().encode()).decode()
+    return f'<a href="data:file/csv;base64,{b64}" download="{filename}">{filename}</a>'
+
+responses_data = []
+edited_responses_data = []
+
+for i in range(row_count):
+    note = st.session_state.get(f"note{i}", "")
+    response = st.session_state.get(f"response{i}", "")
+    followup_response = st.session_state.get(f"followup_response{i}", "")
+
+    if show_notes:
+        responses_data.append(note)
+        edited_responses_data.append(note)
+
+    responses_data.append(response)
+    edited_responses_data.append(followup_response)
+
+download_responses_link = create_download_link(responses_data, "Download Responses.csv")
+download_edited_responses_link = create_download_link(edited_responses_data, "Download Edited Responses.csv")
+
+st.markdown(download_responses_link, unsafe_allow_html=True)
+st.markdown(download_edited_responses_link, unsafe_allow_html=True)