Update app.py

app.py

@@ -10,6 +10,9 @@ import os
 import time
 import csv
 from io import StringIO
+import pandas as pd
+from io import BytesIO
+import base64
 
 def download_pdf(url, output_path):
     urllib.request.urlretrieve(url, output_path)
@@ -190,3 +193,24 @@ for i in range(row_count):
     with col3:
         answer_placeholder = st.empty()  
         answer_placeholder.text_area(f'Answer {i+1}', key=f'answer{i}', value=st.session_state[f'session_answer{i}']) 
+
+# Create a list of lists containing all URLs, questions, and answers
+data = [[st.session_state.get(f'url{i}', ''), st.session_state.get(f'question{i}', ''), st.session_state.get(f'session_answer{i}', '')] for i in range(row_count)]
+
+# Convert the data to a Pandas DataFrame
+df = pd.DataFrame(data, columns=['URL', 'Question', 'Answer'])
+
+# Generate a download link for the DataFrame
+st.markdown(get_table_download_link(df), unsafe_allow_html=True)
+
+def to_csv(data):
+    output = BytesIO()
+    writer = csv.writer(output)
+    writer.writerows(data)
+    return output.getvalue().decode('utf-8')
+
+def get_table_download_link(df, filename="data.csv", text="Download CSV file"):
+    csv = df.to_csv(index=False)
+    b64 = base64.b64encode(csv.encode()).decode()  # some strings <-> bytes conversions necessary here
+    href = f'<a href="data:file/csv;base64,{b64}" download="{filename}">{text}</a>'
+    return href