Update app.py
Browse files
app.py
CHANGED
|
@@ -3,6 +3,7 @@ import json
|
|
| 3 |
import uuid
|
| 4 |
from datetime import datetime
|
| 5 |
import os
|
|
|
|
| 6 |
|
| 7 |
class LegalTechEvaluator:
|
| 8 |
def __init__(self):
|
|
@@ -17,19 +18,23 @@ class LegalTechEvaluator:
|
|
| 17 |
"""Automatically load test_legal_tech.jsonl if it exists"""
|
| 18 |
if os.path.exists("test_legal_tech.jsonl"):
|
| 19 |
self.load_from_file("test_legal_tech.jsonl")
|
|
|
|
| 20 |
return True
|
|
|
|
| 21 |
return False
|
| 22 |
|
| 23 |
def load_from_file(self, filename):
|
| 24 |
"""Load data from JSONL file"""
|
| 25 |
self.data = []
|
| 26 |
try:
|
| 27 |
-
with open(
|
| 28 |
for line in f:
|
| 29 |
self.data.append(json.loads(line))
|
|
|
|
| 30 |
return True
|
| 31 |
except Exception as e:
|
| 32 |
print(f"Error loading file: {e}")
|
|
|
|
| 33 |
return False
|
| 34 |
|
| 35 |
def create_user_session(self, name):
|
|
@@ -45,6 +50,7 @@ class LegalTechEvaluator:
|
|
| 45 |
"completed": False
|
| 46 |
}
|
| 47 |
self.current_index = 0
|
|
|
|
| 48 |
return user_id
|
| 49 |
|
| 50 |
def record_choice(self, sample_choice):
|
|
@@ -56,6 +62,7 @@ class LegalTechEvaluator:
|
|
| 56 |
"timestamp": datetime.now().isoformat()
|
| 57 |
}
|
| 58 |
return True
|
|
|
|
| 59 |
return False
|
| 60 |
|
| 61 |
def get_current_question(self):
|
|
|
|
| 3 |
import uuid
|
| 4 |
from datetime import datetime
|
| 5 |
import os
|
| 6 |
+
from loguru import logger
|
| 7 |
|
| 8 |
class LegalTechEvaluator:
|
| 9 |
def __init__(self):
|
|
|
|
| 18 |
"""Automatically load test_legal_tech.jsonl if it exists"""
|
| 19 |
if os.path.exists("test_legal_tech.jsonl"):
|
| 20 |
self.load_from_file("test_legal_tech.jsonl")
|
| 21 |
+
logger.info("found")
|
| 22 |
return True
|
| 23 |
+
logger.info("not found")
|
| 24 |
return False
|
| 25 |
|
| 26 |
def load_from_file(self, filename):
|
| 27 |
"""Load data from JSONL file"""
|
| 28 |
self.data = []
|
| 29 |
try:
|
| 30 |
+
with open("test_legal_tech.jsonl", 'r') as f:
|
| 31 |
for line in f:
|
| 32 |
self.data.append(json.loads(line))
|
| 33 |
+
logger.info("loaded")
|
| 34 |
return True
|
| 35 |
except Exception as e:
|
| 36 |
print(f"Error loading file: {e}")
|
| 37 |
+
logger.info("failed to load")
|
| 38 |
return False
|
| 39 |
|
| 40 |
def create_user_session(self, name):
|
|
|
|
| 50 |
"completed": False
|
| 51 |
}
|
| 52 |
self.current_index = 0
|
| 53 |
+
logger.info("new session created")
|
| 54 |
return user_id
|
| 55 |
|
| 56 |
def record_choice(self, sample_choice):
|
|
|
|
| 62 |
"timestamp": datetime.now().isoformat()
|
| 63 |
}
|
| 64 |
return True
|
| 65 |
+
logger.info("choice recorded")
|
| 66 |
return False
|
| 67 |
|
| 68 |
def get_current_question(self):
|