move static into invokeai.frontend.web directory for dist install

This commit is contained in:
Lincoln Stein
2023-05-22 16:48:17 -04:00
parent 650d69ef5b
commit d4fb16825e
18 changed files with 899 additions and 823 deletions

View File

@ -4,6 +4,7 @@ from inspect import signature
import uvicorn
import invokeai.backend.util.logging as logger
import invokeai.frontend.web as web_dir
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.openapi.docs import get_redoc_html, get_swagger_ui_html
@ -11,6 +12,7 @@ from fastapi.openapi.utils import get_openapi
from fastapi.staticfiles import StaticFiles
from fastapi_events.handlers.local import local_handler
from fastapi_events.middleware import EventHandlerASGIMiddleware
from pathlib import Path
from pydantic.schema import schema
from .api.dependencies import ApiDependencies
@ -119,7 +121,7 @@ def custom_openapi():
app.openapi = custom_openapi
# Override API doc favicons
app.mount("/static", StaticFiles(directory="static/dream_web"), name="static")
app.mount("/static", StaticFiles(directory=Path(web_dir.__path__[0], 'static/dream_web')), name="static")
@app.get("/docs", include_in_schema=False)
def overridden_swagger():
@ -139,7 +141,7 @@ def overridden_redoc():
)
# Must mount *after* the other routes else it borks em
app.mount("/", StaticFiles(directory="invokeai/frontend/web/dist", html=True), name="ui")
app.mount("/", StaticFiles(directory=Path(web_dir.__path__[0],"dist"), html=True), name="ui")
def invoke_api():
# Start our own event loop for eventing usage