scatterplot: try/except recovered

mpg-age-bioinformatics · May 15, 2020 · ff00e0e · ff00e0e
1 parent e48bd75
commit ff00e0e
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/flaski/apps/routes/scatterplot.py b/flaski/apps/routes/scatterplot.py
@@ -284,21 +284,21 @@ def scatterplot(download=None):
         df=pd.read_json(session["df"])
 
         #CALL FIGURE FUNCTION
-        # try:
-        fig=make_figure(df,plot_arguments)
+        try:
+            fig=make_figure(df,plot_arguments)
 
-        #TRANSFORM FIGURE TO BYTES AND BASE64 STRING
-        figfile = io.BytesIO()
-        plt.savefig(figfile, format='png')
-        plt.close()
-        figfile.seek(0)  # rewind to beginning of file
-        figure_url = base64.b64encode(figfile.getvalue()).decode('utf-8')
+            #TRANSFORM FIGURE TO BYTES AND BASE64 STRING
+            figfile = io.BytesIO()
+            plt.savefig(figfile, format='png')
+            plt.close()
+            figfile.seek(0)  # rewind to beginning of file
+            figure_url = base64.b64encode(figfile.getvalue()).decode('utf-8')
 
-        return render_template('/apps/scatterplot.html', figure_url=figure_url, filename=filename, apps=apps, **plot_arguments)
+            return render_template('/apps/scatterplot.html', figure_url=figure_url, filename=filename, apps=apps, **plot_arguments)
 
-        # except Exception as e:
-        #     flash(e,'error')
-        #     return render_template('/apps/scatterplot.html', filename=filename, apps=apps, **plot_arguments)
+        except Exception as e:
+            flash(e,'error')
+            return render_template('/apps/scatterplot.html', filename=filename, apps=apps, **plot_arguments)
 
     else:
         if download == "download":