Refactor demo module handling in app.py and update navigation link in base.html

Author: ianlintner

flask_app/app.py

@@ -274,14 +274,6 @@ def index():
     # Build categories with additional top-level visualization entries for the dashboard
     categories = {k: v[:] for k, v in get_categories().items()}
 
-    # Add LeetCode problems to categories
-    from src.interview_workbook.leetcode._registry import get_all as get_all_leetcode
-
-    leetcode_problems = get_all_leetcode()
-    for problem in leetcode_problems:
-        category_key = f"leetcode/{problem['category'].value}"
-        categories.setdefault(category_key, []).append(problem)
-
     # Add visualizations
     categories.setdefault("visualizations", [])
     visualizations = [
@@ -349,11 +341,6 @@ def index():
 
 
 def run_demo(module_name: str) -> str:
-    """Import the module and execute its demo() while capturing stdout."""
-    # Defensive: ensure module_name is a valid dotted identifier, not a numeric string
-    if not isinstance(module_name, str) or module_name.strip() == "" or module_name.isdigit():
-        raise ValueError(f"Invalid module name: {module_name!r}")
-
     try:
         mod = importlib.import_module(module_name)
     except ModuleNotFoundError as e:
@@ -402,6 +389,9 @@ def demo_page():
 @app.post("/demo")
 def demo_run():
     module = request.form.get("module")
+    if not module and request.is_json:
+        data = request.get_json(silent=True) or {}
+        module = data.get("module")
     if not module:
         abort(400, description="Missing module")
 

flask_app/templates/base.html

@@ -44,9 +44,7 @@ <h1 class="brand"><a href="{{ url_for('index') }}">Python DSA Demos</a></h1>
            class="{{ 'active' if ep == 'index' else '' }}"
            {% if ep == 'index' %}aria-current="page"{% endif %}>Dashboard</a>
         <a href="{{ url_for('index') }}?view=algorithms"
-           class="{{ 'active' if request.args.get('view') == 'algorithms' else '' }}">Algorithms</a>
-        <a href="{{ url_for('index') }}?view=leetcode"
-           class="{{ 'active' if request.args.get('view') == 'leetcode' else '' }}">LeetCode</a>
+           class="{{ 'active' if request.args.get('view') == 'algorithms' else '' }}">Algorithms & Leetcode</a>
         <a href="{{ url_for('index') }}?view=visualizations"
            class="{{ 'active' if request.args.get('view') == 'visualizations' else '' }}">Visualizations</a>
         <a href="{{ url_for('big_o_guide') }}"