sharktide commited on
Commit
b9aa185
·
verified ·
1 Parent(s): 54b842b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -6
app.py CHANGED
@@ -376,16 +376,40 @@ async def head_text():
376
 
377
  @app.get("/status")
378
  async def get_status():
379
- status = {
380
- "Video Generation": 500,
381
- "Image Generation": 200,
382
- "Lightning-Text v2": 200,
383
- "Music/SFX Generation": 200
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
384
  }
385
 
 
 
 
 
 
386
  return JSONResponse(
387
  status_code=207,
388
- content=status
 
 
 
389
  )
390
 
391
  @app.post("/gen/image")
 
376
 
377
  @app.get("/status")
378
  async def get_status():
379
+ services = {
380
+ "Video Generation": {
381
+ "code": 298,
382
+ "state": "degraded",
383
+ "message": "Experimental video generation services are degraded. If you recieve an error, retry after a few seconds"
384
+ },
385
+ "Image Generation": {
386
+ "code": 200,
387
+ "state": "ok",
388
+ "message": "Running normally"
389
+ },
390
+ "Lightning-Text v2": {
391
+ "code": 200,
392
+ "state": "ok",
393
+ "message": "Running normally"
394
+ },
395
+ "Music/SFX Generation": {
396
+ "code": 200,
397
+ "state": "ok",
398
+ "message": "Running normally"
399
+ }
400
  }
401
 
402
+ overall_state = (
403
+ "ok" if all(s["state"] == "ok" for s in services.values())
404
+ else "degraded"
405
+ )
406
+
407
  return JSONResponse(
408
  status_code=207,
409
+ content={
410
+ "state": overall_state,
411
+ "services": services
412
+ }
413
  )
414
 
415
  @app.post("/gen/image")