190 lines
8.6 KiB
Python
190 lines
8.6 KiB
Python
|
|
from sympy import symbols, Not, Or, And, to_dnf
|
|
from sympy import symbols, simplify_logic
|
|
|
|
import re
|
|
|
|
split_characters = r'[()&|~ ]' # 用正则表达式定义多个字符，包括逗号、句号、分号和感叹号等
|
|
|
|
predicate_list = {"RobotNear","On","Holding","Exists","IsClean","Active","Closed","Low"}
|
|
object_list = {'Coffee', 'Water', 'Dessert', 'Softdrink', 'BottledDrink', 'Yogurt', 'ADMilk', 'MilkDrink', 'Milk','VacuumCup','Chips', 'NFCJuice', 'Bernachon', 'ADMilk', 'SpringWater',
|
|
'Apple','Banana','Mangosteen','Orange','Glass','OrangeJuice','Tray','CoconutMilk','Kettle','PaperCup','Bread','Cake','LunchBox','Teacup','Tissue','Chocolate','Sandwiches','Mugs','Ice',
|
|
'Bar', 'Bar2', 'WaterStation', 'CoffeeStation', 'Table1', 'Table2', 'Table3','WindowTable4','WindowTable5','WindowTable6','QuietTable1','QuietTable2','ReadingNook','Entrance','Exit','LoungeArea','HighSeats','VIPLounge','MerchZone',
|
|
'Table1','Floor','Chairs','AC','TubeLight','HallLight','Curtain','ACTemperature'
|
|
}
|
|
|
|
dic_pred_obj={}
|
|
dic_pred_obj['RobotNear']=['Coffee', 'Water', 'Dessert', 'Softdrink', 'BottledDrink', 'Yogurt', 'ADMilk', 'MilkDrink', 'Milk', 'VacuumCup','Chips', 'NFCJuice', 'Bernachon', 'ADMilk', 'SpringWater', 'Apple', 'Banana', 'Mangosteen', 'Orange','Kettle', 'PaperCup', 'Bread', 'LunchBox','Teacup', 'Chocolate', 'Sandwiches', 'Mugs','Watermelon', 'Tomato', 'CleansingFoam','CocountMilk','SugarlessGum', 'MedicalAdhensiveTape', 'SourMilkDrink', 'PaperCup','Tissue', 'YogurtDrink', 'Newspaper', 'Box','PaperCupStarbucks', 'CoffeeMachine', 'Straw', 'Cake','Tray', 'Bread','Glass', 'Door','Mug', 'Machine','PackagedCoffee', 'CubeSugar','Apple', 'Spoon','Drinks', 'Drink','Ice', 'Saucer','TrashBin', 'Knife','Cube']
|
|
dic_pred_obj['RobotNear']+=['Bar', 'Bar2', 'WaterStation', 'CoffeeStation', 'Table1', 'Table2', 'Table3', 'WindowTable6','WindowTable4', 'WindowTable5','QuietTable1', 'QuietTable2', 'QuietTable3', 'ReadingNook', 'Entrance', 'Exit', 'LoungeArea', 'HighSeats','VIPLounge', 'MerchZone']
|
|
dic_pred_obj['IsClean']=['Table1','Floor','Chairs']
|
|
dic_pred_obj['Active']=['AC','TubeLight','HallLight']
|
|
dic_pred_obj['Closed']=['Curtain']
|
|
dic_pred_obj['Low']=['ACTemperature']
|
|
|
|
# object_list = {'Coffee', 'Water', 'Dessert', 'Softdrink', 'BottledDrink', 'Yogurt', 'ADMilk',
|
|
# 'MilkDrink', 'Milk','VacuumCup','Chips', 'NFCJuice', 'Bernachon', 'ADMilk', 'SpringWater',
|
|
# 'Bar', 'Bar2', 'WaterTable', 'CoffeeTable', 'Table1', 'Table2', 'Table3','BrightTable6',
|
|
# 'Table1','Floor','Chairs',
|
|
# 'AC','TubeLight','HallLight',
|
|
# 'Curtain','ACTemperature'
|
|
# }
|
|
def format_check(result):
|
|
try:
|
|
goal_dnf = str(to_dnf(result, simplify=True))
|
|
except Exception as e:
|
|
# print("Caught an error:", e)
|
|
return False, [str(e),None,None,None]
|
|
|
|
split_sentences = re.split(split_characters, result)
|
|
split_sentences = [s.strip() for s in split_sentences if s.strip()]
|
|
|
|
wrong_format_set = set()
|
|
wrong_predicate_set = set()
|
|
wrong_object_set = set()
|
|
|
|
for sentence in split_sentences:
|
|
if sentence == "": continue
|
|
|
|
try:
|
|
goal_dnf = str(to_dnf(sentence, simplify=True))
|
|
# 格式正确
|
|
word_list = re.split('_|~', sentence)
|
|
word_list = [word for word in word_list if word]
|
|
# word_list = sentence.split('_|~')
|
|
if len(word_list) <=1:
|
|
wrong_format_set.add(sentence)
|
|
continue
|
|
|
|
predicate = word_list[0]
|
|
if predicate not in predicate_list:
|
|
wrong_predicate_set.add(predicate)
|
|
|
|
for object in word_list[1:]:
|
|
if object not in object_list:
|
|
wrong_object_set.add(object)
|
|
if predicate in dic_pred_obj \
|
|
and predicate not in wrong_predicate_set and object not in wrong_object_set \
|
|
and object not in dic_pred_obj[predicate]:
|
|
wrong_format_set.add(sentence)
|
|
|
|
|
|
except:
|
|
wrong_format_set.add(sentence)
|
|
|
|
# print(wrong_format_set)
|
|
# print(wrong_predicate_set)
|
|
# print(wrong_object_set)
|
|
if len(wrong_format_set) == 0 and \
|
|
len(wrong_predicate_set) == 0 and\
|
|
len(wrong_object_set) == 0:
|
|
return True, None
|
|
else:
|
|
return False, [None,wrong_format_set,wrong_predicate_set,wrong_object_set]
|
|
|
|
def word_correct(sentence):
|
|
try:
|
|
goal_dnf = str(to_dnf(sentence, simplify=True))
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
|
|
# 检查 easy_instr_goal 的 goal 中的语法、谓词、obj
|
|
|
|
|
|
# data_set_file = "easy_instr_goal.txt"
|
|
# data_set_file = "medium_instr_goal.txt"
|
|
# data_set_file = "../dataset/hard_instr_goal.txt"
|
|
|
|
# with open(data_set_file, 'r', encoding="utf-8") as f:
|
|
# for i,line in enumerate(f):
|
|
# clean_line = line.strip()
|
|
# if 'Goal' in clean_line:
|
|
# answer = line.replace("Goal:","")
|
|
# format_correct,error_list = format_check(answer)
|
|
# if not format_correct:
|
|
# print(i," ",clean_line," ",error_list)
|
|
|
|
|
|
|
|
# # answer = '~On_Water_Bar & (On_Coffee_Table2 | On_Bernachon_Table2)'
|
|
# answer = 'On_Coffee_Table2 & ~(RobotNear_Table1 | RobotNear_Table3)'
|
|
# split_sentences = re.split(r'[()&|~ ]', answer)
|
|
# split_sentences = [s.strip() for s in split_sentences if s.strip()]
|
|
# print("split_sentences:",split_sentences)
|
|
# goal_dnf = str(to_dnf(answer, simplify=True))
|
|
# # print(goal_dnf)
|
|
# format_correct,error_list = format_check(answer)
|
|
# print(format_correct)
|
|
# print(error_list)
|
|
|
|
def goal_transfer_ls_set(goal):
|
|
goal_dnf = str(to_dnf(goal, simplify=True))
|
|
# print(goal_dnf)
|
|
goal_set =[]
|
|
goal_ls = goal_dnf.split("|")
|
|
for g in goal_ls:
|
|
g_set = set()
|
|
g = g.replace(" ", "").replace("(", "").replace(")", "")
|
|
g = g.split("&")
|
|
for literal in g:
|
|
if '_' in literal:
|
|
first_part, rest = literal.split('_', 1)
|
|
literal = first_part + '(' + rest
|
|
# 添加 ')' 到末尾
|
|
literal += ')'
|
|
# 替换剩余的 '_' 为 ','
|
|
literal = literal.replace('_', ',')
|
|
literal=literal.replace('~', 'Not ')
|
|
g_set.add(literal)
|
|
goal_set.append(g_set)
|
|
goal_set = [sorted(set(item)) for item in goal_set]
|
|
return goal_set
|
|
|
|
|
|
|
|
|
|
def get_feedback_prompt(prompt,result,error_list):
|
|
# wrong_format_set,wrong_predicate_set,wrong_object_set = error_list
|
|
|
|
error_message=""
|
|
|
|
if error_list[0]!=None:
|
|
error_message += "Syntax or Format Error. "+ error_list[0] + ".The answer should consist only of ~, |, &, and the given [Condition] and [Object].. "
|
|
else:
|
|
if error_list[1]!=set():
|
|
error_strings = ", ".join(error_list[1])
|
|
error_message += f"\"{error_strings}\" have format errors. They should consist only of ~, |, &, and the given [Condition] and [Object].\n"
|
|
if error_list[2]!=set():
|
|
error_strings = ", ".join(error_list[2])
|
|
error_message += f"\"{error_strings}\" are not in [Condition]. Please select the appropriate predicates from the [Condition] table to form the answer.\n"
|
|
if error_list[3]!=set():
|
|
error_strings = ", ".join(error_list[3])
|
|
error_message += f"\"{error_strings}\" are not in [Object]. Please select the appropriate parameters from the [Object] table to form the answer.\n"
|
|
|
|
# print("********** error_message: ",error_message,"\n********************")
|
|
|
|
prompt += "\n"+"The last answer is "+result +".\n" + error_message
|
|
return prompt
|
|
|
|
|
|
|
|
# answer = 'On_BottledDrink_Table3 | On_MilkDrink_Table3 |On_Softdrink_Bar' #'~On_Water_Bar & (On_Coffee_Table2 | On_Bernachon_Table2)'
|
|
# goal_dnf = str(to_dnf(answer, simplify=True))
|
|
# print(goal_dnf)
|
|
# print(goal_transfer_str(goal_dnf))
|
|
|
|
# answer = '~Holding_Chips & Is_Chairs_Clean'
|
|
# answer = 'On_Fries_Table6 & On_Yogurt_Table6 | (Active_HallLight & ~Low_ACTemperature) | (~Active_Curtain & Low_ACTemperature) | (Exists_Coffee & ~Holding_ADMilk)'
|
|
# answer="On_Fries_Table6 & On_Yogurt_Table6 | On_Coffee_Table6 & ~Holding_ADMilk | ~Low_ACTemperature | ~Closed_Curtain"
|
|
# answer="On_Fries_Table6 & (On_Yogurt_Table6 | On_Coffee_Table6) & ( OpenCurtain | ~Active_ACTemperature)"
|
|
|
|
|
|
# answer="On_Fries_Table6 & (On_Juice_Table6 | ~Exist_Juice => On_Coffee_Table6) & (OpenCurtain | ~Low_AC)"
|
|
# answer="On_Fries_Table6 & (On_Juice_Table6 | On_Coffee_Table6) & (Open_Curtain | ~Low_ACTemperature)"
|
|
#
|
|
answer="(On_Juice_Table6 | ~Exist_Juice=>On_Coffee_Table6 ) & ( ~Low_AC | Open_Curtain )"
|
|
format_correct,error_list = format_check(answer)
|
|
print(format_correct)
|
|
print(error_list)
|
|
print(get_feedback_prompt("",answer,error_list)) |