24 lines
692 B
Python
24 lines
692 B
Python
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)
|
|
|