System_Dialogowy_Janet/Code/Modules/NLG_module.py

import random

class NLG:
    """
    Moduł, który tworzy reprezentację tekstową aktu systemowego wybranego przez taktykę dialogu.

    Wejście: Akt systemu (rama)

    Wyjście: Tekst
    """
    def __init__(self):
        pass

    def change_to_text(self, system_acts):
        """
        Funkcja zamieniająca akt systemu na tekst rozumiany przez użytkownika.
        """
        if(len(system_acts) == 0):
            return "Nie mam już nic do powiedzenia :("

        act = system_acts.pop(0)
        if(len(act) == 0):
            return "Nie mam już nic do powiedzenia :("

        for variable in act:
            domain, intent = variable

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

            elif(domain == "end_conversation"):
                return random.choice(["Dziękujemy za skorzystanie z naszych usług!", "Do widzenia!", "Do zobaczenia!"])