From 589bed35d6efcf23a9d654ddc9508f3f99c32840 Mon Sep 17 00:00:00 2001 From: Krzysztof Bojakowski Date: Fri, 29 Mar 2024 21:49:58 +0100 Subject: [PATCH] Rozszerzenie zbioru danych Eliza --- eliza.py | 91 ++++++++++++++++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 88 insertions(+), 3 deletions(-) diff --git a/eliza.py b/eliza.py index c9a13d8..f9dc561 100644 --- a/eliza.py +++ b/eliza.py @@ -1,6 +1,82 @@ from nltk.chat.util import Chat, reflections pairs = ( + ( + r"Dzień dobry(.*)", + ("Dla kogo dobry dla tego dobry.", "Dzień dobry."), + ), + ( + r"Cześć(.*)", + ( + "Hej, jak się masz?", + "Jak się dzisiaj czujesz?", + ), + ), + ( + r"Hej(.*)", + ( + "Cześć, co tam u Ciebie?", + "Witam.", + ), + ), + ( + r"Elo(.*)", + ( + "Siema!", + "3 2 0.", + ), + ), + ( + r"Jak (.*)", + ( + "To zależy od Ciebie.", + "To nie takie proste, co o tym myślisz?", + ), + ), + ( + r"Dlaczego (.*)", + ( + "Zastanów się nad tym.", + "Ty mi powiedz dlaczego %1", + "Czemu mnie o to pytasz?", + ), + ), + ( + r"Co (.*)", + ( + "Skąd to pytanie?", + "Co o tym myślisz?", + "Odpowiedz sobie sam.", + ), + ), + ( + r"Czemu (.*)", + ("Tak to już jest.", "Takie życie.", "Bywa."), + ), + ( + r"Czy (.*)", + ( + "Tak", + "Nie", + "Nie wiem czy %1", + ), + ), + ( + r"(.*)\?", + ( + "Czemu o to pytasz?", + "Może odpowiedź jest bliżej niż myślisz.", + "Wytłumacz mi to?", + ), + ), + ( + r"(.*)\!", + ( + "Czemu krzyczysz?", + "Zachowuj się!", + "Czemu się złościsz?", + ), + ), ( r"Potrzebuję (.*)", ( @@ -9,14 +85,23 @@ pairs = ( ), ), ( - r"koniec", + r"(.*) prac(.)(.*)", ( - "Do widzenia", + "Lubisz swoją pracę?", + "Co najbardziej lubisz w swojej pracy?", ), ), + (r"tak", ("Dlaczego tak?",)), + (r"nie", ("Dlaczego nie?")), + ( + r"koniec", + ("Do widzenia",), + ), ( r"(.*)", ( + "Czemu to powiedziałeś?", + "Zmieńmy temat, powiedz mi o swojej pracy.", "Możesz rozwinąć?", "Rozumiem.", "Ciekawe.", @@ -31,7 +116,7 @@ eliza_chatbot = Chat(pairs, reflections) def eliza_chat(): print("Terapeuta\n---------") print("Rozmawiaj po Polsku.") - print('Używaj zarówno wielkich, jak i małych lister; używaj interpunkcji.') + print("Używaj zarówno wielkich, jak i małych lister; używaj interpunkcji.") print('Aby zakończyć, napisz "koniec".') print("=" * 72) print("Witaj, jak się czujesz?")