Add better error handling for UI

routers/root.py

@@ -72,6 +72,25 @@ async def favicon():
     return FileResponse("static/favicon.ico")
 
 
+@app.post("/handleError/")
+async def handle_error(request: Request):
+    context = {"request": request, "settings": settings}
+
+    def not_found():
+        st.html(templates.get_template("errors/404.html").render(**context))
+
+    def unknown_error():
+        st.html(templates.get_template("errors/unknown.html").render(**context))
+
+    body = await request.json()
+
+    match body["status"]:
+        case 404:
+            return st.runner(lambda: page_wrapper(request, not_found))
+        case _:
+            return st.runner(lambda: page_wrapper(request, unknown_error))
+
+
 @app.get("/login/")
 def login(request: Request):
     if request.user and not request.user.is_anonymous:

templates/errors/404.html

@@ -0,0 +1,8 @@
+{% block content %}
+<div style="text-align: center; margin: 2rem 0;">
+    <h1>
+        Oops, we can't find this page!
+    </h1>
+    <img src="https://media.tenor.com/IHdlTRsmcS4AAAAC/404.gif" height="404" alt="404">
+</div>
+{% endblock content %}

templates/errors/unknown.html

@@ -0,0 +1,8 @@
+{% block content %}
+<div style="text-align: center; margin: 2rem 0;">
+    <h1>
+        Uh oh... Something went wrong
+    </h1>
+    <img src="https://media.tenor.com/1Wlr9trTPLwAAAAd/robotically-robot.gif" height="404" alt="404">
+</div>
+{% endblock %}