fix: refresh after submission

app.py

@@ -79,13 +79,19 @@ translation_input_fr = st.text_input(
         )
 st.session_state.translation_input_fr = translation_input_fr
 
-# Submit button
-if st.button("Submit Translation"):
-    if st.session_state.translation_input_fr or st.session_state.translation_input:
+# Save states before refreshing the page
+sentence = st.session_state.sentence
+translation_input = st.session_state.translation_input
+translation_input_fr = st.session_state.translation_input_fr
+
+submit_button = st.button("Submit Translation",
+                           on_click=fetch_sentence, args=(dataset,))
+if submit_button:
+    if translation_input_fr or translation_input:
         store_submission(scheduler,
-                    st.session_state.sentence, 
-                    st.session_state.translation_input,
-                    st.session_state.translation_input_fr
+                    sentence, 
+                    translation_input,
+                    translation_input_fr
                         )
     else:
         st.warning("Please enter a translation before submitting.")

src/components.py

@@ -15,8 +15,12 @@ submissions_folder = "submissions"
 submissions_file = os.path.join(submissions_folder, f"submissions_{uuid.uuid4()}.json")
 
 
-def load_data(repo_id):
-    dataset = load_dataset(f'{repo_id}', name='sentences', split='sentences')
+def load_data(repo_id, column_name="darija_ar"):
+    dataset = load_dataset(
+        f'{repo_id}',
+        name='sentences',
+        split='sentences')
+    dataset = dataset.select_columns(column_name)
     return dataset
 
 def fetch_sentence(dataset, column_name="darija_ar"):
@@ -41,7 +45,7 @@ def store_submission(
     ts = dt.datetime.now().strftime("%Y-%m-%d_%H-%M-%S-%f")
 
     with scheduler.lock:
-        with open(submissions_file, "a") as f:
+        with open(submissions_file, "a", encoding='utf-8') as f:
             f.write(json.dumps({
                 "darija": translation_fr, 
                 "eng": translation,