Update app.py

app.py

@@ -177,7 +177,8 @@ 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.choices[0].message.to_dict())
+        history.append(completion.to_dict())
+        #history.append(completion.choices[0].message.to_dict())
         #history.append(completion["result"].choices[0].message.to_dict())
 
         state['total_tokens'] += completion['usage']['total_tokens']
@@ -187,8 +188,8 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
         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)]
-    chat_messages = [(history[0]['content'],history[1]['content'])]
+    chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]
+    #chat_messages = [(history[0]['content'],history[1]['content'])]
     return '', chat_messages, total_tokens_used_msg, state
 
 def clear_conversation():