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

Restart + "clean" console command after user is finished

Browse Source
This commit is contained in:
Adrian Klessa 2024-06-10 20:48:24 +02:00
parent 6598a84e48
commit 7a5064f19e
1 changed files with 23 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
dialog_with_nlg.py
View File

@ -49,11 +49,11 @@ class Model:
r = self.nlu(msg)
slots = r['slots']
#print(r)
# print(r)
r = self.dst(r)
#print(r)
# print(r)
r = self.dp()
#print(r)
# print(r)
r = self.nlg(r, slots)
if debug:
print(r)
@ -193,7 +193,7 @@ class NLG():
def __init__(self, state):
self.model = AutoModelForSeq2SeqLM.from_pretrained("filnow/nlg-umt5-pol")
self.tokenizer = AutoTokenizer.from_pretrained("google/umt5-small")
self.nlg_pipeline = pipeline('summarization', model=self.model, tokenizer=self.tokenizer)
self.nlg_pipeline = pipeline('summarization', model=self.model, tokenizer=self.tokenizer)
self.messages = {
"welcomemsg": [
@ -257,6 +257,7 @@ class NLG():
elif act == "offer":
return "Proszę oto menu zeskanuj kod QR aby je zobaczyć."
def dialogue_test():
model = Model()
@ -285,12 +286,12 @@ def dialogue_test():
response = model("uniwersytetu poznanskiego 4 61-614 poznan")
print()
# jezeli sprobuje dokonac zamowienia bez podania potrzebnych informacji prosimy o nie
#response = model("Dobrze, nie mogę się już doczekać.")
# response = model("Dobrze, nie mogę się już doczekać.")
response = model("Super, to zatem wszystko!")
print()
# jezeli wybierze rodzaj platnosci to zapisz wybor i poinformuj o nim
# response = model("karta")
#response = model("Poproszę blikiem z góry")
# response = model("Poproszę blikiem z góry")
response = model("Zapłacę kartą przy odbiorze")
print()
# jezeli potwiedzi zamowienie to zakoncz zamawianie sukcesem i wypisz calosc
@ -325,10 +326,17 @@ def dialogue_test():
response = model("Dobrze, nie mogę się już doczekać.")
print()
if __name__ == "__main__":
model = Model()
print("Chatbot Jarvis\n--------------")
while True:
print("\nUżytkownik: ")
user_input = input()
while user_input == "SYSTEM_FINISH":
print("\n\n\n\n\n\n\n\n\n\n\n\n")
model = Model()
print("Chatbot Jarvis\n--------------")
print("\nUżytkownik: ")
user_input = input()
response = model(user_input, debug=False)