Update app.py

app.py

@@ -96,15 +96,16 @@ sp.Load("dungen_dev_preview.model")
 model = tf.keras.models.load_model("dungen_dev_preview_model.keras")
 max_seq_len = 25
 
-# Logging setup
-CSV_FILE = "flagged_outputs.csv"
-if not os.path.exists(CSV_FILE):
-    with open(CSV_FILE, "w", newline="", encoding="utf-8") as csvfile: #added encoding
-        writer = csv.writer(csvfile)
-        writer.writerow(["Timestamp", "Flagged Text"])
-
 logging.basicConfig(filename="app.log", level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
 
+# Dataset setup
+FLAGGED_DATASET_ID = "InfinitodeLTD/DungenDev-FlaggedOutputs"  # Replace with your dataset ID
+try:
+    flagged_dataset = load_dataset(FLAGGED_DATASET_ID)
+except:
+    flagged_dataset = Dataset.from_dict({"Timestamp": [], "Prompt": [], "Flagged Text": []})
+    flagged_dataset.push_to_hub(FLAGGED_DATASET_ID)
+
 def generate_text(seed_text, next_words=30, temperature=0.5):
     seed_text = seed_text.strip().lower()
 
@@ -151,12 +152,17 @@ def generate_text(seed_text, next_words=30, temperature=0.5):
 
     return decoded
 
-def flag_output(text):
-    logging.info(f"Output flagged: {text}")
+def flag_output(prompt_used, text_flagged):
+    if not text_flagged.strip():  # Check for empty or whitespace-only strings
+        return "Cannot flag an empty output."
+
+    logging.info(f"Output flagged: {text_flagged} with prompt: {prompt_used}")
     timestamp = datetime.now().isoformat()
-    with open(CSV_FILE, "a", newline="", encoding="utf-8") as csvfile: #added encoding
-        writer = csv.writer(csvfile)
-        writer.writerow([timestamp, text])
+    new_data = {"Timestamp": [timestamp], "Prompt": [prompt_used], "Flagged Text": [text_flagged]}
+    new_dataset = Dataset.from_dict(new_data)
+    global flagged_dataset
+    flagged_dataset = Dataset.from_pandas(flagged_dataset.to_pandas().append(new_dataset.to_pandas(), ignore_index=True))
+    flagged_dataset.push_to_hub(FLAGGED_DATASET_ID)
     return "Output Flagged. Thank you for your feedback."
 
 with gr.Blocks() as demo: