Update app.py

app.py

@@ -10,6 +10,9 @@ def file_selector(folder_path=r'./Dataset'):
     filenames = os.listdir(folder_path)
     return filenames, folder_path
 
+def convert_df(df):
+    return df.to_csv(index=False).encode('utf-8-sig')
+
 def revert_question_type_id(txt_question_type):
     if txt_question_type == 'What':
         return 0
@@ -112,7 +115,7 @@ txt_goto = col_4.selectbox(label='Sample', label_visibility='collapsed', options
 btn_goto = col_5.button(label=':fast_forward: Move to', use_container_width=True)
 
 if len(df) != 0:
-    col_1, col_2, col_3, col_4, col_5, col_6, col_7, col_8, col_9, col_10 = st.columns(spec=[1, 1, 1, 1, 1, 1, 1, 1, 1, 1])
+    # col_1, col_2, col_3, col_4, col_5, col_6, col_7, col_8, col_9, col_10 = st.columns(spec=[1, 1, 1, 1, 1, 1, 1, 1, 1, 1])
     txt_context = st.text_area(height=200, label='Your context:', value=df['context'][st.session_state.idx])
 
     col_11, col_12, col_13 = st.columns([4.5, 1, 4.5])
@@ -157,13 +160,13 @@ if len(df) != 0:
     d_difficulty = col_eval_5.selectbox(label='[D] Difficulty:', options=options, index=options.index(d_difficulty_value))
 
     if btn_generate_distractor:
-        if filename_input == 'BiologyQA_demo.csv':
+        if filename_input == 'BiologyQA.csv':
             expert = 'biologist'
-        elif filename_input == 'GeographyQA_demo.csv':
+        elif filename_input == 'GeographyQA.csv':
             expert = 'geographer'
-        elif filename_input == 'HistoryQA_demo.csv':
+        elif filename_input == 'HistoryQA.csv':
             expert = 'historian'
-        elif filename_input == 'CivicEduQA_demo.csv':
+        elif filename_input == 'CivicEduQA.csv':
             expert = 'civic educator'
         url = 'https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent'
         headers = {'Content-Type': 'application/json'}
@@ -225,7 +228,10 @@ if len(df) != 0:
         df['d_difficulty'][st.session_state.idx] = d_difficulty
 
         df.to_csv(f'./Dataset/{filename_input}', index=None, encoding='utf-8-sig')
-        st.rerun()
+        csv_file = convert_df(df=df)
+        col_6.download_button(data=csv_file, label=':arrow_down_small: Download file', use_container_width=True, file_name="processed.csv", mime="text/csv")
+        
+        # st.rerun()
 
     if btn_goto:
         st.session_state.idx = txt_goto - 1