From 3c080a9ebfaf1927394981f555c65c72a4698bc3 Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Thu, 20 Feb 2025 04:04:54 +0800
Subject: [PATCH] Enhance webui mounting with root endpoint and directory
 check.

- Added FileResponse for webui root endpoint
- Enabled directory check in StaticFiles mount
- Improved webui static file handling
- Ensured webui directory existence
- Simplified webui access with root endpoint
---
 lightrag/api/lightrag_server.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
index f7f70c62..4083e790 100644
--- a/lightrag/api/lightrag_server.py
+++ b/lightrag/api/lightrag_server.py
@@ -7,6 +7,7 @@ from fastapi import (
     HTTPException,
     Depends,
 )
+from fastapi.responses import FileResponse
 import asyncio
 import threading
 import os
@@ -875,7 +876,11 @@ def create_app(args):
     # Webui mount webui/index.html
     static_dir = Path(__file__).parent / "webui"
     static_dir.mkdir(exist_ok=True)
-    app.mount("/webui", StaticFiles(directory=static_dir, html=True), name="webui")
+    app.mount("/webui", StaticFiles(directory=static_dir, html=True, check_dir=True), name="webui")
+    
+    @app.get("/webui/")
+    async def webui_root():
+        return FileResponse(static_dir / "index.html")
 
     return app