update

app.py

@@ -173,7 +173,6 @@ def select_balanced_samples(samples):
 def generate_random_string(length=8):
     return ''.join(random.choices(string.ascii_letters + string.digits, k=length))
 
-
 @app.route('/', methods=['GET', 'POST'])
 def index():
     logger.info("Rendering index page.")
@@ -215,7 +214,6 @@ def index():
             return "An error occurred", 500
     return render_template('index.html')
 
-
 @app.route('/experiment/<username>', methods=['GET', 'POST'])
 def experiment(username):
     try:
@@ -238,23 +236,8 @@ def experiment(username):
         file_a = None
         file_b = None
 
-        if method_a == 'No-XAI':
-            method_a_dir = ('NO_XAI')
-        elif method_a == 'Dater':
-            method_a_dir = ('DATER')
-        elif method_a == 'Chain-of-Table':
-            method_a_dir = ('COT')
-        elif method_a == 'Plan-of-SQLs':
-            method_a_dir = ('POS')
-
-        if method_b == 'No-XAI':
-            method_b_dir = ('NO_XAI')
-        elif method_b == 'Dater':
-            method_b_dir = ('DATER')
-        elif method_b == 'Chain-of-Table':
-            method_b_dir = ('COT')
-        elif method_b == 'Plan-of-SQLs':
-            method_b_dir = ('POS')
+        method_a_dir = get_method_dir(method_a)
+        method_b_dir = get_method_dir(method_b)
 
         for category in ['TP', 'TN', 'FP', 'FN']:
             dir_a = f'htmls_{method_a_dir.upper()}/{category}'
@@ -295,6 +278,17 @@ Please analyze the explanations and determine which one provides a clearer and m
         logger.exception(f"An error occurred in the experiment route: {e}")
         return "An error occurred", 500
 
+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'
+
+
 def get_visualization_dir(method):
     if method == "No-XAI":
         return 'htmls_NO_XAI'
@@ -305,6 +299,7 @@ def get_visualization_dir(method):
     else:  # Plan-of-SQLs
         return 'htmls_POS'
 
+
 @app.route('/feedback', methods=['POST'])
 def feedback():
     try: