diff --git a/planet/controllers/admin/controls.py b/planet/controllers/admin/controls.py
index 1cf9d1a..2b600e4 100644
--- a/planet/controllers/admin/controls.py
+++ b/planet/controllers/admin/controls.py
@@ -2,7 +2,7 @@
import os
from tempfile import mkstemp
-from flask import Blueprint, Response, redirect, url_for, request, flash, abort, current_app
+from flask import Blueprint, Markup, redirect, url_for, request, flash, abort, current_app
from flask_login import login_required
from planet import cache
@@ -740,9 +740,14 @@ def calculate_enrichment():
:return: Redirect to admin main screen
"""
- CoexpressionCluster.calculate_enrichment()
+ try:
+ CoexpressionCluster.calculate_enrichment()
+ except Exception as e:
+ flash(Markup('An error occurred! Please ensure the file is correctly formatted' +
+ ' and update the counts if necessary'), 'warning')
+ finally:
+ flash('Successfully calculated GO enrichment for co-expression clusters', 'success')
- flash('Successfully calculated GO enrichment for co-expression clusters', 'success')
return redirect(url_for('admin.controls.index'))