Update app.py

app.py

@@ -58,10 +58,9 @@ def submit_message(user_token, prompt, prompt_template, good_foods, bad_foods, t
             max_tokens=max_tokens,
             stream=False)
 
-        history.append(prompt_msg.content)
+        history.append(prompt_msg)
         answer = completion.choices[0].message.content
-        # history.append(completion.choices[0].message.to_dict())
-        history.append(answer)
+        history.append(completion.choices[0].message)
 
         state['total_tokens'] += completion['usage']['total_tokens']
 
@@ -80,14 +79,14 @@ def submit_message(user_token, prompt, prompt_template, good_foods, bad_foods, t
         state['total_tokens'] += completion2['usage']['total_tokens']
 
     except Exception as e:
-        history.append(prompt_msg.content)
+        history.append(prompt_msg)
         history.append({
             "role": "system",
             "content": f"Error: {e}"
         })
 
     total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
-    chat_messages = [(history[i], history[i+1]) for i in range(0, len(history)-1, 2)]
+    chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]
 
     return '', chat_messages, total_tokens_used_msg, state, table