Fix initial bugs

Dockerfile

@@ -24,10 +24,12 @@ RUN mkdir -p csgo/spawn config checkpoints cache && \
 
 # Set environment variables
 ENV PYTHONPATH=/app/src:/app
-ENV CUDA_VISIBLE_DEVICES=""
 ENV OMP_NUM_THREADS=2
 ENV MKL_NUM_THREADS=2
 
+# Don't set CUDA_VISIBLE_DEVICES="" - let HF Spaces GPU be detected
+# ENV CUDA_VISIBLE_DEVICES=""  # Removed - this disables GPU!
+
 # Set cache directories to writable locations
 ENV TORCH_HOME=/tmp/torch
 ENV HF_HOME=/tmp/huggingface

app.py

@@ -3,13 +3,30 @@ Web-based Diamond CSGO AI Player for Hugging Face Spaces
 Uses FastAPI + WebSocket for real-time keyboard input and game streaming
 """
 
+# Fix environment variables FIRST, before any other imports
+import os
+import tempfile
+
+# Fix OMP_NUM_THREADS immediately (before PyTorch/NumPy imports)
+if "OMP_NUM_THREADS" not in os.environ or not os.environ.get("OMP_NUM_THREADS", "").isdigit():
+    os.environ["OMP_NUM_THREADS"] = "2"
+
+# Set up cache directories immediately
+temp_dir = tempfile.gettempdir()
+os.environ.setdefault("TORCH_HOME", os.path.join(temp_dir, "torch"))
+os.environ.setdefault("HF_HOME", os.path.join(temp_dir, "huggingface"))
+os.environ.setdefault("TRANSFORMERS_CACHE", os.path.join(temp_dir, "transformers"))
+
+# Create cache directories
+for cache_var in ["TORCH_HOME", "HF_HOME", "TRANSFORMERS_CACHE"]:
+    cache_path = os.environ[cache_var]
+    os.makedirs(cache_path, exist_ok=True)
+
 import asyncio
 import base64
 import io
 import json
 import logging
-import os
-import tempfile
 from pathlib import Path
 from typing import Dict, List, Optional, Set
 
@@ -53,20 +70,7 @@ os.environ.setdefault("SDL_VIDEODRIVER", "dummy")
 os.environ.setdefault("SDL_AUDIODRIVER", "dummy")
 os.environ.setdefault("PYGAME_HIDE_SUPPORT_PROMPT", "1")
 
-# Fix OMP_NUM_THREADS for HF Spaces (must be positive integer)
-if "OMP_NUM_THREADS" not in os.environ or not os.environ["OMP_NUM_THREADS"].isdigit():
-    os.environ["OMP_NUM_THREADS"] = "2"
-
-# Set up proper cache directories for HF Spaces
-temp_dir = tempfile.gettempdir()
-os.environ.setdefault("TORCH_HOME", os.path.join(temp_dir, "torch"))
-os.environ.setdefault("HF_HOME", os.path.join(temp_dir, "huggingface"))
-os.environ.setdefault("TRANSFORMERS_CACHE", os.path.join(temp_dir, "transformers"))
-
-# Create cache directories
-for cache_var in ["TORCH_HOME", "HF_HOME", "TRANSFORMERS_CACHE"]:
-    cache_path = os.environ[cache_var]
-    os.makedirs(cache_path, exist_ok=True)
+# Environment variables already set at top of file
 connected_clients: Set[WebSocket] = set()
 
 class WebKeyMap:
@@ -212,18 +216,19 @@ class WebGameEngine:
             cfg.agent = OmegaConf.load(config_path / "agent" / "csgo.yaml")
             cfg.env = OmegaConf.load(config_path / "env" / "csgo.yaml") 
             
-            # Use CPU if no GPU available (for free HF spaces)
+            # Use GPU if available, otherwise fall back to CPU
             device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
             logger.info(f"Using device: {device}")
             
-            # Log GPU availability and CUDA info for debugging HF Spaces
+            # Log GPU availability and CUDA info for debugging
             if torch.cuda.is_available():
                 logger.info(f"CUDA available: {torch.cuda.is_available()}")
                 logger.info(f"GPU device count: {torch.cuda.device_count()}")
                 logger.info(f"Current GPU: {torch.cuda.get_device_name(0)}")
                 logger.info(f"GPU memory: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.1f} GB")
+                logger.info("🚀 GPU acceleration enabled!")
             else:
-                logger.info("CUDA not available, using CPU - this is normal for HF Spaces free tier")
+                logger.info("CUDA not available, using CPU mode")
             
             # Initialize agent first
             num_actions = cfg.env.num_actions