Update app.py

app.py

@@ -16,8 +16,17 @@ headers = {
 BASE_SYSTEM_MESSAGE = "I carefully provide accurate, factual, thoughtful, nuanced answers and am brilliant at reasoning."
 
 def call_nvidia_api(message, history_api, system_message, max_tokens, temperature, top_p):
-    messages = [{"role": "system", "content": system_message}] if system_message else []
-    messages.extend([{"role": "user", "content": msg[0]}, {"role": "assistant", "content": msg[1]} for msg in history_api])
+    messages = []
+    if system_message:  # Adiciona a mensagem do sistema, se houver
+        messages.append({"role": "system", "content": system_message})
+    # Adiciona as mensagens do histórico
+    for msg in history_api:
+        messages.append({"role": "user", "content": msg[0]})
+        if msg[1]:  # Garante que não adicionamos respostas vazias
+            messages.append({"role": "assistant", "content": msg[1]})
+
+    # Adiciona a mensagem atual do usuário
+    messages.append({"role": "user", "content": message})
 
     payload = {
         "messages": messages,
@@ -42,6 +51,7 @@ def call_nvidia_api(message, history_api, system_message, max_tokens, temperatur
     else:
         return "Desculpe, ocorreu um erro ao gerar a resposta."
 
+
 def chatbot_function(message, history_api, system_message, max_tokens, temperature, top_p):
     assistant_message = call_nvidia_api(message, history_api, system_message, max_tokens, temperature, top_p)
     history_api.append([message, assistant_message])