update

app.py

@@ -156,30 +156,40 @@ def index():
                 'start_time': start_time,
                 'session_id': session_id
             }
-            session['data'] = session_data  # Store data in session instead of committing
-            return redirect(url_for('explanation', session_id=session_id))
+            session['data'] = session_data  # Store data in session
+            logger.info(f"Session data stored for user {username}, method {method}")
+
+            # Redirect based on the selected method
+            if method == 'No-XAI':
+                return redirect(url_for('experiment', session_id=session_id))
+            else:
+                return redirect(url_for('explanation', session_id=session_id))
         except Exception as e:
             logger.exception(f"Error in index route: {e}")
             return render_template('index.html', error="An error occurred. Please try again.")
     return render_template('index.html')
 
-
-
 @app.route('/explanation/<session_id>')
 def explanation(session_id):
     session_data = session.get('data')
     if not session_data:
+        logger.error(f"No session data found for session ID: {session_id}")
+        return redirect(url_for('index'))
+
+    method = session_data.get('method')
+    if not method:
+        logger.error(f"No method found in session data for session ID: {session_id}")
         return redirect(url_for('index'))
 
-    method = session_data['method']
     if method == 'Chain-of-Table':
         return render_template('cot_intro.html', session_id=session_id)
     elif method == 'Plan-of-SQLs':
         return render_template('pos_intro.html', session_id=session_id)
     elif method == 'Dater':
         return render_template('dater_intro.html', session_id=session_id)
-    else:  # No-XAI
-        return redirect(url_for('experiment', session_id=session_id))
+    else:
+        logger.error(f"Invalid method '{method}' for session ID: {session_id}")
+        return redirect(url_for('index'))
 
 @app.route('/experiment/<session_id>', methods=['GET', 'POST'])
 def experiment(session_id):

templates/index.html

@@ -74,7 +74,6 @@
             background-color: #fff3e0;
             color: #ff9800;
         }
-
         .task-instruction {
             background-color: #f0f8ff;
             border-left: 5px solid #4CAF50;
@@ -101,7 +100,6 @@
             font-size: 18px;
             color: #333;
         }
-
         .method-button:hover {
             opacity: 0.8;
         }
@@ -130,6 +128,11 @@
         button:hover {
             background-color: #45a049;
         }
+        .error-message {
+            color: red;
+            margin-bottom: 10px;
+            font-size: 18px;
+        }
     </style>
     <script>
         function selectMethod(method) {
@@ -156,12 +159,6 @@
             }
             return true;
         }
-
-        function goToExplanation() {
-            if (validateForm()) {
-                document.getElementById('method-form').submit();
-            }
-        }
     </script>
     <link href="https://fonts.googleapis.com/css2?family=Roboto:wght@400;700&display=swap" rel="stylesheet">
 </head>
@@ -187,6 +184,12 @@
             </div>
         </div>
 
+        {% if error %}
+        <div class="error-message">
+            {{ error }}
+        </div>
+        {% endif %}
+
         <form id="method-form" action="{{ url_for('index') }}" method="post" onsubmit="return validateForm();">
             <label for="username">Hi there 👋👋👋 ! What is your name?</label>
             <input type="text" id="username" name="username" required>
@@ -210,7 +213,7 @@
                 </div>
             </div>
 
-            <button type="button" onclick="goToExplanation()">Next</button>
+            <button type="submit">Next</button>
         </form>
     </div>
 </body>