Spaces:

AEUPH
/

AI-NPC-3D-BABYLON

Running

App Files Files Community

AEUPH commited on Aug 14, 2024

Commit

4f23333

verified ·

1 Parent(s): d4050fa

Update index.html

Browse files

Files changed (1) hide show

index.html +117 -18

index.html CHANGED Viewed

@@ -3,27 +3,126 @@
 <head>
     <meta charset="UTF-8" />
-    <link rel="stylesheet" href="style.css" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-    <title>Transformers.js - Object Detection</title>
 </head>
 <body>
-    <h1>Object Detection w/ 🤗 Transformers.js</h1>
-    <label id="container" for="upload">
-        <svg width="25" height="25" viewBox="0 0 25 25" fill="none" xmlns="http://www.w3.org/2000/svg">
-            <path fill="#000"
-                d="M3.5 24.3a3 3 0 0 1-1.9-.8c-.5-.5-.8-1.2-.8-1.9V2.9c0-.7.3-1.3.8-1.9.6-.5 1.2-.7 2-.7h18.6c.7 0 1.3.2 1.9.7.5.6.7 1.2.7 2v18.6c0 .7-.2 1.4-.7 1.9a3 3 0 0 1-2 .8H3.6Zm0-2.7h18.7V2.9H3.5v18.7Zm2.7-2.7h13.3c.3 0 .5 0 .6-.3v-.7l-3.7-5a.6.6 0 0 0-.6-.2c-.2 0-.4 0-.5.3l-3.5 4.6-2.4-3.3a.6.6 0 0 0-.6-.3c-.2 0-.4.1-.5.3l-2.7 3.6c-.1.2-.2.4 0 .7.1.2.3.3.6.3Z">
-            </path>
-        </svg>
-        Click to upload image
-        <label id="example">(or try example)</label>
-    </label>
-    <label id="status">Loading model...</label>
-    <input id="upload" type="file" accept="image/*" />
-    <script src="index.js" type="module"></script>
 </body>
-</html>

 <head>
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Babylon.js 3D VR Game with Transformers.js Chat</title>
+    <style>
+        body, html { margin: 0; padding: 0; overflow: hidden; font-family: 'Trebuchet MS', serif; }
+        canvas { width: 100%; height: 100%; display: block; }
+        #chatUI { position: absolute; top: 10px; right: 10px; width: 300px; }
+        #chatLog { width: 100%; height: 400px; background-color: rgba(0, 0, 0, 0.7); padding: 10px; color: #fff; overflow-y: auto; border: 2px solid #4e342e; box-shadow: 2px 2px 10px rgba(0, 0, 0, 0.9); }
+        #chatInput { width: 100%; height: 40px; margin-top: 10px; background-color: rgba(255, 255, 255, 0.4); border: 2px solid #4e342e; }
+        #sendButton { width: 100%; height: 40px; margin-top: 10px; background: linear-gradient(to bottom, #957d5f, #6c543e); color: #e0d7c5; border: none; cursor: pointer; }
+    </style>
 </head>
 <body>
+    <canvas id="renderCanvas"></canvas>
+    <div id="chatUI">
+        <div id="chatLog"></div>
+        <input type="text" id="chatInput" placeholder="Type your message here..." />
+        <button id="sendButton">Send</button>
+    </div>
+    <script src="https://cdn.babylonjs.com/babylon.js"></script>
+    <script src="https://cdn.babylonjs.com/loaders/babylon.glTF2FileLoader.min.js"></script>
+    <script src="https://unpkg.com/@huggingface/transformers"></script>
+    <script>
+        // Setup Babylon.js
+        const canvas = document.getElementById("renderCanvas");
+        const engine = new BABYLON.Engine(canvas, true);
+        const createScene = function() {
+            const scene = new BABYLON.Scene(engine);
+            // Camera
+            const camera = new BABYLON.ArcRotateCamera("Camera", Math.PI / 2, Math.PI / 2, 10, BABYLON.Vector3.Zero(), scene);
+            camera.attachControl(canvas, true);
+            // Light
+            const light = new BABYLON.HemisphericLight("light", new BABYLON.Vector3(1, 1, 0), scene);
+            // Ground
+            const ground = BABYLON.MeshBuilder.CreateGround("ground", { width: 10, height: 10 }, scene);
+            // Player and NPCs
+            const player = BABYLON.MeshBuilder.CreateBox("player", { size: 1 }, scene);
+            player.position.y = 0.5;
+            const npc1 = BABYLON.MeshBuilder.CreateSphere("npc1", { diameter: 1 }, scene);
+            npc1.position.x = 3;
+            npc1.position.y = 0.5;
+            // Basic movement (Q-learning is complex in a static file, so using basic movement for now)
+            window.addEventListener("keydown", (event) => {
+                switch (event.key) {
+                    case "ArrowUp":
+                        player.position.z -= 0.1;
+                        break;
+                    case "ArrowDown":
+                        player.position.z += 0.1;
+                        break;
+                    case "ArrowLeft":
+                        player.position.x -= 0.1;
+                        break;
+                    case "ArrowRight":
+                        player.position.x += 0.1;
+                        break;
+                }
+                checkCollision(player, npc1);
+            });
+            function checkCollision(mesh1, mesh2) {
+                if (mesh1.intersectsMesh(mesh2)) {
+                    openChat("npc1", "NPC-1", "Hello! How can I assist you today?");
+                }
+            }
+            return scene;
+        };
+        const scene = createScene();
+        engine.runRenderLoop(function() {
+            scene.render();
+        });
+        window.addEventListener("resize", function() {
+            engine.resize();
+        });
+        // Chat System with Transformers.js
+        async function setupChat() {
+            const model = await transformers.AutoModelForCausalLM.from_pretrained('mistralai/Mistral');
+            const tokenizer = await transformers.AutoTokenizer.from_pretrained('mistralai/Mistral');
+            return { model, tokenizer };
+        }
+        let chatSetup = setupChat();
+        document.getElementById("sendButton").addEventListener("click", async () => {
+            const inputField = document.getElementById("chatInput");
+            const chatLog = document.getElementById("chatLog");
+            const message = inputField.value;
+            if (message.trim() === "") return;
+            chatLog.innerHTML += `<p>Player: ${message}</p>`;
+            inputField.value = "";
+            const { model, tokenizer } = await chatSetup;
+            const inputs = tokenizer(message, { return_tensors: "pt" });
+            const response = await model.generate(inputs.input_ids, { max_length: 50 });
+            const output = tokenizer.decode(response[0], { skip_special_tokens: true });
+            chatLog.innerHTML += `<p>NPC: ${output}</p>`;
+            chatLog.scrollTop = chatLog.scrollHeight;
+        });
+        function openChat(npcId, npcName, greeting) {
+            const chatLog = document.getElementById("chatLog");
+            chatLog.innerHTML += `<p>${npcName}: ${greeting}</p>`;
+        }
+    </script>
 </body>
+</html>