Change debug prints to only print with --debug

aiserver.py

@@ -6792,9 +6792,11 @@ def ui2_serve_themes(path):
 #==================================================================#
 @socketio.on('upload_file')
 def upload_file(data):
+    if koboldai_vars.debug:
         print("upload_file {}".format(data['filename']))
     if 'current_folder' in session:
         path = os.path.abspath(os.path.join(session['current_folder'], data['filename']).replace("\\", "/")).replace("\\", "/")
+        if koboldai_vars.debug:
             print("Want to save to {}".format(path))
         if 'popup_jailed_dir' not in session:
             print("Someone is trying to upload a file to your server. Blocked.")
@@ -6815,6 +6817,7 @@ def upload_file(data):
 
 @socketio.on('popup_change_folder')
 def popup_change_folder(data):
+    if koboldai_vars.debug:
         print("Doing popup change folder: {}".format(data))
     if 'popup_jailed_dir' not in session:
         print("Someone is trying to get at files in your server. Blocked.")
@@ -7104,6 +7107,7 @@ def UI_2_var_change(data):
 #==================================================================#
 @socketio.on('save_story')
 def UI_2_save_story(data):
+    if koboldai_vars.debug:
         print("Saving Story")
     if data is None:
         #We need to check to see if there is a file already and if it's not the same story so we can ask the client if this is OK
@@ -7143,6 +7147,7 @@ def UI_2_save_to_json():
 #==================================================================#
 @socketio.on('Set Selected Text')
 def UI_2_Set_Selected_Text(data):
+    if koboldai_vars.quiet:
         print("Updating Selected Text: {}".format(data))
     koboldai_vars.actions[int(data['id'])] = data['text']
 
@@ -7163,8 +7168,8 @@ def UI_2_Use_Option_Text(data):
 #==================================================================#
 @socketio.on('submit')
 def UI_2_submit(data):
-    print(data)
     if not koboldai_vars.noai and data['theme'] != "":
+        if koboldai_vars.debug:
             print("doing random prompt")
         memory = koboldai_vars.memory
         koboldai_vars.memory = "{}\n\nYou generate the following {} story concept :".format(koboldai_vars.memory, data['theme'])
@@ -7172,6 +7177,7 @@ def UI_2_submit(data):
         actionsubmit("", force_submit=True, force_prompt_gen=True)
         koboldai_vars.memory = memory
     else:
+        if koboldai_vars.debug:
             print("doing normal input")
         koboldai_vars.actions.clear_unused_options()
         koboldai_vars.lua_koboldbridge.feedback = None
@@ -7186,9 +7192,9 @@ def UI_2_submit(data):
 #==================================================================#
 @socketio.on('abort')
 def UI_2_abort(data):
+    if koboldai_vars.debug:
         print("got abort")
     koboldai_vars.abort = True
-    print(koboldai_vars.abort)
 
  
 #==================================================================#
@@ -7203,6 +7209,7 @@ def UI_2_Pinning(data):
 #==================================================================#
 @socketio.on('back')
 def UI_2_back(data):
+    if koboldai_vars.debug:
         print("back")
     koboldai_vars.actions.clear_unused_options()
     ignore = koboldai_vars.actions.pop()
@@ -7241,7 +7248,6 @@ def UI_2_load_model_button(data):
 #==================================================================#
 @socketio.on('select_model')
 def UI_2_select_model(data):
-    print(data)
     
     #We've selected a menu
     if data['model'] in model_menu:
@@ -7265,7 +7271,6 @@ def UI_2_select_model(data):
 #==================================================================#
 @socketio.on('load_model')
 def UI_2_load_model(data):
-    print(data)
     if not os.path.exists("settings/"):
         os.mkdir("settings")
     changed = True
@@ -7361,6 +7366,7 @@ def story_sort(base_path, desc=False):
 #==================================================================#
 @socketio.on('load_story')
 def UI_2_load_story(file):
+    if koboldai_vars.debug:
         print("loading {}".format(file))
     loadRequest(file)
 
@@ -7377,7 +7383,6 @@ def UI_2_new_story(data):
 #==================================================================#
 @socketio.on('move_wi')
 def UI_2_move_wi(data):
-    print(data)
     if data['folder'] is None:
         koboldai_vars.worldinfo_v2.reorder(int(data['dragged_id']), int(data['drop_id']))
     else:
@@ -7388,7 +7393,6 @@ def UI_2_move_wi(data):
 #==================================================================#
 @socketio.on('wi_set_folder')
 def UI_2_wi_set_folder(data):
-    print(data)
     koboldai_vars.worldinfo_v2.add_item_to_folder(int(data['dragged_id']), data['folder'])
 
 #==================================================================#
@@ -7396,6 +7400,7 @@ def UI_2_wi_set_folder(data):
 #==================================================================#
 @socketio.on('Rename_World_Info_Folder')
 def UI_2_Rename_World_Info_Folder(data):
+    if koboldai_vars.debug:
         print("Rename_World_Info_Folder")
         print(data)
     koboldai_vars.worldinfo_v2.rename_folder(data['old_folder'], data['new_folder'])
@@ -7405,6 +7410,7 @@ def UI_2_Rename_World_Info_Folder(data):
 #==================================================================#
 @socketio.on('edit_world_info')
 def UI_2_edit_world_info(data):
+    if koboldai_vars.debug:
         print("edit_world_info")
         print(data)
     
@@ -7491,7 +7497,6 @@ def UI_2_import_world_info(data):
 #==================================================================#
 @socketio.on('phrase_bias_update')
 def UI_2_phrase_bias_update(biases):
-    print(biases)
     koboldai_vars.biases = biases
 
 
@@ -7513,6 +7518,7 @@ def socket_io_relay(queue, socketio):
 # Event triggered when program errors out
 #==================================================================#
 def my_except_hook(exctype, value, traceback):
+    if koboldai_vars.debug:
         print("sending error to clients")
     socketio.emit("error", "{}: {}".format(exctype, value), broadcast=True, room="UI_2")
     sys.__excepthook__(exctype, value, traceback)
@@ -7527,6 +7533,7 @@ def handle_exception(e):
         return e
 
     # now you're handling non-HTTP exceptions only
+    if koboldai_vars.debug:
         print("sending error to clients")
     socketio.emit("error", "{}: {}".format(e.message, e.args), broadcast=True, room="UI_2")
     return render_template("500_generic.html", e=e), 500
@@ -7570,6 +7577,7 @@ def get_softprompt_desc(item_full_path, item, valid_selection):
 #==================================================================#
 @socketio.on('load_softprompt')
 def UI_2_load_softprompt(data):
+    if koboldai_vars.debug:
         print("Load softprompt: {}".format(data))
     spRequest(data)
 
@@ -7586,6 +7594,7 @@ def UI_2_load_userscripts_list(data):
                                                                   column_widths=['200px', '150px', 'auto'])
                                                                 
 def valid_userscripts_to_load(file):
+    if koboldai_vars.debug:
         print("{} is valid: {}".format(file, file.endswith(".lua") and os.path.basename(file) not in koboldai_vars.userscripts))
     return file.endswith(".lua") and os.path.basename(file) not in koboldai_vars.userscripts
     
@@ -7637,6 +7646,7 @@ def get_userscripts_desc(item_full_path, item, valid_selection):
 #==================================================================#
 @socketio.on('load_userscripts')
 def UI_2_load_userscripts(data):
+        if koboldai_vars.debug:
     print("Loading Userscripts: {}".format(os.path.basename(data)))
     koboldai_vars.userscripts = [x for x in koboldai_vars.userscripts if x != os.path.basename(data)]+[os.path.basename(data)]
     load_lua_scripts()
@@ -7646,6 +7656,7 @@ def UI_2_load_userscripts(data):
 #==================================================================#
 @socketio.on('unload_userscripts')
 def UI_2_unload_userscripts(data):
+    if koboldai_vars.debug:
         print("Unloading Userscript: {}".format(data))
     koboldai_vars.userscripts = [x for x in koboldai_vars.userscripts if x != data]
     load_lua_scripts()
@@ -7657,6 +7668,7 @@ def UI_2_unload_userscripts(data):
 #==================================================================#
 @socketio.on('load_aidg_club')
 def UI_2_load_aidg_club(data):
+    if koboldai_vars.debug:
         print("Load aidg.club: {}".format(data))
     importAidgRequest(data) 
 
@@ -7671,6 +7683,7 @@ def UI_2_theme_change(data):
         for key, value in data['theme'].items():
             f.write("\t{}: {};\n".format(key, value.replace(";", "").replace("--", "-")))
         f.write("}")
+    if koboldai_vars.debug:
         print("Theme Saved")