From 2dd5b4dd78fc0a30f3d5baa0b99a6b10f002d917 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Mon, 20 Nov 2023 16:51:41 -0500
Subject: [PATCH] Only show last 200 elements in the UI history tab.

---
 execution.py       | 14 ++++++++++++--
 server.py          |  5 ++++-
 web/scripts/api.js |  2 +-
 3 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/execution.py b/execution.py
index 9a2ca5b9..bca48a78 100644
--- a/execution.py
+++ b/execution.py
@@ -750,10 +750,20 @@ class PromptQueue:
                     return True
         return False
 
-    def get_history(self, prompt_id=None):
+    def get_history(self, prompt_id=None, max_items=None, offset=-1):
         with self.mutex:
             if prompt_id is None:
-                return copy.deepcopy(self.history)
+                out = {}
+                i = 0
+                if offset < 0 and max_items is not None:
+                    offset = len(self.history) - max_items
+                for k in self.history:
+                    if i >= offset:
+                        out[k] = self.history[k]
+                        if max_items is not None and len(out) >= max_items:
+                            break
+                    i += 1
+                return out
             elif prompt_id in self.history:
                 return {prompt_id: copy.deepcopy(self.history[prompt_id])}
             else:
diff --git a/server.py b/server.py
index 11bd2a0f..1a8e92b8 100644
--- a/server.py
+++ b/server.py
@@ -431,7 +431,10 @@ class PromptServer():
 
         @routes.get("/history")
         async def get_history(request):
-            return web.json_response(self.prompt_queue.get_history())
+            max_items = request.rel_url.query.get("max_items", None)
+            if max_items is not None:
+                max_items = int(max_items)
+            return web.json_response(self.prompt_queue.get_history(max_items=max_items))
 
         @routes.get("/history/{prompt_id}")
         async def get_history(request):
diff --git a/web/scripts/api.js b/web/scripts/api.js
index b1d245d7..de56b231 100644
--- a/web/scripts/api.js
+++ b/web/scripts/api.js
@@ -256,7 +256,7 @@ class ComfyApi extends EventTarget {
 	 */
 	async getHistory() {
 		try {
-			const res = await this.fetchApi("/history");
+			const res = await this.fetchApi("/history?max_items=200");
 			return { History: Object.values(await res.json()) };
 		} catch (error) {
 			console.error(error);