update

app.py

@@ -154,14 +154,13 @@ def select_balanced_samples(samples):
         # Ensure we have at least 10 unique samples
         unique_samples = list({(s['category'], s['file']) for s in samples})
 
-
         if len(unique_samples) < 10:
             logger.warning(f"Not enough unique samples. Only {len(unique_samples)} available.")
             selected_samples = unique_samples
         else:
             selected_samples = random.sample(unique_samples, 10)
 
-        logger.info(f"Unique sampled samples:\n{unique_samples}")
+        logger.info(f"Unique sampled samples:\n{selected_samples}")
 
         # Convert back to dictionary format
         selected_samples = [{'category': category, 'file': file} for category, file in selected_samples]
@@ -224,10 +223,15 @@ def experiment(username):
             logger.error(f"No session data found for user: {username}")
             return redirect(url_for('index'))
 
+
+        logger.info(f"Session data:\n{session_data}")
+
         selected_samples = session_data['selected_samples']
         methods = session_data['methods']
         current_index = session_data['current_index']
 
+        logger.info(f"current_index:\n{current_index}")
+
         if current_index >= len(selected_samples):
             return redirect(url_for('completed', username=username))