Rozszerzenie zbioru danych Eliza

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?")