Some fixes

Author: lmg-anon

mikupad.html

@@ -3850,15 +3850,15 @@
 				// split the chunk in 2
 				const left = { content: chunk.content.substring(0, chunk.content.indexOf(fillPlaceholder)), type: "user" };
 				const right = { content: chunk.content.substring(chunk.content.indexOf(fillPlaceholder) + fillPlaceholder.length), type: "user" };
-				fillIdx = i + 1;
 				leftPromptChunks = [
-					...promptChunks.slice(0, Math.max(0, i - 1)),
-					...[left]
+					...promptChunks.slice(0, i),
+					left
 				];
 				rightPromptChunks = [
-					...[right],
+					right,
 					...promptChunks.slice(i + 1, promptChunks.length - 1),
 				];
+				fillIdx = i + 1;
 				break;
 			}
 		}
@@ -3872,9 +3872,9 @@
 		const finalPrompt = assembleFinalPrompt(additionalContextPrompt);
 
 		predict(finalPrompt, leftPromptChunks.length, (chunk) => {
-			console.log(chunk);
 			if (rightPromptChunks[0]) {
-				if (chunk.content.trim().startsWith(rightPromptChunks[0].content[0])) {
+				const trimmedChunk = chunk.content.replace(/^ +| +$/gm, "")
+				if (trimmedChunk.startsWith(rightPromptChunks[0].content[0])) {
 					if (chunk.content[0] == ' ' && rightPromptChunks[0].content[0] != ' ') {
 						rightPromptChunks[0].content = ' ' + rightPromptChunks[0].content;
 						setPromptChunks(p => [
@@ -3956,8 +3956,10 @@
 				while (undoStack.current.at(-1) >= chunkCount)
 					undoStack.current.pop();
 				undoStack.current.push(chunkCount);
-				redoStack.current = [];
+			} else {
+				undoStack.current = [];
 			}
+			redoStack.current = [];
 			setUndoHovered(false);
 			setRejectedAPIKey(false);
 			promptArea.current.scrollTarget = undefined;