{header, footer}.html

1 files changed, 13 insertions, 1 deletions

diff --git a/app/main.py b/app/main.py
index ebc4297..56bfb99 100644
--- a/app/main.py
+++ b/app/main.py
@@ -6,6 +6,7 @@ from fastapi import FastAPI, Request, Depends
 from fastapi.responses import HTMLResponse
 from fastapi.staticfiles import StaticFiles
 from starlette.templating import Jinja2Templates
+from jinja2 import Environment, FileSystemLoader
 from pydantic import BaseModel
 
 from utils import get_avatar_urls, get_client_geo
@@ -22,9 +23,20 @@ async def index(
     request: Request,
     avatar_urls: Annotated[list[str], Depends(get_avatar_urls)],
 ):
+    
+    static_env = Environment(
+        loader=FileSystemLoader('./static'),
+        autoescape=True
+    )
+    
     return templates.TemplateResponse(
         "index.html",
-        {"request": request, "avatar_urls": avatar_urls},
+        {
+            "request": request,
+            "avatar_urls": avatar_urls,
+            "header": static_env.get_template('/header.html').render(),
+            "footer": static_env.get_template('/footer.html').render()
+        },
     )