fix: added cache, fixed async caused by data reloading

app.py

@@ -22,6 +22,13 @@ scheduler = CommitScheduler(
     every=10,
 )
 
+# Set page config
+st.set_page_config(page_title="DODa", 
+                   page_icon=DODA_LOGO,)
+
+# add an image
+st.image(DODA_LOGO, use_column_width=True)
+
 # Load the dataset
 dataset = load_data(REPO_ID)
 
@@ -35,12 +42,7 @@ if 'translation_input_fr' not in st.session_state:
 if 'display_new' not in st.session_state:
     st.session_state.display_new = False
 
-# Set page config
-st.set_page_config(page_title="DODa", 
-                   page_icon=DODA_LOGO,)
 
-# add an image
-st.image(DODA_LOGO, use_column_width=True)
 
 # title
 st.title("DODa Labeling App")
@@ -60,7 +62,7 @@ st.write(f"""
     
 
 # Display new sentence button
-st.session_state.display_new = st.button("New Sentence",
+st.session_state.display_new = st.button("Generate a new sentence",
                                             on_click=fetch_sentence,
                                             args=(dataset,))
 
@@ -70,29 +72,29 @@ translation_input = st.text_input(
     "Enter translation to english: ",
     st.session_state.translation_input
     )
-st.session_state.translation_input = translation_input
 
 # Input field for translation in latin characters
 translation_input_fr = st.text_input(
         "Enter translation to darija in latin characters: ",
         st.session_state.translation_input_fr
         )
+
+st.session_state.translation_input = translation_input
 st.session_state.translation_input_fr = translation_input_fr
 
-# 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
+print(st.session_state.sentence)
+print(st.session_state.translation_input)
+print(st.session_state.translation_input_fr)
 
-submit_button = st.button("Submit Translation",
-                           on_click=fetch_sentence, args=(dataset,))
+# Save states before refreshing the page
+submit_button = st.button("Submit Translation")
 if submit_button:
-    if translation_input_fr or translation_input:
+    if st.session_state.translation_input_fr or st.session_state.translation_input:
         store_submission(scheduler,
-                    sentence, 
-                    translation_input,
-                    translation_input_fr
+                    st.session_state.sentence, 
+                    st.session_state.translation_input,
+                    st.session_state.translation_input_fr
                         )
+        _ = fetch_sentence(dataset)
     else:
-        st.warning("Please enter a translation before submitting.")
-        
+        st.warning("Please enter a translation before submitting.")

src/components.py

@@ -5,16 +5,16 @@ import json
 import os
 from huggingface_hub import CommitScheduler
 from datasets import load_dataset
-import uuid
 
+today = dt.datetime.now().strftime("%Y%m%d_%H%M")
 
 REPO_ID = "imomayiz/darija-english"
 DATASET_REPO_URL = f"https://huggingface.co/datasets/{REPO_ID}"
 
 submissions_folder = "submissions"
-submissions_file = os.path.join(submissions_folder, f"submissions_{uuid.uuid4()}.json")
-
+submissions_file = os.path.join(submissions_folder, f"submissions_{today}.json")
 
+@st.cache_data
 def load_data(repo_id, column_name="darija_ar"):
     dataset = load_dataset(
         f'{repo_id}',
@@ -32,7 +32,6 @@ def fetch_sentence(dataset, column_name="darija_ar"):
     st.session_state.sentence = random_sentence
     st.session_state.translation_input = ""
     st.session_state.translation_input_fr = ""
-
     return random_sentence
 
 def store_submission(
@@ -57,7 +56,7 @@ def store_submission(
 
     st.success(
         f"""Translation submitted successfully.
-        You will see your commit in 1 minute at 
+        You will see your commit in a few minutes at 
         {DATASET_REPO_URL}/tree/main/{submissions_folder}.
         You can submit another translation or check the dataset."""
         )