Como executar FastAPI / Uvicorn no Google Colab?

Question

Estou tentando executar um aplicativo da web "local" no Google Colab usando FastAPI / Uvicorn como alguns dos exemplos de código do aplicativo Flask que vi, mas não consigo fazê-lo funcionar. Alguém conseguiu fazer isso? Agradeço.

FastAPI e Uvicorn instalados com sucesso

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

Aplicativo de amostra

from fastapi import FastAPI

app = FastAPI()


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

Executar tentativas

#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

Você pode usar o ngrok para exportar uma porta como url externa. Basicamente, o ngrok pega algo disponível / hospedado em seu host local e o expõe à Internet com uma URL pública temporária.

Primeiro instale as dependências

!pip install fastapi nest-asyncio pyngrok uvicorn

Crie seu aplicativo

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'}

Em seguida, execute-o.

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)