still wip

Code/Janet.py

@@ -1,9 +1,9 @@
-import jsgf
+#import jsgf
 
 from Modules.NLG_module import NLG
 from Modules.DP_module import DP
 from Modules.DST_module import Rules_DST
-from Modules.Book_NLU_module import Book_NLU
+# from Modules.Book_NLU_module import Book_NLU
 from Modules.ML_NLU_module import ML_NLU
 
 import random
@@ -23,7 +23,7 @@ class Janet:
         self.nlg = NLG()
         self.dp = DP()
         self.dst = Rules_DST()
-        self.nlu = Book_NLU(jsgf.parse_grammar_file('book.jsgf'))
+        #self.nlu = Book_NLU(jsgf.parse_grammar_file('book.jsgf'))
         self.nlu_v2 = ML_NLU()
 
     def process(self, command):

Code/Modules/ML_NLU_module.py

@@ -1,5 +1,3 @@
-import jsgf
-from tabulate import tabulate
 from flair.data import Sentence, Token
 from flair.datasets import SentenceDataset
 from flair.models import SequenceTagger

Code/Modules/NLG_module.py

@@ -22,13 +22,19 @@ class NLG:
         if(len(act) == 0):
             return "Nie mam już nic do powiedzenia :("
 
-        for variable in act:
-            domain, intent = variable
+        domain, intent = act[0]
 
-            if (domain == "inform"):
-                return "test"
-            if(domain == "greeting"):
-                return random.choice(["Cześć, mam na imię Janet", "Hej, jestem Janet. W czym mogę pomóc?", "Dzień dobry, nazywam się Janet"])
+        if (domain == "inform"):
+            return "test"
 
-            elif(domain == "end_conversation"):
-                return random.choice(["Dziękujemy za skorzystanie z naszych usług!", "Do widzenia!", "Do zobaczenia!"])
+        if(domain == "greeting"):
+            return random.choice(["Cześć, mam na imię Janet", "Hej, jestem Janet. W czym mogę pomóc?", "Dzień dobry, nazywam się Janet"])
+
+        if(domain == "end_conversation"):
+            return random.choice(["Dziękujemy za skorzystanie z naszych usług!", "Do widzenia!", "Do zobaczenia!"])
+
+        if(domain == "appointment"):
+            if(intent == "book_appointent"):
+                answer = "Zarezerwowano wizytę"
+                print(variable)
+