From 8e51af6786aa4132240d3b1d8b59aa1f7552e24a Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Mon, 13 Feb 2023 17:58:03 +0000
Subject: [PATCH] Changed to serve index.html on root

---
 main.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index 0c1b37c77..7de266a8e 100644
--- a/main.py
+++ b/main.py
@@ -409,6 +409,7 @@ class PromptServer():
         self.socket_handler = socket_handler
         self.number = 0
         self.app = web.Application()
+        self.web_root = os.path.join(os.path.dirname(os.path.realpath(__file__)), "webshit")
         routes = web.RouteTableDef()
 
         @routes.get('/ws')
@@ -417,7 +418,7 @@ class PromptServer():
         
         @routes.get("/")
         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")
         async def get_prompt(request):
@@ -494,7 +495,7 @@ class PromptServer():
 
         self.app.add_routes(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):