Changed to serve index.html on root

This commit is contained in:
pythongosssss 2023-02-13 17:58:03 +00:00 committed by GitHub
parent 540bbbdb82
commit 8e51af6786
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -409,6 +409,7 @@ class PromptServer():
self.socket_handler = socket_handler self.socket_handler = socket_handler
self.number = 0 self.number = 0
self.app = web.Application() self.app = web.Application()
self.web_root = os.path.join(os.path.dirname(os.path.realpath(__file__)), "webshit")
routes = web.RouteTableDef() routes = web.RouteTableDef()
@routes.get('/ws') @routes.get('/ws')
@ -417,7 +418,7 @@ class PromptServer():
@routes.get("/") @routes.get("/")
async def get_root(request): async def get_root(request):
return aiohttp.web.HTTPFound('/index.html') return web.FileResponse(os.path.join(self.web_root, "index.html"))
@routes.get("/prompt") @routes.get("/prompt")
async def get_prompt(request): async def get_prompt(request):
@ -494,7 +495,7 @@ class PromptServer():
self.app.add_routes(routes) self.app.add_routes(routes)
self.app.add_routes([ self.app.add_routes([
web.static('/', os.path.join(os.path.dirname(os.path.realpath(__file__)), "webshit")), web.static('/', self.web_root),
]) ])
async def start_server(server, address, port): async def start_server(server, address, port):