Spaces:
Sleeping
Sleeping
File size: 1,561 Bytes
f7d11f7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
from datetime import datetime
from functools import wraps
from fastapi import Request
import gradio as gr
from loguru import logger
def construct_response(f):
"""Construct a JSON response for an endpoint's results."""
@wraps(f)
def wrap(request: Request, *args, **kwargs):
result = f(request, *args, **kwargs)
response = {
"message": result["message"],
"method": request.method,
"status-code": result["status-code"],
"timestamp": datetime.now().isoformat(),
"url": request.url._url,
}
if "data" in result:
response["data"] = result["data"]
return response
return wrap
def get_model_from_state(request: Request):
"""Retrieve the model from the app state."""
model = getattr(request.app.state, "model", None)
if model is None:
logger.error("Model not loaded in app.state.model")
return model
def load_page(io, fn):
content = gr.Markdown("Loading...")
io.load(fn=fn, inputs=None, outputs=content)
return io
def update_patient_index_choices(df):
"""Populate the dropdown with valid patient indices from the batch results."""
import gradio as gr
if df is None:
return gr.update(choices=[], value=None)
try:
indices = list(df["Patients's index"].astype(int))
except Exception:
return gr.update(choices=[], value=None)
if not indices:
return gr.update(choices=[], value=None)
return gr.update(choices=indices, value=indices[0])
|