diff --git a/chatbot/data/intents.json b/chatbot/data/intents.json index 7544c12..6db5d3e 100644 --- a/chatbot/data/intents.json +++ b/chatbot/data/intents.json @@ -1,5 +1,5 @@ { - "name_query": [ + "ask_name": [ "jak masz na imię", "jak się nazywasz", "twoje imię", diff --git a/chatbot/data/response.json b/chatbot/data/response.json index 407c32f..c6311ac 100644 --- a/chatbot/data/response.json +++ b/chatbot/data/response.json @@ -1,5 +1,5 @@ { - "ask_name": [ + "name_response": [ "Witaj, nazywam się Dia.", "Cześć! Jestem Dia.", "Hej, jestem Dia.", diff --git a/chatbot/main.py b/chatbot/main.py index 9579d09..64b389a 100644 --- a/chatbot/main.py +++ b/chatbot/main.py @@ -27,6 +27,7 @@ def main(): intent = nlp.analyze(user_input) response = generator.generate(intent) + print(Fore.CYAN + "Bot: " + response) diff --git a/chatbot/modules/generator.py b/chatbot/modules/generator.py index fe5af38..b0742b3 100644 --- a/chatbot/modules/generator.py +++ b/chatbot/modules/generator.py @@ -8,6 +8,11 @@ class ResponseGenerator: def __init__(self, config: Config): with config.responses_path.open('r', encoding='utf-8') as file: self.responses: Dict[str, list] = json.load(file) + self.intent_to_response_key = { + "ask_name": "name_response", + "unknown": "unknown" + } def generate(self, response_key: str) -> str: + response_key = self.intent_to_response_key.get(response_key, "unknown") return random.choice(self.responses.get(response_key, ["Przepraszam, nie rozumiem. Możesz to powtórzyć?"])) diff --git a/chatbot/modules/nlp.py b/chatbot/modules/nlp.py index 7aad3e1..45bb459 100644 --- a/chatbot/modules/nlp.py +++ b/chatbot/modules/nlp.py @@ -18,3 +18,4 @@ class NaturalLanguageProcessor: if any(phrase in lower_text for phrase in phrases): return intent return "unknown" +