Update app/main.py

app/main.py

@@ -1,8 +1,10 @@
 import asyncio
 import time
+from pathlib import Path
 
 from fastapi import Depends, FastAPI, Header, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel, Field
 
 from . import config, llama_client
@@ -17,6 +19,11 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Serves test-ui/index.html at "/" so visiting the Space's URL shows the
+# benchmark page instead of a bare 404 - the API routes below are registered
+# first so this catch-all mount can't shadow them.
+_TEST_UI_DIR = Path(__file__).resolve().parent.parent / "test-ui"
+
 
 async def verify_api_key(x_api_key: str | None = Header(default=None)) -> None:
     if config.API_KEY and x_api_key != config.API_KEY:
@@ -142,6 +149,13 @@ async def translate_batch(req: BatchTranslateRequest):
     )
 
 
+# Registered last so it can't shadow the API routes above: Starlette matches
+# routes in registration order, and a "/" mount would otherwise catch
+# everything not matched yet.
+if _TEST_UI_DIR.is_dir():
+    app.mount("/", StaticFiles(directory=_TEST_UI_DIR, html=True), name="test-ui")
+
+
 @app.on_event("shutdown")
 async def shutdown():
     await llama_client.aclose()