update

app.py

@@ -135,14 +135,13 @@ def index():
             return "An error occurred", 500
     return render_template('index.html')
 
-
 from bs4 import BeautifulSoup
 import re
 
-def modify_plan_of_sqls_html(html_content):
+def remove_prediction_from_html(html_content):
     soup = BeautifulSoup(html_content, 'html.parser')
 
-    # Remove only the Prediction line
+    # Remove the Prediction line
     prediction_h3 = soup.find('h3', string=re.compile(r'Prediction:'))
     if prediction_h3:
         prediction_h3.decompose()
@@ -167,14 +166,10 @@ def experiment(username):
         sample = selected_samples[current_index]
         visualization_path = f"{VISUALIZATION_DIRS[method]}/{sample['category']}/{sample['file']}"
 
-        # For Plan-of-SQLs, modify the HTML content directly
-        if method == 'Plan-of-SQLs':
-            with open(visualization_path, 'r', encoding='utf-8') as file:
-                html_content = file.read()
-            modified_html = modify_plan_of_sqls_html(html_content)
-        else:
-            # For other methods, just use the file path
-            modified_html = None
+        # Modify the HTML content for all methods
+        with open(visualization_path, 'r', encoding='utf-8') as file:
+            html_content = file.read()
+        modified_html = remove_prediction_from_html(html_content)
 
         statement = """
 Based on the explanation provided, what do you think the AI model will predict?
@@ -184,14 +179,15 @@ Will it predict the statement as TRUE or FALSE?
         return render_template('experiment.html',
                                sample_id=current_index,
                                statement=statement,
-                               visualization=url_for('send_visualization', filename=visualization_path) if method != 'Plan-of-SQLs' else None,
-                               visualization_content=modified_html if method == 'Plan-of-SQLs' else None,
+                               visualization_content=modified_html,
                                username=username,
                                method=method)
     except Exception as e:
         logger.exception(f"An error occurred in the experiment route: {e}")
         return "An error occurred", 500
 
+
+
 @app.route('/feedback', methods=['POST'])
 def feedback():
     try:

templates/experiment.html

@@ -81,8 +81,6 @@
         <div class="visualization-container">
             {% if visualization_content %}
                 <div>{{ visualization_content | safe }}</div>
-            {% elif visualization %}
-                <iframe src="{{ visualization }}"></iframe>
             {% else %}
                 <p>No visualization available.</p>
             {% endif %}