Spaces:
Running
Running
| # server/engine.py — Shared domain logic for the Supply Chain Simulator | |
| # Both main.py (HTTP API) and openenv_wrapper.py (RL API) import from here | |
| # to respect clean client/server separation. | |
| import random | |
| import uuid | |
| from typing import Any, List, Dict | |
| # ─── CONSTANTS ─────────────────────────────────────────────────────── | |
| INITIAL_FUNDS = 100000 | |
| SALE_PRICE = 150 | |
| STORAGE_COST = 10 | |
| CONTAINER_SIZE = 100 | |
| SHIPPING_COST = 3000 | |
| LATE_PENALTY_RATE = 0.20 | |
| # ─── DATA DEFINITIONS ─────────────────────────────────────────────── | |
| INITIAL_FACTORIES = [ | |
| {"id": "china", "name": "China (Shenzhen)", "baseCost": 50, "lat": 22.5, "lng": 114}, | |
| {"id": "india", "name": "India (Mumbai)", "baseCost": 45, "lat": 19, "lng": 72.8}, | |
| {"id": "mexico", "name": "Mexico (Monterrey)", "baseCost": 70, "lat": 25.6, "lng": -100.3}, | |
| {"id": "vietnam", "name": "Vietnam (Ho Chi Minh)", "baseCost": 42, "lat": 10.8, "lng": 106.6}, | |
| {"id": "brazil", "name": "Brazil (São Paulo)", "baseCost": 65, "lat": -23.5, "lng": -46.6}, | |
| {"id": "germany", "name": "Germany (Stuttgart)", "baseCost": 90, "lat": 48.8, "lng": 9.2}, | |
| ] | |
| INITIAL_PORTS = [ | |
| {"id": "shanghai", "name": "Port of Shanghai", "baseDelay": 1}, | |
| {"id": "singapore", "name": "Port of Singapore", "baseDelay": 1}, | |
| {"id": "rotterdam", "name": "Port of Rotterdam", "baseDelay": 2}, | |
| {"id": "la", "name": "Port of LA", "baseDelay": 1}, | |
| {"id": "busan", "name": "Port of Busan", "baseDelay": 1}, | |
| {"id": "dubai", "name": "Port of Dubai (Jebel Ali)", "baseDelay": 2}, | |
| {"id": "santos", "name": "Port of Santos", "baseDelay": 2}, | |
| ] | |
| WAREHOUSES = [ | |
| {"id": "usa", "name": "USA (Chicago)"}, | |
| {"id": "eu", "name": "EU (Frankfurt)"}, | |
| {"id": "asia", "name": "Asia (Tokyo)"}, | |
| {"id": "mideast", "name": "Middle East (Dubai)"}, | |
| {"id": "latam", "name": "Latin America (São Paulo)"}, | |
| {"id": "oceania", "name": "Oceania (Sydney)"}, | |
| ] | |
| INITIAL_DEMAND = {"usa": 100, "eu": 80, "asia": 120, "mideast": 60, "latam": 70, "oceania": 50} | |
| ALL_WAREHOUSE_IDS = [w["id"] for w in WAREHOUSES] | |
| # ─── GAME STATE ────────────────────────────────────────────────────── | |
| class GameState: | |
| turn: int = 1 | |
| scenario: str = "normal" | |
| balance: float = INITIAL_FUNDS | |
| benchmark_balance: float = INITIAL_FUNDS | |
| inventory: Dict[str, int] = {} | |
| demands: Dict[str, int] = {} | |
| factoryCosts: List[Dict[str, Any]] = [] | |
| portDelays: List[Dict[str, Any]] = [] | |
| inTransit: List[Dict[str, Any]] = [] | |
| history: List[Dict[str, Any]] = [] | |
| events: List[Dict[str, Any]] = [] | |
| alerts: List[Dict[str, Any]] = [] | |
| month_shipping_costs: float = 0.0 | |
| month_goods_costs: float = 0.0 | |
| last_revenue: float = 0.0 | |
| def __init__(self) -> None: | |
| self.reset() | |
| def reset(self): | |
| self.turn = 1 | |
| self.scenario = "normal" | |
| self.balance = INITIAL_FUNDS | |
| self.benchmark_balance = INITIAL_FUNDS | |
| self.inventory = {w["id"]: 0 for w in WAREHOUSES} | |
| self.demands = dict(INITIAL_DEMAND) | |
| self.factoryCosts = [{"id": f["id"], "name": f["name"], "baseCost": f["baseCost"], "currentCost": f["baseCost"]} for f in INITIAL_FACTORIES] | |
| self.portDelays = [{"id": p["id"], "name": p["name"], "baseDelay": p["baseDelay"], "currentDelay": p["baseDelay"], "isClosed": False} for p in INITIAL_PORTS] | |
| self.inTransit = [] | |
| self.history = [{"month": 0, "profit": 0, "revenue": 0, "storage_costs": 0, "shipping_costs": 0, "penalties": 0, "balance": INITIAL_FUNDS, "benchmark": INITIAL_FUNDS}] | |
| self.events = [] | |
| self.alerts = [] | |
| self.month_shipping_costs = 0 | |
| self.month_goods_costs = 0 | |
| self.last_revenue = 0 | |