Update Gpt4oDemo.py

Gpt4oDemo.py

@@ -244,12 +244,12 @@ def new_prompt(prompt):
 
         # Extract the text content from the response and append it to the chat history
         assistant_message = response.choices[0].message.content
-        chat_history.append({'role': 'model', 'content': assistant_message})
+        chat_history.append({'role': 'system', 'content': assistant_message})
         print(assistant_message)
     except Exception as e:
         print("Error: ",e)
         assistant_message = "API rate limit has been reached. Please wait a moment and try again."
-        chat_history.append({'role': 'model', 'content': assistant_message})
+        chat_history.append({'role': 'system', 'content': assistant_message})
 
     # except google.api_core.exceptions.ResourceExhausted:
     #     assistant_message = "API rate limit has been reached. Please wait a moment and try again."