s459312/Generic_DialogSystem
1
0
Fork 0
forked from s151636/Generic_DialogSystem
Code Pull Requests Activity

system rules changes

Browse Source
This commit is contained in:
s459312 2023-06-15 21:01:59 +02:00
parent 01046d677f
commit 256fd00b7b
1 changed files with 31 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
system3.py
View File

@ -14,9 +14,9 @@ class DialogManager:
promotion = "Mamy dzisiaj w promocji ser!" promotion = "Mamy dzisiaj w promocji ser!"
elif r == 2: elif r == 2:
promotion = "Aktualnie w promocji mamy jabłka!" promotion = "Aktualnie w promocji mamy jabłka!"
elif r == 2: elif r == 3:
promotion = "Mleko w super cenie! Tylko dzisiaj!" promotion = "Mleko w super cenie! Tylko dzisiaj!"
elif r == 2: elif r == 4:
promotion = "Chipsy na imprezę w promocji!" promotion = "Chipsy na imprezę w promocji!"
return promotion return promotion
@ -70,6 +70,20 @@ class DialogManager:
response = "Dziękuję i do zobaczenia!" response = "Dziękuję i do zobaczenia!"
print("Agent:", response) print("Agent:", response)
break break
elif "thankyou" in acts[0].act_type:
r = random.randint(1, 4)
if r == 1:
response = "Cieszę się, że mogę pomóc."
elif r == 2:
response = "Nie ma za problemu."
elif r == 3:
response = "Przyjemność po mojej stronie."
elif r == 4:
response = "Czy mógłbym pomóc w czymś jeszcze?"
elif "affirm" in acts[0].act_type:
response = "Czy mógłbym pomóc w czymś jeszcze?"
elif "deny" in acts[0].act_type:
response = "Czy mógłbym pomóc w czymś jeszcze?"
else: else:
response = "Nie rozumiem. Czym mogę Ci pomóc?" response = "Nie rozumiem. Czym mogę Ci pomóc?"
@ -137,19 +151,20 @@ class DialogStateTracker:
product_type_rules = { product_type_rules = {
"pieczywo": ["chleb", "bułka", "bulka", "bułki", "bulki", "rogaliki", "rogalika", "bagietka", "bagietkę", "bagietke", "bagietki"], "pieczywo": ["chleb", "bułka", "bulka", "bulki", "rogalik", "rogalika", "bagietka", "bagietke"],
"owoce": ["jabłko", "jablko", "banana", "gruszkę", "gruszke", "pomarańczę","pomarancze"], "owoce": ["jabłko", "jablko", "banan", "gruszka", "gruszke", "pomarańcza","pomarancze"],
"warzywa": ["marchewkę", "marchewke", "ziemniak", "cebulę", "cebule", "pomidory", "pomidora"], "warzywa": ["marchewka", "ziemniak", "cebula", "cebule", "pomidor", "pomidora"],
"mięso": ["kurczaka", "wołowinę", "wolowine", "wieprzowinę", "wieprzowine", "indyka"], "mięso": ["kurczak", "wołowina", "wolowine", "wieprzowina", "wieprzowine", "indyk"],
"produkty mrożone": ["lody", "frytki", "pierogi", "nuggetsy"], "produkty mrożone": ["lody", "frytki", "pierogi", "nuggetsy"],
"słodycze": ["czekoladę", "czekolade","czekolady", "ciastko", "lizaka", "gumę do żucia", "gume do zucia", "chipsy"], "słodycze": ["czekolada", "czekolade","czekolady", "ciastko", "lizak", "guma do żucia", "gume do zucia"],
"przekąski": ["talarki", "paluszki","orzeszki", "chipsy"],
"przyprawy": ["sól", "sol", "pieprz", "oregano", "cynamon"], "przyprawy": ["sól", "sol", "pieprz", "oregano", "cynamon"],
"napoje": ["wodę", "wode", "sok", "herbatę", "herbate", "kawę", "kawe", "energetyka"], "napoje": ["woda", "wode", "sok", "herbata", "herbate", "kawa", "kawe", "energetyk"],
"napoje alkoholowe": ["piwo", "wino", "wódkę","wodke", "whisky"], "napoje alkoholowe": ["piwo", "wino", "wódka","wodke", "whisky"],
"higiena": ["pastę do zębów", "paste do zebow", "mydło", "mydlo", "szampon", "papier toaletowy"], "higiena": ["pasta do zębów", "paste do zebow", "mydło", "mydlo", "szampon", "papier toaletowy"],
"chemia gospodarcza": ["płyn do naczyń", "plyn do naczyn", "proszek do prania", "odświeżacz powietrza", "odswiezacz powietrza"], "chemia gospodarcza": ["płyn do naczyń", "plyn do naczyn", "proszek do prania", "odświeżacz powietrza", "odswiezacz powietrza"],
"inne": ["długopis", "baterie", "śrubokręt", "nożyczki"], "inne": ["długopis", "dlugopis", "baterie", "srubokret", "śrubokręt", "nożyczki", "nozyczki"],
"nabiał": ["mleko", "ser", "śmietanę","smietane"] "nabiał": ["mleko", "ser", "śmietana","smietane"]
} }
class DialogAct: class DialogAct:
@ -166,11 +181,11 @@ class NLU:
acts = [] acts = []
for token in doc: for token in doc:
if token.lower_ == "cześć" or token.lower_ == "witaj": if token.lower_ == "cześć" or token.lower_ == "witaj" or token.lower_ == "hej" or token.lower_ == "siema" or token.lower_ == "witam" or token.lower_ == "dzień":
acts.append(DialogAct("hello")) acts.append(DialogAct("hello"))
elif token.lower_ == "widzenia" or token.lower_ == "żegnaj": elif token.lower_ == "widzenia" or token.lower_ == "żegnaj" or token.lower_ == "zobaczenia" or token.lower_ == "wszystko":
acts.append(DialogAct("bye")) acts.append(DialogAct("bye"))
elif token.lower_ == "dziękuję": elif token.lower_ == "dziękuję" or token.lower_ == "dzięki" or token.lower_ == "podziękowania" or token.lower_ == "bóg":
acts.append(DialogAct("thankyou")) acts.append(DialogAct("thankyou"))
elif token.lower_ == "proszę": elif token.lower_ == "proszę":
acts.append(DialogAct("request")) acts.append(DialogAct("request"))
@ -183,7 +198,6 @@ class NLU:
elif token.lower_ in ["nie", "nie chcę"]: elif token.lower_ in ["nie", "nie chcę"]:
acts.append(DialogAct("deny")) acts.append(DialogAct("deny"))
elif token.pos_ == "NOUN": elif token.pos_ == "NOUN":
print(token.lemma_)
product_type, product = self.find_product_type(token.lemma_) product_type, product = self.find_product_type(token.lemma_)
if product_type and product: if product_type and product:
act = DialogAct("inform", {"product type": product_type, "product": product}) act = DialogAct("inform", {"product type": product_type, "product": product})