Add system prompt to chat history

app.py

@@ -79,15 +79,16 @@ with gr.Blocks() as demo:
 
     def update_chatbot(message, chat_history):
         print("Updating chatbot...")
+        # Inclui o prompt do sistema como a primeira mensagem, se necessário
+        if not chat_history or (chat_history and chat_history[0]["content"] != BASE_SYSTEM_MESSAGE):
+            chat_history.insert(0, {"role": "system", "content": BASE_SYSTEM_MESSAGE})
         if not chat_history or (chat_history and chat_history[-1]["role"] != "user"):
             chat_history = user(message, chat_history)
-        chat_history, _ = chat(chat_history, system_msg.value, max_tokens.value, temperature.value, top_p.value, 40, 1.1)
+        chat_history, _ = chat(chat_history, BASE_SYSTEM_MESSAGE, max_tokens.value, temperature.value, top_p.value, 40, 1.1)
         # Formata o histórico do chat para o formato esperado pelo gr.Chatbot
-        formatted_chat_history = [(msg["role"], msg["content"]) for msg in chat_history]
+        formatted_chat_history = [f"{msg['role']}: {msg['content']}" for msg in chat_history]
         return formatted_chat_history, chat_history, ""
 
-
-
     submit.click(
         fn=update_chatbot,
         inputs=[message, chat_history_state],