update

app.py

@@ -34,6 +34,16 @@ VISUALIZATION_DIRS = {
     "Plan-of-SQLs": "htmls_POS_mod2"
 }
 
+def get_method_dir(method):
+    if method == 'No-XAI':
+        return 'NO_XAI'
+    elif method == 'Dater':
+        return 'DATER'
+    elif method == 'Chain-of-Table':
+        return 'COT'
+    elif method == 'Plan-of-SQLs':
+        return 'POS'
+
 METHODS = ["No-XAI", "Dater", "Chain-of-Table", "Plan-of-SQLs"]
 
 def save_session_data(username, data):
@@ -232,7 +242,7 @@ def completed(username):
             visualization_file = session_data['selected_samples'][sample_id]['file']
             index = visualization_file.split('-')[1].split('.')[0]
 
-            ground_truth_key = f"{method.upper().replace('-', '_')}_test-{index}.html"
+            ground_truth_key = f"{get_method_dir(method)}_test-{index}.html"
 
             if ground_truth_key in ground_truth:
                 model_prediction = ground_truth[ground_truth_key]['answer'].upper()