fix: Load non-validated examples properly

app.py

@@ -16,27 +16,28 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("foqa")
 
 
-# Load the FoQA dataset in the global scope, as it is used in multiple functions
-foqa = load_dataset(
+dataset = load_dataset(
     "alexandrainst/foqa", split="train", token=os.getenv("HF_HUB_TOKEN")
 )
-assert isinstance(foqa, Dataset)
-df = foqa.to_pandas()
-assert isinstance(df, pd.DataFrame)
+assert isinstance(dataset, Dataset)
+df = pd.DataFrame(dataset.to_pandas())
 
 
-def main():
-    def non_validated_samples() -> Generator[tuple[str, str, str], None, None]:
-        """Iterate over non-validated samples in the FoQA dataset.
+def non_validated_samples() -> Generator[tuple[str, str, str], None, None]:
+    """Iterate over non-validated samples in the FoQA dataset.
+
+    Yields:
+        A tuple (idx, question, answer) of a non-validated sample.
+    """
+    for idx, sample in df.iterrows():
+        if sample.validation is None:
+            yield str(idx), sample.question, sample.answers["text"][0]
 
-        Yields:
-            A tuple (idx, question, answer) of a non-validated sample.
-        """
-        for idx, sample in df.iterrows():
-            if sample.validation is None:
-                yield str(idx), sample.question, sample.answers["text"][0]
 
-    itr = non_validated_samples()
+itr = non_validated_samples()
+
+
+def main():
     idx, question, answer = next(itr)
 
     with gr.Blocks(theme="monochrome", title="FoQA validation") as demo:
@@ -58,7 +59,7 @@ def main():
             with gr.Column():
                 correct_btn = gr.Button(value="Correct")
                 incorrect_btn = gr.Button(value="Incorrect")
-                save_results_btn = gr.Button(value="Save results")
+                load_results_btn = gr.Button(value="Load results")
 
         correct_btn.click(
             fn=partial(assign_correct, itr=itr),
@@ -70,7 +71,7 @@ def main():
             inputs=[idx_box, question_box, answer_box],
             outputs=[idx_box, question_box, answer_box],
         )
-        save_results_btn.click(fn=partial(save_results))
+        load_results_btn.click(fn=load_results)
 
     auth = [
         ("admin", os.environ["ADMIN_PASSWORD"]),
@@ -79,7 +80,7 @@ def main():
     demo.launch(auth=auth)
 
 
-def save_results() -> None:
+def load_results() -> None:
     """Update the FoQA dataset with the validation status of a sample."""
     logger.info("Saving results...")
     gr.Info(message="Saving results...")