update

app.py

@@ -6,7 +6,6 @@ import string
 import logging
 from datetime import datetime
 from huggingface_hub import login, HfApi, hf_hub_download
-from bs4 import BeautifulSoup
 
 # Set up logging
 logging.basicConfig(level=logging.INFO,
@@ -135,25 +134,11 @@ def index():
             return "An error occurred", 500
     return render_template('index.html')
 
-from bs4 import BeautifulSoup
-import re
-
-def remove_prediction_from_html(html_content):
-    soup = BeautifulSoup(html_content, 'html.parser')
-
-    # Remove the Prediction line
-    prediction_h3 = soup.find('h3', string=re.compile(r'Prediction:'))
-    if prediction_h3:
-        prediction_h3.decompose()
-
-    return str(soup)
-
 @app.route('/experiment/<username>', methods=['GET', 'POST'])
 def experiment(username):
     try:
         session_data = load_session_data(username)
         if not session_data:
-            logger.error(f"Failed to load session data for user: {username}")
             return redirect(url_for('index'))
 
         selected_samples = session_data['selected_samples']
@@ -164,12 +149,8 @@ def experiment(username):
             return redirect(url_for('completed', username=username))
 
         sample = selected_samples[current_index]
-        visualization_path = f"{VISUALIZATION_DIRS[method]}/{sample['category']}/{sample['file']}"
-
-        # 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)
+        visualization_dir = VISUALIZATION_DIRS[method]
+        visualization_path = f"{visualization_dir}/{sample['category']}/{sample['file']}"
 
         statement = """
 Based on the explanation provided, what do you think the AI model will predict?
@@ -179,7 +160,7 @@ Will it predict the statement as TRUE or FALSE?
         return render_template('experiment.html',
                                sample_id=current_index,
                                statement=statement,
-                               visualization_content=modified_html,
+                               visualization=url_for('send_visualization', filename=visualization_path),
                                username=username,
                                method=method)
     except Exception as e:
@@ -187,7 +168,6 @@ Will it predict the statement as TRUE or FALSE?
         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: auto;
+            overflow: hidden;
         }
         iframe {
             width: 100%;
@@ -79,11 +79,7 @@
             <p><strong>Task description:</strong> {{ statement }}</p>
         </div>
         <div class="visualization-container">
-            {% if visualization_content %}
-                <div>{{ visualization_content | safe }}</div>
-            {% else %}
-                <p>No visualization available.</p>
-            {% endif %}
+            <iframe src="{{ visualization }}"></iframe>
         </div>
         <div class="buttons">
             <form action="{{ url_for('feedback') }}" method="post">