Upload folder using huggingface_hub

src/envs/pokemon_env/README.md

@@ -50,7 +50,7 @@ print(f"Reward: {result.reward}, Done: {result.done}")
 ```bash
 # Build both images (run from project root directory)
 docker build -t pokemon-showdown:latest -f src/envs/pokemon_env/server/Dockerfile.showdown .
-docker build -t pokemon-env:latest -f src/envs/pokemon_env/server/Dockerfile.env .
+docker build -t pokemon-env:latest -f src/envs/pokemon_env/server/Dockerfile.pokemonenv .
 
 # Create Docker network for container communication
 docker network create pokemon-network
@@ -68,7 +68,7 @@ curl http://localhost:9980/health  # Test OpenEnv server
 ## Configuration
 
 Environment variables:
-- `POKEMON_BATTLE_FORMAT` - Battle format (default: `gen9randombattle`)
+- `POKEMON_BATTLE_FORMAT` - Battle format (default: `gen8randombattle`)
 - `POKEMON_REWARD_MODE` - Reward mode: `sparse` or `dense` (default: `sparse`)
 - `POKEMON_MAX_TURNS` - Maximum turns per battle (default: `1000`)
 - `POKEMON_PLAYER_USERNAME` - Player username (default: auto-generated)

src/envs/pokemon_env/server/Dockerfile.pokemonenv

@@ -0,0 +1,61 @@
+# Dockerfile for Pokemon Battle Environment OpenEnv
+# This image provides Pokemon battles via poke-env
+
+# Build OpenEnv base (can be overridden for CI/CD)
+ARG BASE_IMAGE
+FROM ${BASE_IMAGE:-openenv-base:latest} AS final
+
+# Install dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    supervisor \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install poke-env and dependencies
+RUN pip install --no-cache-dir \
+    poke-env>=0.9.0 \
+    gymnasium>=0.29.0
+
+# Copy OpenEnv core (base image already set WORKDIR=/app)
+COPY src/core/ /app/src/core/
+
+# Copy Pokemon environment code
+COPY src/envs/pokemon_env/ /app/src/envs/pokemon_env/
+
+# Copy README for web interface documentation
+COPY src/envs/pokemon_env/README.md /app/README.md
+
+# Pokemon environment variables
+ENV SHOWDOWN_SERVER_URL=pokemon-showdown:8000
+ENV POKEMON_BATTLE_FORMAT=gen8randombattle
+ENV POKEMON_PLAYER_USERNAME=player
+ENV POKEMON_REWARD_MODE=sparse
+ENV POKEMON_MAX_TURNS=1000
+
+# Expose OpenEnv port
+EXPOSE 9980
+
+# Create supervisor config for OpenEnv
+RUN echo '[supervisord]\n\
+nodaemon=true\n\
+logfile=/dev/null\n\
+logfile_maxbytes=0\n\
+\n\
+[program:openenv]\n\
+command=uvicorn envs.pokemon_env.server.app:app --host 0.0.0.0 --port 9980\n\
+directory=/app\n\
+environment=PYTHONPATH="/app/src"\n\
+autostart=true\n\
+autorestart=true\n\
+stdout_logfile=/dev/fd/1\n\
+stdout_logfile_maxbytes=0\n\
+stderr_logfile=/dev/fd/2\n\
+stderr_logfile_maxbytes=0\n\
+startsecs=10\n' > /etc/supervisor/conf.d/pokemon-env.conf
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=3s --start-period=15s --retries=3 \
+    CMD curl -f http://localhost:9980/health || exit 1
+
+# Run supervisor
+CMD ["/usr/bin/supervisord", "-c", "/etc/supervisor/supervisord.conf"]

src/envs/pokemon_env/server/pokemon_environment.py

@@ -15,6 +15,7 @@ Key Design:
 import asyncio
 import logging
 import uuid
+import os
 from dataclasses import asdict
 from threading import Event, Lock
 from typing import Any, Dict, List, Optional
@@ -25,7 +26,7 @@ from ..models import PokemonAction, PokemonObservation, PokemonData, PokemonStat
 
 try:
     # Import from top-level poke_env module
-    from poke_env import Player, RandomPlayer, AccountConfiguration, LocalhostServerConfiguration
+    from poke_env import Player, RandomPlayer, AccountConfiguration, ServerConfiguration
     # Import battle orders from player submodule
     from poke_env.player import BattleOrder, ForfeitBattleOrder
     # Import concurrency from concurrency submodule
@@ -247,13 +248,17 @@ class PokemonEnvironment(Environment):
         self.player_username = player_username or f"player_{uuid.uuid4().hex[:8]}"
         self.reward_mode = reward_mode
         self.max_turns = max_turns
+        self.showdown_server_url = os.getenv("SHOWDOWN_SERVER_URL", "localhost:8000")
 
         # Initialize player on POKE_LOOP
         logger.info(f"Creating player {self.player_username} for format {battle_format}")
 
         self.player = OpenEnvPokemonPlayer(
             account_configuration=AccountConfiguration(self.player_username, None),
-            server_configuration=LocalhostServerConfiguration,
+            server_configuration=ServerConfiguration(
+                f"ws://{self.showdown_server_url}/showdown/websocket",
+                "https://play.pokemonshowdown.com/action.php?"
+            ),
             battle_format=self.battle_format,
             max_concurrent_battles=1,  # One battle at a time
         )
@@ -264,7 +269,10 @@ class PokemonEnvironment(Environment):
             logger.info(f"Creating random opponent {opponent_username}")
             self.opponent = RandomPlayer(
                 account_configuration=AccountConfiguration(opponent_username, None),
-                server_configuration=LocalhostServerConfiguration,
+                server_configuration=ServerConfiguration(
+                    f"ws://{self.showdown_server_url}/showdown/websocket",
+                    "https://play.pokemonshowdown.com/action.php?"
+                ),
                 battle_format=self.battle_format,
                 max_concurrent_battles=1,
             )
@@ -275,7 +283,7 @@ class PokemonEnvironment(Environment):
         self._state = PokemonState(
             battle_format=battle_format,
             player_username=self.player_username,
-            server_url="localhost:8000",
+            server_url=self.showdown_server_url,
         )
 
         # Battle tracking