From 35829302e935d86d4da45ef4c0f8ed99e4920016 Mon Sep 17 00:00:00 2001
From: Nils Kruthoff <nils.kruthoff@dlr.de>
Date: Mon, 27 Jul 2026 19:34:29 +0200
Subject: [PATCH] added a DocumentManager.cs to buffer only the active document
 page and its images

---
 .../Tools/DocumentManager.cs                  | 58 +++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 app/MindWork AI Studio/Tools/DocumentManager.cs

diff --git a/app/MindWork AI Studio/Tools/DocumentManager.cs b/app/MindWork AI Studio/Tools/DocumentManager.cs
new file mode 100644
index 00000000..8a033b41
--- /dev/null
+++ b/app/MindWork AI Studio/Tools/DocumentManager.cs	
@@ -0,0 +1,58 @@
+using System.Text;
+
+namespace AIStudio.Tools;
+
+/// <summary>
+/// Buffers only the active document page so that its image segments can follow
+/// the page Markdown without retaining the complete document in memory.
+/// </summary>
+public sealed class DocumentManager
+{
+    private int currentPageNumber;
+    private StringBuilder? currentPageContent;
+
+    public string? AddPage(ContentStreamDocumentMetadata metadata, string? content, bool extractImages)
+    {
+        var pageNumber = metadata.Document?.PageNumber ?? 0;
+        if (pageNumber == 0)
+            return content;
+
+        var image = metadata.Document?.Image;
+        if (image is null)
+        {
+            var completedPage = this.Flush();
+            this.currentPageNumber = pageNumber;
+            this.currentPageContent = new StringBuilder();
+            this.currentPageContent.AppendLine($"# Page {pageNumber}");
+            this.currentPageContent.Append(content);
+            return completedPage;
+        }
+
+        if (!extractImages || this.currentPageContent is null || string.IsNullOrWhiteSpace(image.Id))
+            return null;
+
+        if (ContentStreamSseHandler.ProcessImageSegment(image.Id, image))
+        {
+            var base64 = ContentStreamSseHandler.BuildImage(image.Id);
+            if (!string.IsNullOrWhiteSpace(base64))
+            {
+                var mediaType = string.IsNullOrWhiteSpace(image.MediaType) ? "image/jpeg" : image.MediaType;
+                this.currentPageContent.AppendLine();
+                this.currentPageContent.AppendLine($"![Image](data:{mediaType};base64,{base64})");
+            }
+        }
+
+        return null;
+    }
+
+    public string? Flush()
+    {
+        if (this.currentPageContent is null)
+            return null;
+
+        var result = this.currentPageContent.ToString();
+        this.currentPageContent = null;
+        this.currentPageNumber = 0;
+        return string.IsNullOrWhiteSpace(result) ? null : result;
+    }
+}