88 lines
4.3 KiB
Python
88 lines
4.3 KiB
Python
import re
|
|
import json
|
|
import numpy as np
|
|
from typing import List
|
|
|
|
class sorter:
|
|
''' 重点穴位按摩排列器 '''
|
|
def __init__(self, body_part = 'back', massage_side = 'both'):
|
|
with open('config/acupoint_metadata.json','r',encoding='utf-8') as file:
|
|
metadata = json.load(file)
|
|
self.acupoints_metadata = metadata
|
|
self.body_part_list = ['back','shoulder','waist']
|
|
self.massage_side_list = ['left','right','both']
|
|
if body_part in self.body_part_list:
|
|
self.body_part = body_part # 默认部位为'back'
|
|
else:
|
|
raise ValueError("按摩位置不在可按摩区域")
|
|
if massage_side in self.massage_side_list:
|
|
self.massage_side = massage_side # 默认为双边'both'
|
|
else:
|
|
raise ValueError("未指定按摩在左侧、右侧或两侧")
|
|
def _extract_acupoints(self,respnse_from_llm:str)->List[str]:
|
|
pattern = r"[0-9]+\.\s*([\u4e00-\u9fa5]{2,5}穴)"
|
|
matches = re.findall(pattern, respnse_from_llm)
|
|
# 去重 & 排除空值
|
|
unique_names = sorted(set(name for name in matches if name.strip()))
|
|
print(unique_names)
|
|
return unique_names
|
|
|
|
def sort_acupoints(self,respnse_from_llm:List[str])->List[str]:
|
|
if self.body_part == 'back':
|
|
allowed_names = []
|
|
if self.body_part == 'shoulder':
|
|
allowed_names = ["肩中左俞","肩外左俞","秉风左","天宗左","曲垣左","附分左","大杼左","风门左",
|
|
"肩中右俞","肩外右俞","秉风右","天宗右","曲垣右","附分右","大杼右","风门右"]
|
|
if self.body_part == 'waist':
|
|
allowed_names = ["志室左","肓门左","胃仓左","意舍左","阳纲左","胞肓左","气海左俞",
|
|
"大肠左俞","小肠左俞","中膂左俞","肾俞左","关元左俞","膀胱左俞","白环左俞","秩边左","京门左",
|
|
"志室右","肓门右","胃仓右","意舍右","阳纲右","胞肓右","气海右俞","大肠右俞","小肠右俞",
|
|
"中膂右俞","肾俞右","关元右俞","膀胱右俞","白环右俞","秩边右","京门右"]
|
|
|
|
def __filter_acupoints(acupoints:List[str],allowed_names: List[str])->List[str]:
|
|
acupoints_cleaned = []
|
|
matched_keys = []
|
|
for acupoint in acupoints:
|
|
res = re.sub(r"[穴]$","",acupoint)
|
|
res = re.sub(r"[俞]$","",res)
|
|
acupoints_cleaned.append(res)
|
|
for name in acupoints_cleaned:
|
|
# 第一步：模糊匹配 metadata 中包含 name 的 key
|
|
keys_candidates = [k for k in self.acupoints_metadata.keys() if name in k]
|
|
|
|
# 第二步：如果不是背部，限制匹配在 allowed_names 范围内
|
|
if self.body_part != "back":
|
|
keys_candidates = [k for k in keys_candidates if k in allowed_names]
|
|
|
|
matched_keys.extend(keys_candidates)
|
|
return matched_keys
|
|
acupoints_filtered = __filter_acupoints(respnse_from_llm,allowed_names)
|
|
# 侧边过滤逻辑
|
|
if hasattr(self,"massage_side"):
|
|
if self.massage_side == 'left':
|
|
acupoints_filtered = [pt for pt in acupoints_filtered if "左" in pt]
|
|
elif self.massage_side == 'right':
|
|
acupoints_filtered = [pt for pt in acupoints_filtered if "右" in pt]
|
|
# 'both'或者无定义的情况下则不做筛选
|
|
|
|
def __get_coords(name:str):
|
|
return self.acupoints_metadata[name]["pos"]
|
|
|
|
left_group = [pt for pt in acupoints_filtered if __get_coords(pt)[0] < 4]
|
|
right_group = [pt for pt in acupoints_filtered if __get_coords(pt)[0] > 4]
|
|
|
|
left_group_sorted = sorted(left_group,key=lambda pt:(__get_coords(pt)[0],-__get_coords(pt)[1]))
|
|
right_group_sorted = sorted(right_group,key=lambda pt:(__get_coords(pt)[0],__get_coords(pt)[1]),reverse=True)
|
|
|
|
pt_sorted = left_group_sorted + right_group_sorted
|
|
|
|
return pt_sorted
|
|
|
|
|
|
if __name__ == "__main__":
|
|
mySorter = sorter()
|
|
print(type(mySorter.acupoints_metadata))
|
|
test_response = ["肩井穴","天宗穴","肺俞穴"]
|
|
pts_sorted = mySorter.sort_acupoints(test_response)
|
|
print(pts_sorted)
|