Update index.html

index.html

@@ -37,20 +37,18 @@
                 this.canvas_ctx.font = '1px monospace';
                 this.canvas_ctx.fillText(text, 1, 1);
             };
-            this.submit_board = async function () {
-                document.body.style.cursor = 'progress';
-                const response = await fetch("submit", {
-                    method: 'POST',
-                    mode: 'cors', // no-cors, *cors, same-origin
-                    cache: 'no-cache', // *default, no-cache, reload, force-cache, only-if-cached
-                    credentials: 'same-origin', // include, *same-origin, omit
-                    headers: { 'Content-Type': 'application/json' },
-                    redirect: 'follow', // manual, *follow, error
-                    referrerPolicy: 'no-referrer', // no-referrer, *no-referrer-when-downgrade, origin, origin-when-cross-origin, same-origin, strict-origin, strict-origin-when-cross-origin, unsafe-url
-                    body: JSON.stringify({ "board": this_.board, "ai-player": this_.ai_player }) // body data type must match "Content-Type" header
-                });
-                return response.json(); // parses JSON response into native JavaScript objects
-            }
+            this.submit_board = async function (action_) {
+                const obs = tf.tensor(this_.board, [this_.num_rows, this_.num_cols], 'float32');
+                const normalized_obs = tf.mul(obs, this_.ai_player);
+                normalized_obs.print();
+                const [action_logits, value] = this_.agent.predict(normalized_obs);
+                action_logits.print();
+                const action = await tf.argMax(action_logits).array();
+                return {
+                    "terminated": false,
+                    "action": action,
+                };
+            };
             this.end_game = function () {
                 this.game_ended = true;
                 setTimeout(function() {this_.reset();}, 3000);
@@ -178,9 +176,17 @@
                 setTimeout(function () { this_.render(); });
             };
         };
+        const modelUrl = 'policy/model.json';
+        const init_fn = async function() {
+            const model = await tf.loadGraphModel(modelUrl);
+            await tf.setBackend('wasm');
+            return model;
+        };f
         document.addEventListener("DOMContentLoaded", function (event) {
-            game = new BoardGame(6, 7);
-            game.render();
+            init_fn().then(function (agent) {
+                game = new BoardGame(agent, 6, 7);
+                game.render();                
+            });
         });
     </script>
 </body>