update

app.py

@@ -139,7 +139,6 @@ def index():
 from bs4 import BeautifulSoup
 import re
 
-
 def modify_plan_of_sqls_html(html_content):
     soup = BeautifulSoup(html_content, 'html.parser')
 
@@ -150,7 +149,6 @@ def modify_plan_of_sqls_html(html_content):
 
     return str(soup)
 
-
 @app.route('/experiment/<username>', methods=['GET', 'POST'])
 def experiment(username):
     try:
@@ -169,16 +167,14 @@ def experiment(username):
         sample = selected_samples[current_index]
         visualization_path = f"{VISUALIZATION_DIRS[method]}/{sample['category']}/{sample['file']}"
 
-        # Modify the HTML content for Plan-of-SQLs
+        # 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)
-            # Save the modified HTML to a temporary file
-            temp_path = f"/tmp/{sample['file']}"
-            with open(temp_path, 'w', encoding='utf-8') as file:
-                file.write(modified_html)
-            visualization_path = temp_path
+        else:
+            # For other methods, just use the file path
+            modified_html = None
 
         statement = """
 Based on the explanation provided, what do you think the AI model will predict?
@@ -188,14 +184,14 @@ 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),
+                               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,
                                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

@@ -35,7 +35,7 @@
             display: flex;
             justify-content: center;
             align-items: center;
-            overflow: hidden;
+            overflow: auto;
         }
         iframe {
             width: 100%;
@@ -79,7 +79,13 @@
             <p><strong>Task description:</strong> {{ statement }}</p>
         </div>
         <div class="visualization-container">
-            <iframe src="{{ visualization }}"></iframe>
+            {% if visualization_content %}
+                <div>{{ visualization_content | safe }}</div>
+            {% elif visualization %}
+                <iframe src="{{ visualization }}"></iframe>
+            {% else %}
+                <p>No visualization available.</p>
+            {% endif %}
         </div>
         <div class="buttons">
             <form action="{{ url_for('feedback') }}" method="post">