diff --git a/AI/WxMini/Input.txt b/AI/WxMini/Input.txt index 152b47d5..0304421a 100644 --- a/AI/WxMini/Input.txt +++ b/AI/WxMini/Input.txt @@ -6,4 +6,4 @@ 6. 网络成瘾 7. 创伤后应激反应 8. 偷窃癖 -9. 不想活了,自杀倾向 \ No newline at end of file +9. 自杀倾向 \ No newline at end of file diff --git a/AI/WxMini/Start.py b/AI/WxMini/Start.py index 87242355..3733f2a4 100644 --- a/AI/WxMini/Start.py +++ b/AI/WxMini/Start.py @@ -518,10 +518,10 @@ async def process_image(image_url: str, current_user: dict = Depends(get_current try: if await is_text_dominant(client, image_url): logger.info("检测到图片主要是文字内容,开始识别文字:") - return StreamingResponse(recognize_text(client, image_url), media_type="text/plain") + return StreamingResponse(recognize_text(client,app.state.mysql_pool, image_url), media_type="text/plain") else: print("检测到图片主要是物体/场景,开始识别内容:") - return StreamingResponse(recognize_content(client, image_url), media_type="text/plain") + return StreamingResponse(recognize_content(client,app.state.mysql_pool, image_url), media_type="text/plain") except Exception as e: raise HTTPException(status_code=500, detail=str(e)) diff --git a/AI/WxMini/Utils/ImageUtil.py b/AI/WxMini/Utils/ImageUtil.py index 1675b2ad..f25e5e5f 100644 --- a/AI/WxMini/Utils/ImageUtil.py +++ b/AI/WxMini/Utils/ImageUtil.py @@ -1,7 +1,8 @@ import time import re -async def is_text_dominant(client,image_url): + +async def is_text_dominant(client, image_url): """ 判断图片是否主要是文字内容 :param image_url: 图片 URL @@ -34,7 +35,7 @@ async def is_text_dominant(client,image_url): return True -async def recognize_text(client,image_url): +async def recognize_text(client, pool, image_url): """ 识别图片中的文字,流式输出 """ @@ -64,7 +65,7 @@ async def recognize_text(client,image_url): time.sleep(0.1) -async def recognize_content(client,image_url): +async def recognize_content(client, pool, image_url): """ 识别图片中的内容,流式输出 """ @@ -80,4 +81,4 @@ async def recognize_content(client,image_url): if chunk.choices[0].delta.content is not None: for char in chunk.choices[0].delta.content: yield char - time.sleep(0.1) \ No newline at end of file + time.sleep(0.1)