'commit'

dsLightRag/Start.py

@@ -1,33 +1,32 @@
-import uvicorn
 import asyncio
-from fastapi import FastAPI
-from starlette.staticfiles import StaticFiles
-from fastapi.middleware.cors import CORSMiddleware  # 添加此导入
+import logging  # 添加此导入
+from contextlib import asynccontextmanager
 
-from Routes.TeachingModel.tasks.BackgroundTasks import train_document_task
-from Util.PostgreSQLUtil import init_postgres_pool, close_postgres_pool
+import uvicorn
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware  # 添加此导入
+from starlette.staticfiles import StaticFiles
 
 from Routes.Ggb import router as ggb_router
-from Routes.Knowledge import router as knowledge_router
-from Routes.Oss import router as oss_router
-from Routes.Rag import router as rag_router
-from Routes.ZuoWen import router as zuowen_router
-from Routes.Llm import router as llm_router
-from Routes.TeachingModel.api.LoginController import router as login_router
-from Routes.TeachingModel.api.UserController import router as user_router
-from Routes.TeachingModel.api.DmController import router as dm_router
-from Routes.TeachingModel.api.ThemeController import router as theme_router
-from Routes.TeachingModel.api.DocumentController import router as document_router
-from Routes.TeachingModel.api.TeachingModelController import router as teaching_model_router
-from Routes.QA import router as qa_router
 from Routes.JiMengRoute import router as jimeng_router
-from Routes.SunoRoute import router as suno_router
-from Routes.XueBanRoute import router as xueban_router
+from Routes.Knowledge import router as knowledge_router
+from Routes.Llm import router as llm_router
 from Routes.MjRoute import router as mj_router
+from Routes.Oss import router as oss_router
+from Routes.QA import router as qa_router
 from Routes.QWenImageRoute import router as qwen_image_router
-from Util.LightRagUtil import *
-from contextlib import asynccontextmanager
-import logging  # 添加此导入
+from Routes.Rag import router as rag_router
+from Routes.SunoRoute import router as suno_router
+from Routes.TeachingModel.api.DmController import router as dm_router
+from Routes.TeachingModel.api.DocumentController import router as document_router
+from Routes.TeachingModel.api.LoginController import router as login_router
+from Routes.TeachingModel.api.TeachingModelController import router as teaching_model_router
+from Routes.TeachingModel.api.ThemeController import router as theme_router
+from Routes.TeachingModel.api.UserController import router as user_router
+from Routes.TeachingModel.tasks.BackgroundTasks import train_document_task
+from Routes.XueBanRoute import router as xueban_router
+from Routes.ZuoWen import router as zuowen_router
+from Routes.RecognizeEduQuestion import router as ocr_router
 
 # 控制日志输出
 logger = logging.getLogger('lightrag')
@@ -39,8 +38,8 @@ logger.addHandler(handler)
 
 @asynccontextmanager
 async def lifespan(_: FastAPI):
-    #pool = await init_postgres_pool()
-    #app.state.pool = pool
+    # pool = await init_postgres_pool()
+    # app.state.pool = pool
 
     asyncio.create_task(train_document_task())
 
@@ -48,7 +47,7 @@ async def lifespan(_: FastAPI):
         yield
     finally:
         # 应用关闭时销毁连接池
-        #await close_postgres_pool(pool)
+        # await close_postgres_pool(pool)
         pass
 
 
@@ -78,8 +77,8 @@ app.include_router(jimeng_router)  # 即梦路由
 app.include_router(suno_router)  # Suno路由
 app.include_router(xueban_router)  # 学伴路由
 app.include_router(mj_router)  # Midjourney路由
-app.include_router(qwen_image_router) # Qwen Image 路由
-
+app.include_router(qwen_image_router)  # Qwen Image 路由
+app.include_router(ocr_router)  # 教育场景识别
 
 # Teaching Model 相关路由
 # 登录相关（不用登录）