Spaces:
Running
Running
File size: 1,861 Bytes
02eac4b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import asyncio
import logging
from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
handlers=[logging.StreamHandler(), logging.FileHandler("server.log")],
)
logger = logging.getLogger(__name__)
@asynccontextmanager
async def lifespan(app: FastAPI):
"""FastAPI lifespan manager (preserves original startup/shutdown)."""
# Startup
async def _startup():
# Wait 0.1s before starting
await asyncio.sleep(0.1)
print("Starting server...")
await _startup()
yield
# Shutdown
async def _shutdown():
# Wait 0.1s before shutting down
await asyncio.sleep(0.1)
print("Shutting down server...")
await _shutdown()
# FastAPI application (preserves original configuration)
app = FastAPI(
title="LeRobot Arena Server",
description="WebRTC video streaming server for robotics applications",
version="2.0.0",
lifespan=lifespan,
)
# CORS middleware (preserves original settings)
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
from .robotics.api import robotics_router
from .video.api import video_router
# Include the API routers
app.include_router(video_router)
app.include_router(robotics_router)
@app.get("/health")
async def health_check():
return {
"status": "healthy",
"server_running": True,
}
if __name__ == "__main__":
import uvicorn
logger.info("🤖 Starting LeRobot Arena Modular Server...")
print("🤖 Starting LeRobot Arena Modular Server...")
uvicorn.run(
"api:app",
reload=False,
log_level="info",
)
|