hisotry now working, also with system prompt.

query.py

@@ -265,23 +265,24 @@ def load_history(state, history_id):
         history = history[:-1]
     else:
         user_input = ''
+    
+    if history:
+        state["system"] = get_system_prompt(history[0][0])
 
     return state, history, history, user_input   # state, Chatbot, ChatInterface.state, ChatInterface.textbox
  
+def get_system_prompt(message):
+    system_prompt = prompt_chain.invoke(message)
+    return system_prompt.messages[0]
+
 def chat(message, history, state, request:gr.Request):
     if not_authenticated(state):
         yield "You need to authenticate first"
     else:
         if AI:
             if not history:
-                system_prompt = prompt_chain.invoke(message)
-                system_prompt = system_prompt.messages[0]
-                state["system"] = system_prompt
-                
-                # Next is commented out because astra has a limit on document size
-                # add_history(state, request, "system", system_prompt, name=message)
-            else:
-                system_prompt = state["system"]
+                state["system"] = get_system_prompt(message)
+            system_prompt = state["system"]
 
             add_history(state, request, "user", message)