Update app.py

app.py

@@ -18,6 +18,11 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=904
     temperature = max(float(temperature), 1e-2)
     top_p = float(top_p)
 
+        if 'chat_history' in localStorage:
+        history = JSON.parse(localStorage.getItem('chat_history'))
+    else:
+        history = []
+
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -31,6 +36,8 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=904
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
+localStorage.setItem('chat_history', JSON.stringify(history))
+
     for response in stream:
         output += response.token.text
         yield output