small refactor
This commit is contained in:
parent
f7d23ef603
commit
0edf05669e
2 changed files with 1 additions and 25 deletions
|
@ -1,24 +0,0 @@
|
|||
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)
|
||||
|
|
@ -25,7 +25,7 @@ def index():
|
|||
elif option == "chat":
|
||||
messages.append(f"User: {user_message}")
|
||||
|
||||
ai_response = "AI: " + api.send_message(user_message, 1)
|
||||
ai_response = "AI: " + api.send_message(user_message, 3)
|
||||
messages.append(ai_response)
|
||||
|
||||
return render_template('index.html', messages=messages)
|
||||
|
|
Loading…
Reference in a new issue