ai_novelist_endpoint.diff

--- ainovel_v2_081.js	Fri Jan 21 23:27:08 2022
+++ ainovel_v2_081_patch.js	Fri Jan 21 23:27:20 2022
@@ -2141,8 +2141,17 @@
 
 					// write the text to data_edit
 					TextSharding(all_textdata,false);
-					all[all.length - 1].innerHTML = all[all.length - 1].innerHTML + "<span id=\"ai_output\" style=\"color:"+document.getElementById("vis_fontcolor_ai").value+"\">" + result + "</span>";
-					$("#data_container").scrollTop($("#data_edit").height()*4096);
+					let isFound = false;
+					for (const edit_area of all){
+						if (edit_area.innerHTML.indexOf('<br>@endpoint<br>') !== -1) {
+							edit_area.innerHTML = edit_area.innerHTML.replace('<br>@endpoint<br>', "<span id=\"ai_output\" style=\"color:"+document.getElementById("vis_fontcolor_ai").value+"\">" + result + "</span><br>@endpoint<br>");
+							isFound = true;
+						}
+					}
+					if ( ! isFound) {
+						all[all.length - 1].innerHTML = all[all.length - 1].innerHTML + "<span id=\"ai_output\" style=\"color:"+document.getElementById("vis_fontcolor_ai").value+"\">" + result + "</span>";
+						$("#data_container").scrollTop($("#data_edit").height()*4096);
+					}
 
 		 			document.getElementById("retryoptions").style.display = "";