small refactor

This commit is contained in:
Patrick_Pluto 2024-09-16 16:45:46 +02:00
parent f7d23ef603
commit 0edf05669e
2 changed files with 1 additions and 25 deletions

View file

@ -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)

View file

@ -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)