s464915/JARVIS
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Dialogue loop

Browse Source
This commit is contained in:
Adrian Klessa 2024-06-10 20:39:26 +02:00
parent 4b1c33bf98
commit 6598a84e48
1 changed files with 17 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
dialog_with_nlg.py
View File

@ -42,8 +42,9 @@ class Model:
self.dp = DP(self.state) self.dp = DP(self.state)
self.nlg = NLG(self.state) self.nlg = NLG(self.state)
def __call__(self, prompt) -> Any: def __call__(self, prompt, debug=True) -> Any:
print(prompt) if debug:
print(prompt)
msg = prompt.lower() msg = prompt.lower()
r = self.nlu(msg) r = self.nlu(msg)
@ -54,7 +55,10 @@ class Model:
r = self.dp() r = self.dp()
#print(r) #print(r)
r = self.nlg(r, slots) r = self.nlg(r, slots)
print(r) if debug:
print(r)
else:
print(f"JARVIS: {r}")
return r return r
@ -253,8 +257,7 @@ class NLG():
elif act == "offer": elif act == "offer":
return "Proszę oto menu zeskanuj kod QR aby je zobaczyć." return "Proszę oto menu zeskanuj kod QR aby je zobaczyć."
def dialogue_test():
if __name__ == "__main__":
model = Model() model = Model()
# jezeli sie przywita to przywitaj uzytkownika (i pokaz menu) # jezeli sie przywita to przywitaj uzytkownika (i pokaz menu)
@ -320,4 +323,12 @@ if __name__ == "__main__":
response = model("Zapłacę kartą przy odbiorze") response = model("Zapłacę kartą przy odbiorze")
print() print()
response = model("Dobrze, nie mogę się już doczekać.") response = model("Dobrze, nie mogę się już doczekać.")
print() print()
if __name__ == "__main__":
model = Model()
print("Chatbot Jarvis\n--------------")
while True:
print("\nUżytkownik: ")
user_input = input()
response = model(user_input, debug=False)