Update app.py

app.py

@@ -84,41 +84,41 @@ class HuggingFaceDatasetSaver(FlaggingCallback):
     flag_option: str | None = None,
     flag_index: int | None = None,
     username: str | None = None,
-) -> int:
-    self.repo.git_pull(lfs=True)
-
-    is_new = not Path(self.log_file).exists()
+    ) -> int:
+        self.repo.git_pull(lfs=True)
 
-    with open(self.log_file, "a", newline="", encoding="utf-8") as csvfile:
-        writer = csv.writer(csvfile)
+        is_new = not Path(self.log_file).exists()
 
-        # File previews for certain input and output types
-        infos, file_preview_types, headers = _get_dataset_features_info(
-            is_new, self.components
-        )
+        with open(self.log_file, "a", newline="", encoding="utf-8") as csvfile:
+           writer = csv.writer(csvfile)
+    
+           # File previews for certain input and output types
+           infos, file_preview_types, headers = _get_dataset_features_info(
+                is_new, self.components
+           )
 
-        # Generate the headers and dataset_infos
-        if is_new:
-            writer.writerow(utils.sanitize_list_for_csv(headers))
+            # Generate the headers and dataset_infos
+            if is_new:
+                writer.writerow(utils.sanitize_list_for_csv(headers))
 
-        # Generate the row corresponding to the flagged sample
-        csv_data = []
-        for component, sample in zip(self.components, flag_data):
-            save_dir = Path(
-                self.dataset_dir
-            ) / utils.strip_invalid_filename_characters(component.label or "")
-            filepath = component.deserialize(sample, save_dir, None)
-            csv_data.append(filepath)
-            if isinstance(component, tuple(file_preview_types)):
-                csv_data.append(
-                    "{}/resolve/main/{}".format(self.path_to_dataset_repo, filepath)
+            # Generate the row corresponding to the flagged sample
+            csv_data = []
+            for component, sample in zip(self.components, flag_data):
+                save_dir = Path(
+                    self.dataset_dir
+                ) / utils.strip_invalid_filename_characters(component.label or "")
+                filepath = component.deserialize(sample, save_dir, None)
+                csv_data.append(filepath)
+                if isinstance(component, tuple(file_preview_types)):
+                    csv_data.append(
+                        "{}/resolve/main/{}".format(self.path_to_dataset_repo, filepath)
                 )
 
-        # Truncate flagged output to first 100 characters
-        flagged_output = csv_data[-1][:100] if csv_data else ""
-        csv_data.append(flagged_output)
-        csv_data.append(flag_option if flag_option is not None else "")
-        writer.writerow(utils.sanitize_list_for_csv(csv_data))
+    # Truncate flagged output to first 100 characters
+    flagged_output = csv_data[-1][:100] if csv_data else ""
+    csv_data.append(flagged_output)
+    csv_data.append(flag_option if flag_option is not None else "")
+    writer.writerow(utils.sanitize_list_for_csv(csv_data))
 
     if is_new:
         json.dump(infos, open(self.infos_file, "w"))