Update app.py

app.py

@@ -78,24 +78,24 @@ class HuggingFaceDatasetSaver(FlaggingCallback):
         self.log_file = Path(self.dataset_dir) / "data.csv"
         self.infos_file = Path(self.dataset_dir) / "dataset_infos.json"
 
-   def flag(
-    self,
-    flag_data: List[Any],
-    flag_option: str | None = None,
-    flag_index: int | None = None,
-    username: str | None = None,
+    def flag(
+        self,
+        flag_data: List[Any],
+        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()
 
         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(
+            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:
@@ -112,23 +112,19 @@ class HuggingFaceDatasetSaver(FlaggingCallback):
                 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))
+                    )
+            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"))
+        if is_new:
+            json.dump(infos, open(self.infos_file, "w"))
 
-    with open(self.log_file, "r", encoding="utf-8") as csvfile:
-        line_count = len([None for row in csv.reader(csvfile)]) - 1
+        with open(self.log_file, "r", encoding="utf-8") as csvfile:
+            line_count = len([None for row in csv.reader(csvfile)]) - 1
 
-    self.repo.push_to_hub(commit_message="Flagged sample #{}".format(line_count))
+        self.repo.push_to_hub(commit_message="Flagged sample #{}".format(line_count))
 
-    return line_count
+        return line_count
 
 ##################