Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
| from __future__ import annotations | |
| from fastapi.responses import HTMLResponse | |
| from openenv.core import create_fastapi_app | |
| from fusion_lab.models import StellaratorAction, StellaratorObservation | |
| from server.environment import BUDGET, N_FIELD_PERIODS, StellaratorEnvironment | |
| from server.physics import ASPECT_RATIO_MAX, AVERAGE_TRIANGULARITY_MAX, EDGE_IOTA_OVER_NFP_MIN | |
| app = create_fastapi_app( | |
| env=StellaratorEnvironment, | |
| action_cls=StellaratorAction, | |
| observation_cls=StellaratorObservation, | |
| ) | |
| def landing_page() -> str: | |
| return """<!DOCTYPE html> | |
| <html lang="en"> | |
| <head> | |
| <meta charset="utf-8"> | |
| <meta name="viewport" content="width=device-width, initial-scale=1"> | |
| <title>Fusion Design Lab</title> | |
| <style> | |
| * { margin: 0; padding: 0; box-sizing: border-box; } | |
| body { font-family: system-ui, -apple-system, sans-serif; background: #0a0e1a; color: #e0e0e0; min-height: 100vh; display: flex; align-items: center; justify-content: center; } | |
| .container { max-width: 720px; padding: 3rem 2rem; } | |
| h1 { font-size: 2.2rem; font-weight: 700; margin-bottom: 0.5rem; background: linear-gradient(135deg, #60a5fa, #a78bfa); -webkit-background-clip: text; -webkit-text-fill-color: transparent; } | |
| .subtitle { color: #94a3b8; font-size: 1.1rem; margin-bottom: 2rem; } | |
| .card { background: #141926; border: 1px solid #1e293b; border-radius: 12px; padding: 1.5rem; margin-bottom: 1.2rem; } | |
| .card h2 { font-size: 1rem; color: #94a3b8; text-transform: uppercase; letter-spacing: 0.05em; margin-bottom: 0.8rem; } | |
| .constraint { display: flex; justify-content: space-between; padding: 0.3rem 0; border-bottom: 1px solid #1e293b; } | |
| .constraint:last-child { border: none; } | |
| .constraint .name { color: #60a5fa; font-family: monospace; } | |
| .constraint .bound { color: #e0e0e0; font-family: monospace; } | |
| .endpoints { list-style: none; } | |
| .endpoints li { padding: 0.4rem 0; } | |
| .endpoints code { background: #1e293b; padding: 0.2rem 0.5rem; border-radius: 4px; font-size: 0.9rem; color: #a78bfa; } | |
| .endpoints .method { color: #4ade80; font-weight: 600; font-family: monospace; margin-right: 0.3rem; } | |
| .meta { display: flex; gap: 2rem; flex-wrap: wrap; } | |
| .meta-item { flex: 1; min-width: 140px; } | |
| .meta-item .value { font-size: 1.8rem; font-weight: 700; color: #60a5fa; } | |
| .meta-item .label { color: #64748b; font-size: 0.85rem; } | |
| a { color: #a78bfa; text-decoration: none; } | |
| a:hover { text-decoration: underline; } | |
| .footer { margin-top: 2rem; color: #475569; font-size: 0.85rem; text-align: center; } | |
| </style> | |
| </head> | |
| <body> | |
| <div class="container"> | |
| <h1>Fusion Design Lab</h1> | |
| <p class="subtitle">OpenEnv P1 stellarator optimization environment powered by <code>constellaration</code></p> | |
| <div class="card"> | |
| <h2>Task</h2> | |
| <p style="margin-bottom:1rem;">Minimize <strong>max elongation</strong> of a stellarator boundary using 4 geometric knobs, subject to physics constraints.</p> | |
| <div class="meta"> | |
| <div class="meta-item"><div class="value">6</div><div class="label">evaluation budget</div></div> | |
| <div class="meta-item"><div class="value">26</div><div class="label">discrete actions</div></div> | |
| <div class="meta-item"><div class="value">3</div><div class="label">field periods</div></div> | |
| </div> | |
| </div> | |
| <div class="card"> | |
| <h2>Constraints</h2> | |
| <div class="constraint"><span class="name">aspect_ratio</span><span class="bound">≤ 4.0</span></div> | |
| <div class="constraint"><span class="name">average_triangularity</span><span class="bound">≤ −0.5</span></div> | |
| <div class="constraint"><span class="name">abs(edge_iota_over_nfp)</span><span class="bound">≥ 0.3</span></div> | |
| </div> | |
| <div class="card"> | |
| <h2>API Endpoints</h2> | |
| <ul class="endpoints"> | |
| <li><span class="method">GET</span> <code>/health</code> — liveness check</li> | |
| <li><span class="method">GET</span> <code>/task</code> — task specification</li> | |
| <li><span class="method">POST</span> <code>/reset</code> — start a new episode</li> | |
| <li><span class="method">POST</span> <code>/step</code> — execute an action</li> | |
| </ul> | |
| </div> | |
| <div class="footer"> | |
| <a href="https://github.com/jungdaesuh/fusion-design-lab">GitHub</a> | |
| · OpenEnv Hackathon 2026 | |
| </div> | |
| </div> | |
| </body> | |
| </html>""" | |
| def task_summary() -> dict[str, object]: | |
| return { | |
| "description": ( | |
| "Optimize the P1 benchmark with a custom low-dimensional boundary family " | |
| "derived from a rotating-ellipse seed." | |
| ), | |
| "constraints": { | |
| "aspect_ratio_max": ASPECT_RATIO_MAX, | |
| "average_triangularity_max": AVERAGE_TRIANGULARITY_MAX, | |
| "abs_edge_iota_over_nfp_min": EDGE_IOTA_OVER_NFP_MIN, | |
| }, | |
| "n_field_periods": N_FIELD_PERIODS, | |
| "budget": BUDGET, | |
| "actions": ["run", "submit", "restore_best"], | |
| "parameters": [ | |
| "aspect_ratio", | |
| "elongation", | |
| "rotational_transform", | |
| "triangularity_scale", | |
| ], | |
| "directions": ["increase", "decrease"], | |
| "magnitudes": ["small", "medium", "large"], | |
| "evaluation_modes": { | |
| "run": "low-fidelity constellaration evaluation", | |
| "submit": "low-fidelity constellaration terminal evaluation", | |
| }, | |
| } | |
| def main() -> None: | |
| import uvicorn | |
| uvicorn.run("server.app:app", host="0.0.0.0", port=8000, reload=True) | |
| if __name__ == "__main__": | |
| main() | |