Update app.py

app.py

@@ -125,7 +125,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
 # https://colab.research.google.com/drive/1dzdNDZyofRB0f2KIB4gHXmIza7ehMX30?usp=sharing#scrollTo=b-ejDn_JfpWW
         
         history.append(prompt_msg)
-        history.append(completion)
+        history.append(completion.message.to_dict())
         #history.append(completion.choices[0].message.to_dict())
         #history.append(completion["result"].choices[0].message.to_dict())
 
@@ -133,10 +133,7 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     
     except Exception as e:
         history.append(prompt_msg)
-        history.append({
-            "role": "system",
-            "content": f"Error: {e}"
-        })
+        history.append("error")
 
     total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
     chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]