modify:优化企业微信输出图片链接

coze_bot_api.py

@@ -9,6 +9,7 @@ Email: me@gentlecp.com
 Create Date: 2021/6/19
 -----------------End-----------------------------
 """
+import re
 import time
 
 from django.db.models.fields import return_None
@@ -132,10 +133,29 @@ def consumer(user_query,decrypt_data):
     coze_response = call_llm(prompt=user_query,bot_id=bot_id,coze_access_token = coze_access_token)
     # answer = coze_response['messages'][1]['content']#v2
     answer = coze_response['data'][1]['content'].replace(" ","") #v3 删除图片url中的空格
+
+    ##处理图片链接
+    image_counter = 1
+    # 定义一个替换函数，用于在替换时添加序号
+    def replace_with_counter(match):
+        nonlocal image_counter
+        alt_text = match.group(1) or f"示例图片{image_counter}"
+        url = match.group(2)
+        replacement = f'<a href="{url}">{alt_text}</a>'
+        image_counter += 1
+        return replacement
+    # 将Markdown格式的图片链接转换为HTML格式的文字链接，并添加序号
+    answer = re.sub(r'!\[(.*?)\]\((https?://[^)]+)\)', replace_with_counter, answer)
+
     print(f"结果:{answer}")
     # 主动发结果给qiwei
     qiwei_post(username, answer, agentid)
 
+# @app.get("/ok")
+# async def ok():
+#     return "ok"
+
+
 @app.get("/bot")
 async def verify(msg_signature: str, timestamp: str, nonce: str, echostr: str):
     ret, sEchoStr = wxcpt.VerifyURL(msg_signature, timestamp, nonce, echostr)
@@ -160,4 +180,6 @@ async def recv(msg_signature: str, timestamp: str, nonce: str, request: Request,
 
 
 if __name__ == "__main__":
+    # coze_response = call_llm(prompt="房快贷是什么",bot_id=bot_id,coze_access_token = coze_access_token)
+    # print(coze_response)
     uvicorn.run("coze_bot_api:app", port=port, host='0.0.0.0', reload=False,ssl_keyfile="./key.pem", ssl_certfile="./cert.pem")