OOPified.

py/simple_chat.py

@@ -1,20 +1,24 @@
 from api import API
 
-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("ungültiges Modell")
-        continue
-    while True:
-        print("")
-        print("Ask a question")
-        inputText = input ()
-        if inputText == "change":
-            break
-        outputText = chat1.send_message(inputText, model)
-        print(outputText)
+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)