Feat(viewer picker): Guess config files & aws cmds

klieret · klieret · commit 80341b7a3815 · 2025-10-20T18:38:44.000-04:00
diff --git a/codeclash/viewer/app.py b/codeclash/viewer/app.py
@@ -1409,4 +1409,84 @@ def compute():
         return jsonify({"success": False, "error": str(e)}), 500
 
 
+@app.route("/picker/api/guess-config-names", methods=["POST"])
+@print_timing
+def guess_config_names():
+    """Guess config file names based on metadata.json files"""
+    try:
+        data = request.get_json()
+        folder_paths = data.get("folder_paths", [])
+
+        if not folder_paths:
+            return jsonify({"success": False, "error": "No folder paths provided"})
+
+        results = []
+        for folder_path in folder_paths:
+            folder_path_obj = LOG_BASE_DIR / folder_path
+
+            if not folder_path_obj.exists() or not is_game_folder(folder_path_obj):
+                results.append({"folder": folder_path, "config_name": None, "error": "Invalid folder"})
+                continue
+
+            metadata = load_metadata(folder_path_obj)
+            if not metadata.is_valid:
+                results.append({"folder": folder_path, "config_name": None, "error": "No metadata"})
+                continue
+
+            # Extract components for config name
+            game_name = metadata.game_name
+            rounds = metadata.get_path("config.tournament.rounds")
+            sims = metadata.get_path("config.game.sims_per_round")
+            players = metadata.get_path("config.players", [])
+
+            # Extract player names
+            player_names = []
+            for player in players:
+                if isinstance(player, dict):
+                    name = player.get("name", "")
+                    if name:
+                        player_names.append(name)
+
+            # Build config file name
+            if game_name and rounds and sims and len(player_names) >= 2:
+                # Determine correct model ordering by checking folder name
+                folder_name = Path(folder_path).name
+                model1, model2 = player_names[0], player_names[1]
+
+                # Check which order appears in the folder name
+                if f"{model1}.{model2}" in folder_name:
+                    # Order is correct
+                    pass
+                elif f"{model2}.{model1}" in folder_name:
+                    # Swap order
+                    model1, model2 = model2, model1
+                # If neither appears, keep original order
+
+                config_name = f"{game_name}__{model1}__{model2}__r{rounds}__s{sims}.yaml"
+                results.append({"folder": folder_path, "config_name": config_name, "error": None})
+            else:
+                missing = []
+                if not game_name:
+                    missing.append("game_name")
+                if not rounds:
+                    missing.append("rounds")
+                if not sims:
+                    missing.append("sims")
+                if len(player_names) < 2:
+                    missing.append(f"players (found {len(player_names)})")
+                results.append(
+                    {
+                        "folder": folder_path,
+                        "config_name": None,
+                        "error": f"Missing: {', '.join(missing)}",
+                    }
+                )
+
+        return jsonify({"success": True, "results": results})
+
+    except Exception as e:
+        logger.error(f"Error guessing config names: {e}", exc_info=True)
+        return jsonify({"success": False, "error": str(e)}), 500
+
+
 # Use run_viewer.py to launch the application
diff --git a/codeclash/viewer/static/js/picker.js b/codeclash/viewer/static/js/picker.js
@@ -373,6 +373,157 @@ function fillTextareaWithAWSResubmitCommands() {
   }
 }
 
+async function fillTextareaWithGuessedConfigNames() {
+  const selectedCheckboxes = getVisibleCheckedCheckboxes();
+
+  if (selectedCheckboxes.length === 0) {
+    showModalWarning(
+      "No folders selected. Please select folders from the table first.",
+    );
+    document.getElementById("bulk-actions-textarea").value = "";
+    return;
+  }
+
+  // Show loading message
+  const textarea = document.getElementById("bulk-actions-textarea");
+  textarea.value = "Loading...";
+  hideModalWarning();
+
+  // Extract folder paths
+  const folderPaths = Array.from(selectedCheckboxes).map((checkbox) =>
+    checkbox.getAttribute("data-path"),
+  );
+
+  try {
+    // Call backend API
+    const response = await fetch("/picker/api/guess-config-names", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({ folder_paths: folderPaths }),
+    });
+
+    const data = await response.json();
+
+    if (!data.success) {
+      showModalWarning(`Error: ${data.error}`);
+      textarea.value = "";
+      return;
+    }
+
+    // Process results
+    const successfulResults = data.results.filter(
+      (result) => result.config_name,
+    );
+    const failedResults = data.results.filter((result) => !result.config_name);
+
+    if (successfulResults.length === 0) {
+      showModalWarning("No config names could be guessed from the metadata.");
+      textarea.value = "";
+      return;
+    }
+
+    // Format output
+    const configNames = successfulResults.map((result) => result.config_name);
+    textarea.value = configNames.join("\n");
+
+    // Show warning if some folders failed
+    if (failedResults.length > 0) {
+      const errorMessages = failedResults.map((result) => {
+        const folderName = result.folder.split("/").pop();
+        return `${folderName}: ${result.error}`;
+      });
+      showModalWarning(
+        `Warning: ${failedResults.length} folder(s) failed:\n${errorMessages.join("\n")}`,
+      );
+    } else {
+      hideModalWarning();
+    }
+  } catch (error) {
+    console.error("Error guessing config names:", error);
+    showModalWarning(`Error: ${error.message}`);
+    textarea.value = "";
+  }
+}
+
+async function fillTextareaWithAWSSubmitCommands() {
+  const selectedCheckboxes = getVisibleCheckedCheckboxes();
+
+  if (selectedCheckboxes.length === 0) {
+    showModalWarning(
+      "No folders selected. Please select folders from the table first.",
+    );
+    document.getElementById("bulk-actions-textarea").value = "";
+    return;
+  }
+
+  // Show loading message
+  const textarea = document.getElementById("bulk-actions-textarea");
+  textarea.value = "Loading...";
+  hideModalWarning();
+
+  // Extract folder paths
+  const folderPaths = Array.from(selectedCheckboxes).map((checkbox) =>
+    checkbox.getAttribute("data-path"),
+  );
+
+  try {
+    // Call backend API
+    const response = await fetch("/picker/api/guess-config-names", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({ folder_paths: folderPaths }),
+    });
+
+    const data = await response.json();
+
+    if (!data.success) {
+      showModalWarning(`Error: ${data.error}`);
+      textarea.value = "";
+      return;
+    }
+
+    // Process results
+    const successfulResults = data.results.filter(
+      (result) => result.config_name,
+    );
+    const failedResults = data.results.filter((result) => !result.config_name);
+
+    if (successfulResults.length === 0) {
+      showModalWarning("No config names could be guessed from the metadata.");
+      textarea.value = "";
+      return;
+    }
+
+    // Format output as AWS submit commands
+    const commands = successfulResults.map(
+      (result) =>
+        `aws/run_job.py -- aws/docker_and_sync.sh python main.py configs/main/${result.config_name}`,
+    );
+    textarea.value = commands.join("\n");
+
+    // Show warning if some folders failed
+    if (failedResults.length > 0) {
+      const errorMessages = failedResults.map((result) => {
+        const folderName = result.folder.split("/").pop();
+        return `${folderName}: ${result.error}`;
+      });
+      showModalWarning(
+        `Warning: ${failedResults.length} folder(s) skipped:\n${errorMessages.join("\n")}`,
+      );
+    } else {
+      hideModalWarning();
+    }
+  } catch (error) {
+    console.error("Error generating AWS submit commands:", error);
+    showModalWarning(`Error: ${error.message}`);
+    textarea.value = "";
+  }
+}
+
 function copyFromModal() {
   const textarea = document.getElementById("bulk-actions-textarea");
   const text = textarea.value;
diff --git a/codeclash/viewer/templates/picker.html b/codeclash/viewer/templates/picker.html
@@ -173,6 +173,12 @@ <h2>Bulk Actions</h2>
                 <button class="modal-action-btn" onclick="fillTextareaWithAWSResubmitCommands()">
                     <i class="bi bi-arrow-repeat"></i> AWS resubmit
                 </button>
+                <button class="modal-action-btn" onclick="fillTextareaWithGuessedConfigNames()">
+                    <i class="bi bi-file-earmark-text"></i> Guess config file name
+                </button>
+                <button class="modal-action-btn" onclick="fillTextareaWithAWSSubmitCommands()">
+                    <i class="bi bi-cloud-upload"></i> Guess AWS submit cmd
+                </button>
             </div>
             <div id="modal-warning" class="modal-warning">
                 <i class="bi bi-exclamation-triangle"></i>
