From efff4fb1a872480bfd213b71d5c051f7f9096c8e Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Fri, 30 Sep 2022 13:07:52 -0400
Subject: [PATCH] Fix for large UI1 story loads

---
 aiserver.py          | 7 +++++--
 koboldai_settings.py | 1 -
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/aiserver.py b/aiserver.py
index 981a194c..1338d859 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -7053,8 +7053,11 @@ def load_story_v1(js):
         else:
             koboldai_vars.gamestarted = False
     if(koboldai_vars.gamestarted):
-        for s in actions:
-            koboldai_vars.actions.append(s)
+        koboldai_vars.actions.action_count += 1
+        for i in range(len(js["actions"])):
+            if i == len(js["actions"])-1:
+                koboldai_vars.actions.action_count -= 1
+            koboldai_vars.actions.append(js["actions"][i])
 
     if "actions_metadata" in js:
         if type(js["actions_metadata"]) == dict:
diff --git a/koboldai_settings.py b/koboldai_settings.py
index 36a20fdd..9eab7394 100644
--- a/koboldai_settings.py
+++ b/koboldai_settings.py
@@ -1113,7 +1113,6 @@ class KoboldStoryRegister(object):
         self.set_game_saved()
         self.story_settings.save_story()
         
-            
     def append(self, text):
         self.clear_unused_options()
         self.action_count+=1