add html pages to package data

vtuber-plan · Sep 6, 2024 · d1cdc94 · d1cdc94
1 parent e9777fa
commit d1cdc94
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/olah/server.py b/olah/server.py
@@ -161,8 +161,10 @@ async def lifespan(app: FastAPI):
 # ======================
 # Application
 # ======================
+code_file_path = os.path.abspath(__file__)
 app = FastAPI(lifespan=lifespan, debug=False)
-templates = Jinja2Templates(directory="static")
+templates = Jinja2Templates(directory=os.path.join(os.path.dirname(code_file_path), "..", "static"))
+
 
 class AppSettings(BaseSettings):
     # The address of the model controller.

diff --git a/pyproject.toml b/pyproject.toml
@@ -31,5 +31,8 @@ olah-cli = "olah.server:cli"
 [tool.setuptools.packages.find]
 exclude = ["assets*", "benchmark*", "docs", "dist*", "playground*", "scripts*", "tests*"]
 
+[tool.setuptools.package-data]
+static = ["*.html"]
+
 [tool.wheel]
 exclude = ["assets*", "benchmark*", "docs", "dist*", "playground*", "scripts*", "tests*"]