¿Cómo ejecutar FastAPI / Uvicorn en Google Colab?

Question

Estoy intentando ejecutar una aplicación web "local" en Google Colab usando FastAPI / Uvicorn como algunos de los códigos de muestra de la aplicación Flask que he visto, pero no puedo hacer que funcione. ¿Fue alguien capaz de hacer esto? Lo aprecio.

FastAPI y Uvicorn instalados correctamente

!pip install FastAPI -q
!pip install uvicorn -q

Aplicación de muestra

from fastapi import FastAPI

app = FastAPI()


@app.get("/")
async def root():
    return {"message": "Hello World"}

Ejecutar intentos

#attempt 1
if __name__ == "__main__":
    uvicorn.run("/content/fastapi_002:app", host="127.0.0.1", port=5000, log_level="info")

#attempt 2
#uvicorn main:app --reload
!uvicorn "/content/fastapi_001.ipynb:app" --reload

Answer 1

Puede utilizar ngrok para exportar un puerto como una URL externa. Básicamente, ngrok toma algo disponible / alojado en su localhost y lo expone a Internet con una URL pública temporal.

Primero instale las dependencias

!pip install fastapi nest-asyncio pyngrok uvicorn

Crea tu aplicación

from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=['*'],
    allow_credentials=True,
    allow_methods=['*'],
    allow_headers=['*'],
)

@app.get('/')
async def root():
    return {'hello': 'world'}

Entonces bájala.

import nest_asyncio
from pyngrok import ngrok
import uvicorn

ngrok_tunnel = ngrok.connect(8000)
print('Public URL:', ngrok_tunnel.public_url)
nest_asyncio.apply()
uvicorn.run(app, port=8000)