server/app.py

"""
app.py — Multi-task OpenEnv server with openenv-core bug fix
"""
import os, sys
sys.path.insert(0, os.path.dirname(__file__))

# ── Monkey-patch openenv-core bug (HTTP_422_UNPROCESSABLE_CONTENT missing) ──
import starlette.status as _status
if not hasattr(_status, "HTTP_422_UNPROCESSABLE_CONTENT"):
    _status.HTTP_422_UNPROCESSABLE_CONTENT = 422

from openenv.core import create_fastapi_app
from environment import PromptSensitivityEnvironment, SensitivityAction, SensitivityObservation
import uvicorn

TASK = os.environ.get("TASK", "1")

if TASK == "2":
    from environment2 import HallucinationDetectionEnvironment, HallucinationAction, HallucinationObservation
    app = create_fastapi_app(HallucinationDetectionEnvironment, HallucinationAction, HallucinationObservation)
else:
    app = create_fastapi_app(PromptSensitivityEnvironment, SensitivityAction, SensitivityObservation)

if __name__ == "__main__":
    uvicorn.run("app:app", host="0.0.0.0", port=7860, reload=False)