48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
from server import PromptServer
|
|
from .WebBuilder import WebBuilder
|
|
from .WebClient import WebClient
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class WebServer:
|
|
def __init__(self):
|
|
self.url_prefix = '/custom_nodes/presentation-ComfyUI'
|
|
self.webBuilder = WebBuilder()
|
|
self.webClient = WebClient()
|
|
|
|
def start(self):
|
|
logger.info("WebServer.start")
|
|
self.webBuilder.install_nuxt()
|
|
|
|
PromptServer.instance.app.router.add_post(
|
|
self.url_prefix + '/save',
|
|
self.webBuilder.web_save
|
|
)
|
|
|
|
PromptServer.instance.app.router.add_get(
|
|
self.url_prefix + '/restart',
|
|
self.webBuilder.web_restart
|
|
)
|
|
|
|
PromptServer.instance.app.router.add_get(
|
|
self.url_prefix + '/stop',
|
|
self.webBuilder.web_stop
|
|
)
|
|
|
|
PromptServer.instance.app.router.add_get(
|
|
self.url_prefix + '/rebuild',
|
|
self.webBuilder.web_rebuild
|
|
)
|
|
|
|
PromptServer.instance.app.router.add_get(
|
|
self.url_prefix + '/files',
|
|
self.webClient.web_files
|
|
)
|
|
|
|
PromptServer.instance.app.router.add_get(
|
|
self.url_prefix + '/history_move/{prompt_id}',
|
|
self.webClient.web_history_move
|
|
)
|