Enh(viewer): Allow to view different directories

Author: klieret

codeclash/viewer/__init__.py

@@ -4,6 +4,6 @@
 A Flask-based web application to visualize AI agent game trajectories
 """
 
-from .app import app
+from .app import app, set_log_base_directory
 
-__all__ = ["app"]
+__all__ = ["app", "set_log_base_directory"]

codeclash/viewer/app.py

@@ -12,6 +12,15 @@
 
 from flask import Flask, jsonify, render_template, request
 
+# Global variable to store the directory to search for logs
+LOG_BASE_DIR = Path.cwd() / "logs"
+
+
+def set_log_base_directory(directory: str | Path):
+    """Set the logs directory directly"""
+    global LOG_BASE_DIR
+    LOG_BASE_DIR = Path(directory).resolve()
+
 
 @dataclass
 class GameMetadata:
@@ -139,9 +148,8 @@ def nl2br(value):
 @app.route("/")
 def index():
     """Main viewer page"""
-    # Get available log directories (relative to project root)
-    project_root = Path(__file__).parent.parent.parent
-    logs_dir = project_root / "logs"
+    # Get available log directories
+    logs_dir = LOG_BASE_DIR
     log_folders = []
     if logs_dir.exists():
         log_folders = [d.name for d in logs_dir.iterdir() if d.is_dir()]
@@ -183,8 +191,7 @@ def trajectory_detail(player_id: int, round_num: int):
     if not selected_folder:
         return jsonify({"error": "No folder specified"})
 
-    project_root = Path(__file__).parent.parent.parent
-    logs_dir = project_root / "logs"
+    logs_dir = LOG_BASE_DIR
     parser = LogParser(logs_dir / selected_folder)
     trajectory = parser.parse_trajectory(player_id, round_num)
 

run_viewer.py

@@ -3,8 +3,30 @@
 Launch script for the CodeClash Trajectory Viewer
 """
 
+import argparse
+from pathlib import Path
+
 if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="CodeClash Trajectory Viewer")
+    parser.add_argument(
+        "-d",
+        "--directory",
+        type=str,
+        default=None,
+        help="Logs directory to search for game trajectories (defaults to ./logs)",
+    )
+
+    args = parser.parse_args()
+
     from codeclash.viewer import app
+    from codeclash.viewer.app import set_log_base_directory
+
+    # Set the logs directory if provided
+    if args.directory:
+        set_log_base_directory(args.directory)
+        print(f"📁 Using logs directory: {Path(args.directory).resolve()}")
+    else:
+        print(f"📁 Using logs directory: {Path.cwd() / 'logs'}")
 
     print("🎮 Starting CodeClash Trajectory Viewer...")
     print("📊 Navigate to http://localhost:5001 to view game trajectories")