Fix save

app.py

@@ -4,6 +4,8 @@ import gradio as gr
 from download_url import download_text_and_title
 from cache_system import CacheHandler
 from gradio_client import Client
+from collections import OrderedDict
+from typing import Any
 
 print(f"CPU cores: {os.cpu_count()}.")
 
@@ -13,6 +15,36 @@ auth_token = os.environ.get("TOKEN") or True
 client = Client(server)
 
 
+class HuggingFaceDatasetSaver_custom(gr.HuggingFaceDatasetSaver):
+    def _deserialize_components(
+        self,
+        data_dir,
+        flag_data: list[Any],
+        flag_option: str = "",
+        username: str = "",
+    ) -> tuple[dict[Any, Any], list[Any]]:
+        """Deserialize components and return the corresponding row for the flagged sample.
+
+        Images/audio are saved to disk as individual files.
+        """
+        # Components that can have a preview on dataset repos
+        file_preview_types = {gr.Audio: "Audio", gr.Image: "Image"}
+
+        # Generate the row corresponding to the flagged sample
+        features = OrderedDict()
+        row = []
+        for component, sample in zip(self.components, flag_data):
+            label = component.label or ""
+            features[label] = {"dtype": "string", "_type": "Value"}
+            row.append(sample)
+
+        features["flag"] = {"dtype": "string", "_type": "Value"}
+        features["username"] = {"dtype": "string", "_type": "Value"}
+        row.append(flag_option)
+        row.append(username)
+        return features, row
+
+
 def finish_generation(text: str) -> str:
     return f"{text}\n\n⬇️ Ayuda a mejorar la herramienta marcando si el resumen es correcto o no.⬇️"
 
@@ -86,7 +118,10 @@ def generate_text(
 
 
 cache_handler = CacheHandler(max_cache_size=1000)
-hf_writer = gr.HuggingFaceDatasetSaver(auth_token, "Iker/Clickbait-News")
+hf_writer = HuggingFaceDatasetSaver_custom(
+    auth_token, "Iker/Clickbait-News", private=True, separate_dirs=False
+)
+
 
 demo = gr.Interface(
     generate_text,