diff --git a/dsLightRag/Routes/RecognizeEduQuestion.py b/dsLightRag/Routes/RecognizeEduQuestion.py
new file mode 100644
index 00000000..38775db4
--- /dev/null
+++ b/dsLightRag/Routes/RecognizeEduQuestion.py
@@ -0,0 +1,70 @@
+import logging
+
+from fastapi import APIRouter, Request, HTTPException
+
+from Util.FormulaRecognizer import FormulaRecognizer
+from Util.ShiTiRecognizer import ShiTiRecognizer
+
+# 创建路由路由器
+router = APIRouter(prefix="/api", tags=["教育场景识别"])
+
+# 配置日志
+logger = logging.getLogger(__name__)
+
+@router.post("/recognize-formula")
+async def recognize_formula(request: Request):
+ """
+ 公式识别接口
+ 参数:
+ image_url: 图片URL
+ 返回:
+ JSON响应，包含公式识别结果
+ """
+ try:
+ # 获取请求参数
+ data = await request.json()
+ image_url = data.get("image_url")
+
+ # 验证参数
+ if not image_url:
+ raise HTTPException(status_code=400, detail="缺少必要参数: image_url")
+
+ # 调用公式识别器
+ recognizer = FormulaRecognizer()
+ result = recognizer.recognize_formula(image_url)
+
+ return result
+ except HTTPException as e:
+ raise e
+ except Exception as e:
+ logger.error(f"公式识别接口异常: {str(e)}")
+ raise HTTPException(status_code=500, detail=f"服务器内部错误: {str(e)}")
+
+@router.post("/recognize-question")
+async def recognize_question(request: Request):
+ """
+ 试题识别接口
+ 参数:
+ image_url: 图片URL
+ 返回:
+ JSON响应，包含试题识别结果
+ """
+ try:
+ # 获取请求参数
+ data = await request.json()
+ image_url = data.get("image_url")
+
+ # 验证参数
+ if not image_url:
+ raise HTTPException(status_code=400, detail="缺少必要参数: image_url")
+
+ # 调用试题识别器
+ recognizer = ShiTiRecognizer()
+ result = recognizer.recognize_question(image_url)
+
+ return result
+ except HTTPException as e:
+ raise e
+ except Exception as e:
+ logger.error(f"试题识别接口异常: {str(e)}")
+ raise HTTPException(status_code=500, detail=f"服务器内部错误: {str(e)}")
\ No newline at end of file
diff --git a/dsLightRag/Routes/__pycache__/RecognizeEduQuestion.cpython-310.pyc b/dsLightRag/Routes/__pycache__/RecognizeEduQuestion.cpython-310.pyc
new file mode 100644
index 00000000..3c253b5a
Binary files /dev/null and b/dsLightRag/Routes/__pycache__/RecognizeEduQuestion.cpython-310.pyc differ
diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py
index 183caef3..71532839 100644
--- a/dsLightRag/Start.py
+++ b/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 相关路由
# 登录相关（不用登录）
diff --git a/dsLightRag/Util/ShiTiRecognizer.py b/dsLightRag/Util/ShiTiRecognizer.py
index 22d70cdd..6c8e5ff0 100644
--- a/dsLightRag/Util/ShiTiRecognizer.py
+++ b/dsLightRag/Util/ShiTiRecognizer.py
@@ -88,6 +88,7 @@ class ShiTiRecognizer:
response = self.client.recognize_edu_question_ocr_with_options(request, runtime)
result = self._parse_response(response)
logger.info("试题识别成功")
+ print( result)
return result
except Exception as error:
logger.error(f"试题识别失败: {str(error)}")
diff --git a/dsLightRag/Util/__pycache__/FormulaRecognizer.cpython-310.pyc b/dsLightRag/Util/__pycache__/FormulaRecognizer.cpython-310.pyc
new file mode 100644
index 00000000..811a3421
Binary files /dev/null and b/dsLightRag/Util/__pycache__/FormulaRecognizer.cpython-310.pyc differ
diff --git a/dsLightRag/Util/__pycache__/ShiTiRecognizer.cpython-310.pyc b/dsLightRag/Util/__pycache__/ShiTiRecognizer.cpython-310.pyc
new file mode 100644
index 00000000..fea2efaf
Binary files /dev/null and b/dsLightRag/Util/__pycache__/ShiTiRecognizer.cpython-310.pyc differ
diff --git a/dsLightRag/static/RecognizeEduQuestionOcr.html b/dsLightRag/static/RecognizeEduQuestionOcr.html
new file mode 100644
index 00000000..6fab9f36
--- /dev/null
+++ b/dsLightRag/static/RecognizeEduQuestionOcr.html
@@ -0,0 +1,461 @@
+
+
+
+
+
+ 教育场景识别
+
+
+
+
+
+
+
+
+
+
教育场景识别
+
+
+
+
+
+
+
+
+
+
示例公式图片
+
+
+
+
数学公式示例
+
+
+
+
+
+
+
+
+
+
+
示例试题图片
+
+
+
+
试题示例
+
+
+
+
+
+
+