add helper script for reviewing input into dataset

1 files changed, 3 insertions, 1 deletions

diff --git a/highlight.py b/highlight.py
index 361a0d2..ffcdd2c 100644
--- a/highlight.py
+++ b/highlight.py
@@ -1,6 +1,7 @@
 import sys
 import json
 import re
+import pathlib
 from datetime import timedelta
 
 # ANSI escape codes for terminal colors
@@ -100,7 +101,8 @@ def visualize_entry(entry: dict):
 
 def main(filepath: str):
     try:
-        with open(filepath, 'r', encoding='utf-8') as f:
+        p = pathlib.Path(filepath)
+        with p.open('r', encoding='utf-8') as f:
             for line in f:
                 try:
                     entry = json.loads(line)