HemanM commited on
Commit
bb51229
Β·
verified Β·
1 Parent(s): 9e38e5c

Update logger.py

Browse files
Files changed (1) hide show
  1. logger.py +6 -2
logger.py CHANGED
@@ -1,5 +1,6 @@
1
  import csv
2
  from datetime import datetime
 
3
 
4
  def log_feedback(query, rag_context, evo_output, feedback):
5
  row = {
@@ -10,8 +11,11 @@ def log_feedback(query, rag_context, evo_output, feedback):
10
  "label": 1 if "πŸ‘" in feedback else 0 if "πŸ‘Ž" in feedback else "",
11
  }
12
 
13
- with open("feedback_log.csv", mode="a", newline="", encoding="utf-8") as file:
 
 
 
14
  writer = csv.DictWriter(file, fieldnames=row.keys())
15
- if file.tell() == 0:
16
  writer.writeheader()
17
  writer.writerow(row)
 
1
  import csv
2
  from datetime import datetime
3
+ import os
4
 
5
  def log_feedback(query, rag_context, evo_output, feedback):
6
  row = {
 
11
  "label": 1 if "πŸ‘" in feedback else 0 if "πŸ‘Ž" in feedback else "",
12
  }
13
 
14
+ file_path = "feedback_log.csv"
15
+
16
+ file_exists = os.path.exists(file_path)
17
+ with open(file_path, mode="a", newline="", encoding="utf-8") as file:
18
  writer = csv.DictWriter(file, fieldnames=row.keys())
19
+ if not file_exists:
20
  writer.writeheader()
21
  writer.writerow(row)