improve aimulator

Author: VictorTaelin

aimulator.mjs

@@ -1,12 +1,41 @@
-#!/usr/bin/env node
-import * as GPT from './GPT.mjs';
-import * as Claude from './Claude.mjs';
 import process from "process";
 import fs from 'fs/promises';
 import os from 'os';
 import path from 'path';
+import { Anthropic } from '@anthropic-ai/sdk';
+
+async function getAnthropicKey() {
+  const keyPath = path.join(process.env.HOME, '.config', 'anthropic.token');
+  return (await fs.readFile(keyPath, 'utf8')).trim();
+}
+
+export async function ask({ system, messages, max_tokens, model = 'claude-3-opus-20240229', temperature = 1, debug = true }) {
+  const anthropic = new Anthropic({ apiKey: await getAnthropicKey() });
+  if (debug) {
+    const stream = anthropic.messages.stream({
+      model,
+      messages,
+      max_tokens: max_tokens || 4096,
+      temperature,
+      ...(system && { system }),
+    }).on('text', (text) => process.stdout.write(text));
+    const message = await stream.finalMessage();
+    console.log(); // Add a newline at the end
+    return message.content[0].text;
+  } else {
+    const message = await anthropic.messages.create({
+      model,
+      messages,
+      max_tokens: max_tokens || 4096,
+      temperature,
+      ...(system && { system }),
+    });
+    return message.content[0].text;
+  }
+}
 
 const MODEL = "claude-3-opus-20240229";
+//const MODEL = "gpt-4-32k-0314";
 
 const SYSTEM = `
 You're a game emulator. You can emulate ANY game, but text-based. Your goal is
@@ -15,87 +44,192 @@ original as possible, from start to end.
 
 You'll be provided with:
 1. The chosen game.
-2. The current game log / history.
+2. The current message context.
 
-You'll must answer with:
-1. A description of the current game screen.
-2. A text-based UI of the current game screen.
+Your responses must include:
+1. A short description of the current game screen or state.
+2. A textual 2D UI of the current game screen, using emojis and symbols.
 3. A labelled list of options that the player can take.
 
-Note that the screen must emulate all game screen elements in a well-positioned,
-well-aligned 2D canvas. IT IS NOT ASCII ART. It is a textual UI. For example:
-
-# Example 1: Pokémon Red Battle Screen
-
-You're in a Pokémon battle.
-  
-  Blastoise LV30     💦🐢💣
-  HP: |||.......     🔫🐚🛡️
-
-  Charizard LV32     🔥🐉🦇
-  HP: ||||||....     🌋🦖😤
-
-Options:
-A) [FIGHT] B) [PKMN]
-C) [ITEM]  D) [RUN]
-
-Notes:
-1. The screen was drawn as compactly as possible.
-2. Key in-game screen elements were positioned in 2D.
-3. HP bars were drawn visually, to make it appealing.
-4. Emojis (NOT ASCII art) were used to represent images.
-5. We expanded the FIGHT option for faster interactions.
+Always follow this template:
 
-# Example 2: Zelda Majora's Mask - Odolwa Boss Fight Room
+<<description>>
+<<game_screen>>
+<<options>>
 
-HP   ❤️ ❤️ ❤️ 🤍🤍🤍🤍       :: [A] PutAway [B] 🗡
-MANA 🟩🟩🟩⬜⬜⬜⬜⬜⬜⬜ :: [<] 🪈 [V] 💣 [>] 🎣
+Guidelines for the game screen UI:
+- Draw it as compactly as possible while maintaining readability.
+- When handy, add a description / narration above the screen.
+- Use a 2D textual grid to position key game elements spatially.
+- Represent sprites, characters, items etc with 1-3 emojis each.
+- Draw HP/mana bars, item counts etc visually to form a nice UI.
+- Use ASCII diagrams very sparingly, mainly for progress bars.
+- Include menu options like Pause, Inventory etc for completeness.
+- Expand item/action options (e.g. Use X, Attack, Defend) for faster play.
 
- Link      Navi  Door-A
- [🗡️🧝🛡️]  [🧚]  [🚪🔒]
-                 
- Odolwa    Jar   Door-B  Chest
- [🗡️🎭🗡️]  [🏺]  [🚪🔒]  [🎁🔒]
+Here are some examples of how your game screen should look.
 
- Grss Grss Grss
- [🌿] [🌿] [🌿]
+# Example: Pokémon Red - Battle Screen
 
-💎 000 :: [_|7|_|_|_|_|_|_|_|_|_|_] ☀️  1st 
-
-Options:               
-A) Talk to Navi  B) Use Item
-C) Attack Odolwa D) Attack Jar
-E) Open Door-A   F) Open Door-B
-G) Move to Grass H) Press Start
-
-Notes:
-1. The screen was drawn as compactly as possible.
-2. The room layout was positioned in a 2D grid.
-3. Key room elements like Link, Odolwa, doors, etc were positioned spatially.
-4. HP/Mana bars and Rupee count were drawn visually.
-5. Emojis represent Link's current weapon, characters, items, etc.
-6. ASCII diagrams used for Rupee count bar and day cycle.
-7. Start button menu option included for completeness.
-8. Expanded item usage controls for faster interactions.
-
-IMPORTANT: You ARE the videogame. Stay in character. Answer ONLY with the
-game screen. Do NOT answer with assistant-like explanations.
-
-IMPORTANT: Stay LOYAL to the original game, including its core mechanics, order
-of events and gameplay, from the initial menu all the way to the end screen.
-
-At some points of the interaction, the player may add comments and hints after a
-hashtag ('#'). Use this feedback to adjust and improve the experience.`;
+You're in a Pokémon battle.
+,-----------------------------,
+   Blastoise LV30    [💦🐢💣]  
+   HP: |||.......    [🔫🐚🛡️]  
+                               
+   Charizard LV32    [🔥🐉🦇]  
+   HP: ||||||....    [🌋🦖😤]  
+'-----------------------------'
+A) FIGHT
+B) PKMN
+C) ITEM
+D) RUN
+
+# Example: Zelda Majora's Mask - Odolwa Boss Room
+
+You're in Odolwa's boss room in Woodfall Temple.
+Odolwa is dancing and swinging his swords menacingly.
+,--------------------------------------------------,
+  HP   ❤️ ❤️ ❤️ 🤍🤍🤍🤍                               
+  MANA 🟩🟩🟩⬜⬜⬜⬜⬜⬜⬜                         
+                                                    
+   Link      Navi  Door0                            
+   [🗡️🧝🛡️]  [🧚]  [🚪🔒]                           
+                                                    
+   Odolwa    Jar   Door1   Chest                    
+   [🗡️🎭🗡️]  [🏺]  [🚪🔒]  [🎁🔒]                   
+                                                    
+   Grs0 Grs1 Grs2                                   
+   [🌿] [🌿] [🌿]                                   
+                                                    
+  💎 000                     🕒 7 AM :: ☀️  1st Day  
+'--------------------------------------------------'
+A) Talk to Navi
+B) Enter Door0
+C) Attack Odolwa
+D) Break the Jar
+E) Enter Door1
+F) Check Grs0
+G) Check Grs1
+H) Check Grs2
+
+# Example: Mario 64 - Inside Castle
+
+You're in the main entrance hall of Princess Peach's castle.
+,---------------------------------.
+  🍄x4                       🌟x7  
+                                   
+    Door0       Door1      Door2   
+    [🚪🌟]      [🚪🔒]     [🚪0]   
+                                   
+  Door3   Door4    Door5   Door6   
+  [🚪0]   [🚪3]    [🚪7]   [🚪1]   
+                                   
+    Exit    Mario   Coin0  Coin1   
+    [🚪]    [🍄]     [🟡]   [🟡]   
+'---------------------------------'
+A) Enter Door0
+B) Enter Door1
+C) Enter Door2
+D) Enter Door3
+E) Enter Door4
+F) Enter Door5
+G) Enter Door6
+H) Check Coin0
+I) Check Coin1
+J) Exit
+
+# Example: Pokémon Red - Title Screen
+
+,-------------------------------,
+             Pokémon             
+              Red                
+                                 
+            [🔥🐉🦇]             
+                                 
+         ©1996 Nintendo          
+          Creatures Inc.         
+         GAME FREAK inc.         
+                                 
+       Press Start Button        
+'-------------------------------'
+A) New Game
+B) Continue
+C) Options
+
+# Example: Pokémon Red - Introduction
+
+,-------------------------------.
+                                 
+            OAK                  
+  Hello there! Welcome to the    
+  world of POKÉMON!              
+                                 
+            OAK                  
+  My name is OAK!                
+  People call me the             
+  POKÉMON PROF!                  
+                                 
+           NIDORAN♂              
+           [🐭💜🦏]                 
+'-------------------------------'
+A) Next
+
+# Example: Pokémon Red - Pallet Town
+
+You're in Pallet Town, your hometown.
+,--------------------------,
+      🌳 [Route 1] 🌳       
+                            
+   House0        House1     
+   [🏠]          [🏠]       
+                            
+   Grass         Oak's Lab  
+   [🌿]          [🏫]       
+                            
+   Beach         Sign   🌸  
+   [🌊]          [🪧]   🌼  
+'--------------------------'
+A) Enter House0
+B) Enter House1
+C) Enter Oak's Lab
+D) Check the Sign
+E) Walk in the Grass
+F) Exit to Route 1
+
+# Example: Pokémon Red - Protagonist's House
+
+You're inside your house in Pallet Town.
+,---------------------------.
+  PC        TV      Stairs   
+  [💻]      [📺]     [┗┓]    
+                             
+  Bed       You              
+  [🛏️]      [👦]             
+'---------------------------'
+A) Check the PC
+B) Play SNES on TV
+C) Rest in Bed
+B) Go Downstairs
+
+IMPORTANT: 
+- You ARE the videogame. Stay in character. 
+- Start from the game's initial menus and emulate each level in order.
+- Emulate the game loyally, following its original sequence of events.
+- Design a well-aligned UI for each screen. Position elements in 2D.
+- Respond with ONLY the next emulation step and its options.
+
+If the player provides feedback after a '#', use it to improve the experience.
+`;
 
 (async () => {
   console.clear();
 
   const ASCII_ART = `
-\x1b[1m\x1b[36m█▀▀▀▀▀█ ▀ ▄▀▄ █▀▀▀▀▀█\x1b[0m
+\x1b[1m\x1b[36m█▀▀▀▀▀█ ▀ ▄▀▄ █▀▀▀▀▀█\x1b[0m  
 \x1b[1m\x1b[36m█ ███ █ ▀ ▀█▀ █ ███ █\x1b[0m
 \x1b[1m\x1b[36m█ ▀▀▀ █ █ ▄█▄ █ ▀▀▀ █\x1b[0m
 \x1b[1m\x1b[36m▀▀▀▀▀▀▀ ▀ ▀▀▀ ▀▀▀▀▀▀▀\x1b[0m
-\x1b[2mA I   E M U L A T O R\x1b[0m
+\x1b[2mA I   E M U L A T O R\x1b[0m  
 `.trim();
 
   console.log(ASCII_ART);
@@ -104,31 +238,33 @@ hashtag ('#'). Use this feedback to adjust and improve the experience.`;
   console.log(`\x1b[32mUsing \x1b[1m${MODEL}\x1b[0m`);
   console.log("");
 
-  // TODO: get game input
   process.stdout.write("Game: ");
   const game = (await new Promise(resolve => process.stdin.once('data', data => resolve(data.toString())))).trim();
 
   console.log(`Emulating ${game}...\n\n`);
 
-  let log = '';
+  let messages = [
+    {role: "user", content: `# GAME: ${game}`},
+  ];
 
   while (true) {
     console.clear();
 
-    const response = await Claude.ask({
-      system: SYSTEM, 
-      model: MODEL,
-      prompt: `# GAME: ${game}\n# LOG:\n${log}\n\n# TASK: You must continue the game from here. Write your answer below, including the next screen's description, textual UI and player options:`,
-      max_tokens: 4096,
-      temperature: 0.9,
+    const response = await ask({
+      model: MODEL, 
+      messages,
+      max_tokens: 2048,
+      temperature: 0.5,
+      system: SYSTEM,
     });
 
-    log += `# SCREEN:\n\n${response}\n\n`;
+    messages.push({role: "assistant", content: response});
 
     process.stdout.write("\n\nEnter your choice: ");
-    const choice = (await new Promise(resolve => process.stdin.once('data', data => resolve(data.toString())))).trim().toUpperCase();
-    log += `# ACTION: ${choice}\n\n`;
+    const choice = (await new Promise(resolve => process.stdin.once('data', data => resolve(data.toString())))).trim();
+    messages.push({role: "user", content: choice});
 
-    await fs.writeFile(path.join(os.homedir(), '.log.txt'), log);
+    await fs.writeFile("./log.txt", messages.map(m => `${m.role === "user" ? "# PLAYER" : "# EMULATOR"}:\n\n${m.content}\n\n`).join(""));
   }
 })();
+