Showing checkpoint directory in headlines

jvm123 · jvm123 · commit 46142c9f1c59 · 2025-06-06T08:58:33.000+02:00
diff --git a/scripts/visualizer.py b/scripts/visualizer.py
@@ -189,6 +189,10 @@
                     renderGraph(data);
                     lastDataStr = dataStr;
                 }
+
+                // set headline to include data.checkpoint_dir
+                let title = "OpenEvolve Evolution Visualizer | Checkpoint: " + data.checkpoint_dir;
+                document.querySelector('h1').textContent = title;
             });
     }
     fetchAndRender();
@@ -218,7 +222,7 @@
         </style>
     </head>
     <body>
-        <h1>Program ID: {{ program_data.id }}</h1>
+        <h1>Checkpoint: {{checkpoint_dir}} | Program ID: {{ program_data.id }}</h1>
         <h2>Island: {{ program_data.island }}</h2>
         <h3>Generation: {{ program_data.generation }}</h3>
         <h3>Parent ID: {{ program_data.parent_id or 'None' }}</h3>
@@ -236,12 +240,13 @@
 
 logger = logging.getLogger("openevolve.visualizer")
 
+
 def find_latest_checkpoint(base_folder):
     # Check whether the base folder is itself a checkpoint folder
-    if os.path.basename(base_folder).startswith('checkpoint_'):
+    if os.path.basename(base_folder).startswith("checkpoint_"):
         return base_folder
 
-    checkpoint_folders = glob.glob('**/checkpoint_*', root_dir=base_folder, recursive=True)
+    checkpoint_folders = glob.glob("**/checkpoint_*", root_dir=base_folder, recursive=True)
     if not checkpoint_folders:
         logger.info(f"No checkpoint folders found in {base_folder}")
         return None
@@ -250,50 +255,55 @@ def find_latest_checkpoint(base_folder):
     logger.debug(f"Found checkpoint folder: {checkpoint_folders[0]}")
     return checkpoint_folders[0]
 
+
 def load_evolution_data(checkpoint_folder):
-    meta_path = os.path.join(checkpoint_folder, 'metadata.json')
-    programs_dir = os.path.join(checkpoint_folder, 'programs')
+    meta_path = os.path.join(checkpoint_folder, "metadata.json")
+    programs_dir = os.path.join(checkpoint_folder, "programs")
     if not os.path.exists(meta_path) or not os.path.exists(programs_dir):
         logger.info(f"Missing metadata.json or programs dir in {checkpoint_folder}")
-        return {"nodes": [], "edges": []}
+        return {"nodes": [], "edges": [], "checkpoint_dir": checkpoint_folder}
     with open(meta_path) as f:
         meta = json.load(f)
 
     nodes = []
     id_to_program = {}
-    for island_idx, id_list in enumerate(meta.get('islands', [])):
+    for island_idx, id_list in enumerate(meta.get("islands", [])):
         for pid in id_list:
             prog_path = os.path.join(programs_dir, f"{pid}.json")
             if os.path.exists(prog_path):
                 with open(prog_path) as pf:
                     prog = json.load(pf)
-                prog['island'] = island_idx
+                prog["island"] = island_idx
                 nodes.append(prog)
                 id_to_program[pid] = prog
             else:
                 logger.debug(f"Program file not found: {prog_path}")
 
     edges = []
     for prog in nodes:
-        parent_id = prog.get('parent_id')
+        parent_id = prog.get("parent_id")
         if parent_id and parent_id in id_to_program:
-            edges.append({"source": parent_id, "target": prog['id']})
+            edges.append({"source": parent_id, "target": prog["id"]})
 
     logger.info(f"Loaded {len(nodes)} nodes and {len(edges)} edges from {checkpoint_folder}")
-    return {"nodes": nodes, "edges": edges}
+    return {"nodes": nodes, "edges": edges, "checkpoint_dir": checkpoint_folder}
 
-@app.route('/')
+
+@app.route("/")
 def index():
     return render_template_string(HTML_TEMPLATE)
 
+
 checkpoint_dir = None  # Global variable to store the checkpoint directory
-@app.route('/api/data')
+
+
+@app.route("/api/data")
 def data():
-    base_folder = os.environ.get('EVOLVE_OUTPUT', 'examples/')
+    base_folder = os.environ.get("EVOLVE_OUTPUT", "examples/")
     checkpoint = find_latest_checkpoint(base_folder)
     if not checkpoint:
         logger.info(f"No checkpoints found in {base_folder}")
-        return jsonify({"nodes": [], "edges": []})
+        return jsonify({"nodes": [], "edges": [], "checkpoint_dir": ""})
 
     # Memorize checkpoint directory for usage in other routes
     global checkpoint_dir
@@ -318,27 +328,34 @@ def program_page(program_id):
     with open(program_path) as f:
         program_data = json.load(f)
 
-    return render_template_string(HTML_TEMPLATE_PROGRAM_PAGE, program_data=program_data)
+    return render_template_string(HTML_TEMPLATE_PROGRAM_PAGE, program_data=program_data, checkpoint_dir=checkpoint_dir)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     import argparse
 
-    parser = argparse.ArgumentParser(description='OpenEvolve Evolution Visualizer')
-    parser.add_argument('--path', type=str, default='examples/',
-                        help='Path to openevolve_output or checkpoints folder')
-    parser.add_argument('--host', type=str, default='127.0.0.1')
-    parser.add_argument('--port', type=int, default=8080)
-    parser.add_argument('--log-level', type=str, default='INFO',
-                        help='Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)')
+    parser = argparse.ArgumentParser(description="OpenEvolve Evolution Visualizer")
+    parser.add_argument(
+        "--path",
+        type=str,
+        default="examples/",
+        help="Path to openevolve_output or checkpoints folder",
+    )
+    parser.add_argument("--host", type=str, default="127.0.0.1")
+    parser.add_argument("--port", type=int, default=8080)
+    parser.add_argument(
+        "--log-level",
+        type=str,
+        default="INFO",
+        help="Logging level (DEBUG, INFO, WARNING, ERROR, CRITICAL)",
+    )
     args = parser.parse_args()
 
     log_level = getattr(logging, args.log_level.upper(), logging.INFO)
-    logging.basicConfig(
-        level=log_level,
-        format='[%(asctime)s] %(levelname)s %(name)s: %(message)s'
-    )
+    logging.basicConfig(level=log_level, format="[%(asctime)s] %(levelname)s %(name)s: %(message)s")
 
-    os.environ['EVOLVE_OUTPUT'] = args.path
-    logger.info(f"Starting server at http://{args.host}:{args.port} with log level {args.log_level.upper()}")
+    os.environ["EVOLVE_OUTPUT"] = args.path
+    logger.info(
+        f"Starting server at http://{args.host}:{args.port} with log level {args.log_level.upper()}"
+    )
     app.run(host=args.host, port=args.port, debug=True)
