can exit kudasai when selecting translation method

Bikatr7 · May 15, 2024 · eec366c · eec366c
1 parent 24da5b1
commit eec366c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/modules/common/translator.py b/modules/common/translator.py
@@ -185,7 +185,12 @@ async def initialize() -> None:
             "3": ("deepl", FileEnsurer.deepl_api_key_path),
         }
 
-        method = input("What method would you like to use for translation? (1 for OpenAI, 2 for Gemini, 3 for Deepl) : \n")
+        method = input("What method would you like to use for translation? (1 for OpenAI, 2 for Gemini, 3 for Deepl, or any other key to exit) : \n")
+
+        if(method not in translation_methods.keys()):
+            print("\nThank you for using Kudasai, goodbye.")
+            time.sleep(2)
+            FileEnsurer.exit_kudasai()
 
         Translator.TRANSLATION_METHOD, api_key_path = translation_methods.get(method, ("deepl", FileEnsurer.deepl_api_key_path))