Adds human controls (Linux only) and simplify code

Makefile

@@ -1,8 +1,8 @@
 run:
-	diambra -r ~/.diambra/roms run python3 script.py
+	diambra -r ~/.diambra/roms run -l python3 script.py
 
 demo:
-	diambra -r ~/.diambra/roms run python3 mistral.py && python3 result.py
+	diambra -r ~/.diambra/roms run -l python3 mistral.py && python3 result.py
 
 install:
 	pip3 install -r requirements.txt

agent/config.py

@@ -139,6 +139,7 @@
         if "Punch" in move_name or "Kick" in move_name
     },
     "Jump Closer": {"right": [4, 4, 4, 4], "left": [2, 2, 2, 2]},
+    "Jump Away": {"right": [2, 2, 2, 2], "left": [4, 4, 4, 4]},
 }
 META_INSTRUCTIONS_WITH_LOWER = {
     **META_INSTRUCTIONS,

agent/language_models.py

@@ -1,7 +1,8 @@
 import os
-import dotenv
 from typing import Tuple
 
+import dotenv
+
 dotenv.load_dotenv()
 
 try:

agent/observer.py

@@ -1,7 +1,6 @@
 import numpy as np
 
 KEN_RED = [248, 0, 0]
-# KEN_GREEN = [48, 88, 152]
 KEN_GREEN = [88, 176, 40]
 
 
@@ -19,8 +18,7 @@ def detect_position_from_color(
     # the screen is a np.array of RGB colors (3 channels)
     # Select the frames where the characters play: between 80 vertical and 200 vertical
 
-    # dump the observation to a file
-
+    # dump the observation to a file for debugging
     if save_frame:
         np.save("observation.npy", frame)