Update app.py

app.py

@@ -59,14 +59,14 @@ def submit_message(prompt, prompt_template, good_foods, bad_foods, temperature,
             stream=False)
 
         history.append(prompt_msg)
-        answer = completion.choices[0].message.content
-        history.append(completion.choices[0].message)
+        answer = {'role': 'assistant', 'content': completion.choices[0].message.content }
+        history.append(answer)
 
         state['total_tokens'] += completion['usage']['total_tokens']
 
         # completion2 = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=system_prompt + food_priming_prompt + history[-1:] + [classification_msg], temperature=temperature, max_tokens=max_tokens)
         # print(completion2)
-        messages2 = system_prompt + food_priming_prompt + history[-1:] + [classification_msg]
+        messages2 = system_prompt + food_priming_prompt + answer + [classification_msg]
         completion2 = client.chat.completions.create(
             model=LLM_MODEL, 
             messages=messages2,