Update app.py

app.py

@@ -8,9 +8,21 @@ import json
 # Set the OpenAI API key
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
+class CustomHuggingFaceDatasetSaver(gr.FlaggingCallback):
+    def __init__(self, hf_token, dataset_name):
+        super().__init__()
+        self.hf_token = hf_token
+        self.dataset_name = dataset_name
+
+    def on_flag(self, sample):
+        if len(sample["output"]) > 1000:
+            sample["output"] = sample["output"][:1000]  # Truncate output to 1000 characters
+
+        super().on_flag(sample)
+
 # Set up flagging callback function
 HF_TOKEN = os.getenv("HF_TOKEN")
-hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
+hf_writer = CustomHuggingFaceDatasetSaver(HF_TOKEN, "CannaTech/Flagged")
 
 # Define the authentication function
 def check_auth(username, password):