Fix bug 9

app.py

@@ -46,6 +46,12 @@ logger = logging.getLogger(__name__)
 
 # Global variables
 app = FastAPI(title="Diamond CSGO AI Player")
+
+# Set safe defaults for headless CI/Spaces environments
+os.environ.setdefault("SDL_VIDEODRIVER", "dummy")
+os.environ.setdefault("SDL_AUDIODRIVER", "dummy")
+os.environ.setdefault("PYGAME_HIDE_SUPPORT_PROMPT", "1")
+os.environ.setdefault("OMP_NUM_THREADS", "1")
 connected_clients: Set[WebSocket] = set()
 
 class WebKeyMap:

src/csgo/action_processing.py

@@ -6,7 +6,10 @@ from dataclasses import dataclass
 from typing import Dict, List, Set, Tuple
 
 import numpy as np
-import pygame
+try:
+    import pygame  # type: ignore
+except Exception:
+    pygame = None  # type: ignore
 import torch
 
 from .keymap import CSGO_FORBIDDEN_COMBINATIONS, CSGO_KEYMAP
@@ -26,7 +29,18 @@ class CSGOAction:
 
     @property
     def key_names(self) -> List[str]:
-        return [pygame.key.name(key) for key in self.keys] 
+        # Use pygame to convert key codes when available; else assume keys already strings
+        names: List[str] = []
+        for key in self.keys:
+            if pygame is not None:
+                try:
+                    names.append(pygame.key.name(key))
+                    continue
+                except Exception:
+                    pass
+            # Fallback for headless: keep as-is or cast to string
+            names.append(str(key))
+        return names 
 
     def process_mouse(self) -> None:
         # Clip and match mouse to closest in list of possibles
@@ -208,7 +222,12 @@ def decode_csgo_action(y_preds: torch.Tensor) -> CSGOAction:
     id = np.argmax(mouse_y_pred)
     mouse_y = MOUSE_Y_POSSIBLES[id]
 
-    keys_pressed = [pygame.key.key_code(x) for x in keys_pressed]
+    # Map string names back to pygame key codes when pygame is available; otherwise keep as strings
+    if pygame is not None:
+        try:
+            keys_pressed = [pygame.key.key_code(x) for x in keys_pressed]
+        except Exception:
+            pass
 
     return CSGOAction(keys_pressed, mouse_x, mouse_y, bool(l_click), bool(r_click))
 

src/csgo/keymap.py

@@ -1,25 +1,49 @@
-import pygame
+try:
+    import pygame  # type: ignore
+except Exception:
+    pygame = None  # type: ignore
 
 
-CSGO_KEYMAP = {
-    pygame.K_w: "up",
-    pygame.K_d: "right",
-    pygame.K_a: "left",
-    pygame.K_s: "down",
-    pygame.K_SPACE: "jump",
-    pygame.K_LCTRL: "crouch",
-    pygame.K_LSHIFT: "walk",
-    pygame.K_1: "weapon1",
-    pygame.K_2: "weapon2",
-    pygame.K_3: "weapon3",
-    pygame.K_r: "reload",
+if pygame is not None:
+    CSGO_KEYMAP = {
+        pygame.K_w: "up",
+        pygame.K_d: "right",
+        pygame.K_a: "left",
+        pygame.K_s: "down",
+        pygame.K_SPACE: "jump",
+        pygame.K_LCTRL: "crouch",
+        pygame.K_LSHIFT: "walk",
+        pygame.K_1: "weapon1",
+        pygame.K_2: "weapon2",
+        pygame.K_3: "weapon3",
+        pygame.K_r: "reload",
 
-    # Override mouse movement with arrows
-    pygame.K_UP: "camera_up",
-    pygame.K_RIGHT: "camera_right",
-    pygame.K_LEFT: "camera_left",
-    pygame.K_DOWN: "camera_down",
-}
+        # Override mouse movement with arrows
+        pygame.K_UP: "camera_up",
+        pygame.K_RIGHT: "camera_right",
+        pygame.K_LEFT: "camera_left",
+        pygame.K_DOWN: "camera_down",
+    }
+else:
+    # Headless fallback without pygame
+    CSGO_KEYMAP = {
+        "w": "up",
+        "d": "right",
+        "a": "left",
+        "s": "down",
+        "space": "jump",
+        "left ctrl": "crouch",
+        "left shift": "walk",
+        "1": "weapon1",
+        "2": "weapon2",
+        "3": "weapon3",
+        "r": "reload",
+        # Override mouse movement with arrows (string identifiers)
+        "arrow_up": "camera_up",
+        "arrow_right": "camera_right",
+        "arrow_left": "camera_left",
+        "arrow_down": "camera_down",
+    }
 
 
 CSGO_FORBIDDEN_COMBINATIONS = [

src/game/play_env.py

@@ -3,7 +3,10 @@ import math
 from pathlib import Path
 from typing import Any, Dict, List, Tuple
 
-import pygame
+try:
+    import pygame  # type: ignore
+except Exception:  # Handle ImportError and headless environments gracefully
+    pygame = None  # type: ignore
 import torch
 from torch import Tensor
 
@@ -42,7 +45,13 @@ class PlayEnv:
         print("\nEnvironment actions:\n")
         for key, action_name in self.keymap.items():
             if key is not None:
-                key_name = pygame.key.name(key)
+                # Use pygame key names when available, otherwise fallback to string representation
+                key_name = str(key)
+                if pygame is not None:
+                    try:
+                        key_name = pygame.key.name(key)
+                    except Exception:
+                        pass
                 key_name = "⎵" if key_name == "space" else key_name
                 print(f"{key_name} : {action_name}")