'commit'

AI/WxMini/Input.txt

@@ -6,4 +6,4 @@
 6. 网络成瘾
 7. 创伤后应激反应
 8. 偷窃癖
-9. 不想活了，自杀倾向
+9. 自杀倾向

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))
 

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)
+                time.sleep(0.1)