Upload folder using huggingface_hub

Dockerfile

@@ -1,5 +1,8 @@
-ARG BASE_IMAGE=openenv-base:latest
-FROM ${BASE_IMAGE}
+# Use public Python base image for HuggingFace compatibility
+FROM python:3.11-slim
+
+# Set working directory
+WORKDIR /app/env
 
 # Install system dependencies for Playwright and browsers
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -27,29 +30,22 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     # Additional dependencies
     git \
     wget \
+    curl \
     && rm -rf /var/lib/apt/lists/*
 
-# Install BrowserGym and related packages
-COPY src/envs/browsergym_env/server/requirements.txt /tmp/browsergym_requirements.txt
-RUN pip install --no-cache-dir -r /tmp/browsergym_requirements.txt && \
-    rm /tmp/browsergym_requirements.txt
+# Copy environment files first (for better caching)
+COPY . .
+
+# Install Python dependencies using pip install -e . (from pyproject.toml)
+RUN pip install --no-cache-dir -e .
 
 # Install Playwright browsers (Chromium by default)
 RUN playwright install chromium
 
 # Install MiniWoB++ tasks
-RUN pip install browsergym-miniwob
 RUN git clone --depth 1 https://github.com/Farama-Foundation/miniwob-plusplus.git /app/miniwob-plusplus
 
-# Copy OpenEnv core and browsergym_env code
-WORKDIR /app
-COPY src/core/ /app/src/core/
-COPY src/envs/browsergym_env/ /app/src/envs/browsergym_env/
-COPY src/envs/browsergym_env/README.md /app/README.md
-RUN chmod +x /app/src/envs/browsergym_env/server/start.sh
-
 # Set environment variables
-ENV PYTHONPATH=/app/src
 ENV PYTHONUNBUFFERED=1
 ENV BROWSERGYM_BENCHMARK=miniwob
 ENV BROWSERGYM_TASK_NAME="click-test"
@@ -57,9 +53,11 @@ ENV BROWSERGYM_HEADLESS=true
 ENV BROWSERGYM_VIEWPORT_WIDTH=1280
 ENV BROWSERGYM_VIEWPORT_HEIGHT=720
 ENV BROWSERGYM_TIMEOUT=10000
+ENV BROWSERGYM_PORT=8000
 ENV MINIWOB_HTML_DIR=/app/miniwob-plusplus/miniwob/html
 ENV MINIWOB_HTTP_PORT=8888
 ENV MINIWOB_URL=http://127.0.0.1:8888/miniwob/
+ENV ENABLE_WEB_INTERFACE=true
 
 # For WebArena tasks, these should be set by the user when running the container:
 # ENV SHOPPING=
@@ -70,11 +68,13 @@ ENV MINIWOB_URL=http://127.0.0.1:8888/miniwob/
 # ENV WIKIPEDIA=
 # ENV HOMEPAGE=
 
+# Expose ports
 EXPOSE 8000
 EXPOSE 8888
 
+# Health check
 HEALTHCHECK --interval=30s --timeout=3s --start-period=10s --retries=3 \
     CMD curl -f http://localhost:8000/health || exit 1
 
-ENV ENABLE_WEB_INTERFACE=true
-CMD ["/app/src/envs/browsergym_env/server/start.sh"]
+# Run the server using the start script
+CMD ["/app/env/server/start.sh"]

client.py

@@ -2,8 +2,8 @@
 
 from typing import Any, Dict
 
-from core.http_env_client import HTTPEnvClient, StepResult
-from envs.browsergym_env.models import (
+from openenv_core.http_env_client import HTTPEnvClient, StepResult
+from browsergym_env.models import (
     BrowserGymAction,
     BrowserGymObservation,
     BrowserGymState,

models.py

@@ -8,7 +8,7 @@ and more under a single Gymnasium-compatible API.
 from dataclasses import dataclass
 from typing import List, Optional
 
-from core.env_server.types import Action, Observation, State
+from openenv_core.env_server.types import Action, Observation, State
 
 
 @dataclass(kw_only=True)

server/app.py

@@ -2,12 +2,9 @@
 
 import os
 
-from core.env_server.http_server import create_app
-from envs.browsergym_env.models import (
-    BrowserGymAction,
-    BrowserGymObservation,
-)
-from envs.browsergym_env.server.browsergym_environment import BrowserGymEnvironment
+from openenv_core.env_server.http_server import create_app
+from browsergym_env.models import BrowserGymAction, BrowserGymObservation
+from browsergym_env.server.browsergym_environment import BrowserGymEnvironment
 
 # Get configuration from environment variables
 benchmark = os.environ.get("BROWSERGYM_BENCHMARK", "miniwob")

server/browsergym_environment.py

@@ -15,8 +15,8 @@ from uuid import uuid4
 
 import gymnasium as gym
 
-from core.env_server.interfaces import Environment
-from envs.browsergym_env.models import (
+from openenv_core.env_server.interfaces import Environment
+from browsergym_env.models import (
     BrowserGymAction,
     BrowserGymObservation,
     BrowserGymState,

server/start.sh

@@ -25,5 +25,5 @@ cleanup() {
 
 trap cleanup EXIT INT TERM
 
-exec python -m uvicorn envs.browsergym_env.server.app:app --host 0.0.0.0 --port "${BROWSERGYM_PORT}"
+exec python -m uvicorn browsergym_env.server.app:app --host 0.0.0.0 --port "${BROWSERGYM_PORT}"