Move logic for loading streamlined-UI templates to reusable function

1 files changed, 4 insertions, 1 deletions

diff --git a/uploader/base_routes.py b/uploader/base_routes.py
index 80a15a0..cc2a270 100644
--- a/uploader/base_routes.py
+++ b/uploader/base_routes.py
@@ -11,6 +11,9 @@ from flask import (flash,
                    current_app as app,
                    send_from_directory)
 
+
+from uploader.sui import sui_template
+
 from uploader.flask_extensions import url_for
 from uploader.ui import make_template_renderer
 from uploader.oauth2.client import user_logged_in
@@ -42,7 +45,7 @@ def index():
         print("We found a species ID. Processing...")
         if not bool(request.args.get("species_id")):
             return render_template(
-                "sui-index.html",# TODO: Rename: sui-index.html, sui_base.html
+                sui_template("index.html"),
                 gn2server_intro=urljoin(app.config["GN2_SERVER_URL"], "/intro"),
                 species=all_species(conn),
                 streamlined_ui=streamlined_ui)