Enhance server.py with CORS support, versioning, and health check endpoints

- Added CORS configuration to allow requests from specified domains.
- Introduced a version constant for the API to improve version management.
- Expanded trusted hosts for better network access.
- Implemented new root and health check endpoints for improved API usability.
- Updated status endpoint to return the API version from the constant instead of using importlib.metadata.

server.py

@@ -1,6 +1,5 @@
 # server.py
 from dataclasses import dataclass, asdict
-import importlib.metadata
 import secrets
 import logging
 import asyncio
@@ -9,17 +8,25 @@ from typing import Tuple
 
 from quart import Quart, websocket, request
 from quart_schema import QuartSchema, validate_request, validate_response
+from quart_cors import cors
 from uvicorn.middleware.proxy_headers import ProxyHeadersMiddleware
 
 from broker import SessionBroker, SessionDoesNotExist, ClientRequest, ClientResponse, ClientError
 
 # Configuración
+VERSION = "1.0.0"  # Versión de la API
 TIMEOUT: int = 40
 LOG_LEVEL: int = logging.DEBUG
-TRUSTED_HOSTS: list[str] = ["127.0.0.1"]
+TRUSTED_HOSTS: list[str] = ["127.0.0.1", "10.16.38.136", "10.16.3.13", "10.16.13.73"]
 
 # Create app
 app = Quart(__name__)
+app = cors(app, 
+    allow_origin=["https://*.hf.space", "https://*.huggingface.co"],
+    allow_methods=["GET", "POST", "OPTIONS"],
+    allow_headers=["Content-Type"],
+    max_age=3600
+)
 QuartSchema(app)
 app.asgi_app = ProxyHeadersMiddleware(app.asgi_app, trusted_hosts=TRUSTED_HOSTS)
 app.logger.setLevel(LOG_LEVEL)
@@ -74,7 +81,7 @@ class ErrorResponse:
 @app.get("/status")
 @validate_response(Status)
 async def status() -> Status:
-    return Status(status="OK", version=importlib.metadata.version('your-package-name'))
+    return Status(status="OK", version=VERSION)
 
 @app.websocket('/session')
 async def session_handler():
@@ -156,5 +163,17 @@ async def write(data: Write) -> Tuple[WriteResponse | ErrorResponse, int]:
     except asyncio.TimeoutError:
         return ErrorResponse('Timeout when waiting for client.'), 500
 
+# Agregar un endpoint de health check y root
+@app.route("/")
+async def root():
+    return {"message": "Kaio API Server", "version": VERSION}
+
+@app.route("/health")
+async def health_check():
+    return {"status": "healthy"}
+
 def run():
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860)
+
+if __name__ == "__main__":
+    run()