Update app.py

app.py

@@ -131,8 +131,8 @@ if selected_app == "1) Create CSVs":
     if st.session_state.success != None:
         st.success("Success! Download the Q/A pairs below / Click reset to upload more PDFs")
         st.download_button(
-            label="Download CSV",
-            data=st.session_state.success,
+            label=f"Download CSV: length = {st.session_state.success[1]}",
+            data=st.session_state.success[0],
             file_name='questions_answers.csv',
             mime='text/csv',
         )
@@ -188,11 +188,12 @@ if selected_app == "1) Create CSVs":
                     df['answers'] = df.apply(lambda row: get_answers(row, answer_protocol), axis=1)
             
                     df = df.drop('context', axis=1)
+                    length = len(df)
                     
                     csv = df.to_csv(index=False).encode('utf-8')
 
                     st.session_state.clear()
-                    st.session_state.success = csv
+                    st.session_state.success = (csv, length)
                     st.rerun()
             else:
                 st.session_state.clear()
@@ -206,8 +207,8 @@ if selected_app == "2) Merge CSVs":
     if st.session_state.success != None:
         st.success("Success! Download the merged CSV with Q/A pairs below / Reset to merge more CSVs")
         st.download_button(
-            label="Download CSV",
-            data=st.session_state.success,
+            label=f"Download CSV: length = {st.session_state.success[1]}",
+            data=st.session_state.success[0],
             file_name='questions_answers.csv',
             mime='text/csv',
         )
@@ -238,10 +239,12 @@ if selected_app == "2) Merge CSVs":
                         st.rerun()
                         
                 df = pd.concat(dfs, ignore_index=True)
+                length = len(df)
+                
                 csv = df.to_csv(index=False).encode('utf-8')
                 
                 st.session_state.clear()
-                st.session_state.success = csv
+                st.session_state.success = (csv, length)
                 st.rerun()
                 
             else: