BlueskyPreviewServer.py

"""
This serves a directory of static files.
It also tells you what files are available for a run.
"""

from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
import os


directory = os.environ["THUMBNAIL_DIRECTORY"]
os.makedirs(directory, exist_ok=True)
app = FastAPI()


@app.get("/{uid}")
async def get_files(uid: str):
    run_directory = os.path.join(directory, uid)

    if os.path.exists(run_directory):
        return [f for f in os.listdir(run_directory)
                if os.path.isfile(os.path.join(run_directory, f))]
    else:
        return []


print("SERVING STATIC FILES:", directory)
app.mount('', StaticFiles(directory=directory))