From 98c102d0ef1c5baa29565efcba012cf3e4e27240 Mon Sep 17 00:00:00 2001
From: matatonic <matatonic-git@zhero.org>
Date: Fri, 1 Dec 2023 14:33:13 -0500
Subject: [PATCH] simpler health check

---
 main.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/main.py b/main.py
index 38acfb7..b2ec4db 100755
--- a/main.py
+++ b/main.py
@@ -7,7 +7,7 @@ import subprocess
 import tempfile
 import yaml
 from fastapi import FastAPI
-from fastapi.responses import StreamingResponse, JSONResponse
+from fastapi.responses import StreamingResponse, Response
 from fastapi.middleware.cors import CORSMiddleware
 import uvicorn
 from pydantic import BaseModel
@@ -57,13 +57,11 @@ app.add_middleware(
     allow_headers=["*"]
 )
 
-@app.get("/livez")
-async def livez():
-    return "OK"
-
+@app.get("/")
+@app.head("/")
 @app.options("/")
-async def options():
-    return JSONResponse(content="OK")
+async def root() -> Response:
+    return Response(status_code=200, content="", media_type="text/plain")
 
 class GenerateSpeechRequest(BaseModel):
     model: str = "tts-1" # or "tts-1-hd"