update

app.py

@@ -31,14 +31,16 @@ app.config['SECRET_KEY'] = 'supersecretkey'  # Change this to a random secret ke
 SESSION_DIR = '/tmp/sessions'
 os.makedirs(SESSION_DIR, exist_ok=True)
 
-# Directories for visualizations
+# Update the VISUALIZATION_DIRS dictionary
 VISUALIZATION_DIRS = {
     "No-XAI": "htmls_NO_XAI_mod",
     "Dater": "htmls_DATER_mod2",
     "Chain-of-Table": "htmls_COT_mod",
-    "Plan-of-SQLs": "htmls_POS_mod2"
+    "Plan-of-SQLs": "htmls_POS_mod2",
+    "Text2SQL": "htmls_Text2SQL"
 }
 
+# Update the get_method_dir function
 def get_method_dir(method):
     if method == 'No-XAI':
         return 'NO_XAI'
@@ -48,10 +50,13 @@ def get_method_dir(method):
         return 'COT'
     elif method == 'Plan-of-SQLs':
         return 'POS'
+    elif method == 'Text2SQL':
+        return 'Text2SQL'
     else:
         return None
 
-METHODS = ["No-XAI", "Dater", "Chain-of-Table", "Plan-of-SQLs"]
+# Update the METHODS list
+METHODS = ["No-XAI", "Dater", "Chain-of-Table", "Plan-of-SQLs", "Text2SQL"]
 
 def generate_session_id():
     return str(uuid.uuid4())
@@ -153,7 +158,7 @@ def index():
         method = request.form.get('method')
         if not username or not seed or not method:
             return render_template('index.html', error="Please fill in all fields and select a method.")
-        if method not in ['Chain-of-Table', 'Plan-of-SQLs', 'Dater']:
+        if method not in ['Chain-of-Table', 'Plan-of-SQLs', 'Dater', 'Text2SQL']:
             return render_template('index.html', error="Invalid method selected.")
         try:
             seed = int(seed)
@@ -202,6 +207,8 @@ def explanation(session_id):
         return render_template('pos_intro.html', session_id=session_id)
     elif method == 'Dater':
         return render_template('dater_intro.html', session_id=session_id)
+    elif method == 'Text2SQL':
+        return render_template('text2sql_intro.html', session_id=session_id)
     else:
         logger.error(f"Invalid method '{method}' for session ID: {session_id}")
         return redirect(url_for('index'))
@@ -298,6 +305,7 @@ def feedback():
         logger.exception(f"Error in feedback route: {e}")
         return "An error occurred", 500
 
+# Update the completed route to include Text2SQL
 @app.route('/completed/<session_id>')
 def completed(session_id):
     try:
@@ -318,6 +326,8 @@ def completed(session_id):
             json_file = 'Tabular_LLMs_human_study_vis_6_DATER.json'
         elif method == "No-XAI":
             json_file = 'Tabular_LLMs_human_study_vis_6_NO_XAI.json'
+        elif method == "Text2SQL":
+            json_file = 'Tabular_LLMs_human_study_vis_6_Text2SQL.json'
         else:
             return "Invalid method", 400
 

templates/index.html

@@ -66,6 +66,10 @@
             background-color: #ffcc80;
             color: #e65100;
         }
+        .method-button.Text2SQL {
+            background-color: #e1bee7;
+            color: #7b1fa2;
+        }
         .method-button.Dater {
             background-color: #e8f5e9;
             color: #4caf50;
@@ -210,6 +214,9 @@
                 <div class="method-button Dater" onclick="selectMethod('Dater')">
                     Dater
                 </div>
+                <div class="method-button Text2SQL" onclick="selectMethod('Text2SQL')">
+                    Text2SQL
+                </div>
                 {% if show_no_xai %}
                 <div class="method-button No-XAI" onclick="selectMethod('No-XAI')">
                     No-XAI