From acbeca40c0318ccc44d2bf8d4bc22d8e5c0c7f0c Mon Sep 17 00:00:00 2001
From: Cian Hughes <chughes000@gmail.com>
Date: Thu, 8 May 2025 11:10:29 +0100
Subject: [PATCH] Another attempt to fix plugin

---
 invenio/plugin/custom_invenio_plugin/__init__.py | 13 ++++++-------
 invenio/plugin/custom_invenio_plugin/views.py    |  6 +++++-
 2 files changed, 11 insertions(+), 8 deletions(-)

diff --git a/invenio/plugin/custom_invenio_plugin/__init__.py b/invenio/plugin/custom_invenio_plugin/__init__.py
index c4249ea..63d67c8 100644
--- a/invenio/plugin/custom_invenio_plugin/__init__.py
+++ b/invenio/plugin/custom_invenio_plugin/__init__.py
@@ -1,15 +1,14 @@
 from flask import Blueprint
 
-blueprint = Blueprint(
-    "custom_menu",
-    __name__,
-)
-
 
 def init_app(app):
     """Initialize application."""
-    app.register_blueprint(blueprint)
-
     from . import views
 
+    app.register_blueprint(views.blueprint)
+
+    @app.before_first_request
+    def init_menus():
+        views.init_menu()
+
     return app
diff --git a/invenio/plugin/custom_invenio_plugin/views.py b/invenio/plugin/custom_invenio_plugin/views.py
index 0b90748..d0dee7f 100644
--- a/invenio/plugin/custom_invenio_plugin/views.py
+++ b/invenio/plugin/custom_invenio_plugin/views.py
@@ -7,7 +7,6 @@ blueprint = Blueprint(
 )
 
 
-@blueprint.before_app_first_request
 def init_menu():
     """Add custom items to main menu."""
     current_menu.submenu("main").register(
@@ -18,6 +17,11 @@ def init_menu():
     )
 
 
+@blueprint.before_app_first_request
+def init_menu_on_request():
+    init_menu()
+
+
 @blueprint.route("/debug-menu")
 def debug_menu():
     """Debug endpoint to view menu structure."""