[ENH] Changed the favicon to neurobagel logo (#222)

* Changed the favicon to neurobagel logo

* Added docstrings

* Changed the titles

* Updated favicon_url

app/main.py

@@ -8,12 +8,16 @@
 import uvicorn
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import ORJSONResponse
+from fastapi.openapi.docs import get_redoc_html, get_swagger_ui_html
+from fastapi.responses import ORJSONResponse, RedirectResponse
 
 from .api import utility as util
 from .api.routers import attributes, query
 
-app = FastAPI(default_response_class=ORJSONResponse)
+app = FastAPI(
+    default_response_class=ORJSONResponse, docs_url=None, redoc_url=None
+)
+favicon_url = "https://raw.githubusercontent.com/neurobagel/documentation/main/docs/imgs/logo/neurobagel_favicon.png"
 
 app.add_middleware(
     CORSMiddleware,
@@ -24,6 +28,38 @@
 )
 
 
+@app.get("/favicon.ico", include_in_schema=False)
+async def favicon():
+    """
+    Overrides the default favicon with a custom one.
+    """
+    return RedirectResponse(url=favicon_url)
+
+
+@app.get("/docs", include_in_schema=False)
+def overridden_swagger():
+    """
+    Overrides the Swagger UI HTML for the "/docs" endpoint.
+    """
+    return get_swagger_ui_html(
+        openapi_url="/openapi.json",
+        title="Neurobagel API",
+        swagger_favicon_url=favicon_url,
+    )
+
+
+@app.get("/redoc", include_in_schema=False)
+def overridden_redoc():
+    """
+    Overrides the Redoc HTML for the "/redoc" endpoint.
+    """
+    return get_redoc_html(
+        openapi_url="/openapi.json",
+        title="Neurobagel API",
+        redoc_favicon_url=favicon_url,
+    )
+
+
 @app.on_event("startup")
 async def auth_check():
     """Checks whether username and password environment variables are set."""