ai-virtual-assistant/py/simple_chat.py

from api import API

class CLIChat:
    @staticmethod
    def chat():
        chat1 = API()

        while True:
            print("")
            print("Which AI Model do you want to use? Write as a Number (1-5).")
            model_input = input()
            model = int(model_input)
            if model <= 0 or model > 5:
                print("Invalid model.")
                continue
            while True:
                print("")
                print("Ask a question")
                input_text = input()
                if input_text == "change":
                    break
                output_text = chat1.send_message(input_text, model)
                print(output_text)