Update app.py

app.py

@@ -34,10 +34,13 @@ def user(message, history, system_message=None):
     return history
 
 
-def call_nvidia_api(history, max_tokens, temperature, top_p):
+def call_nvidia_api(history, system_message, max_tokens, temperature, top_p):
     """Calls the NVIDIA API to generate a response."""
+    messages = [{"role": "system", "content": system_message}]
+    messages.extend([{"role": "user", "content": h[0]} for h in history])
+
     payload = {
-        "messages": history,
+        "messages": messages,
         "temperature": temperature,
         "top_p": top_p,
         "max_tokens": max_tokens,
@@ -53,23 +56,23 @@ def call_nvidia_api(history, max_tokens, temperature, top_p):
         response.raise_for_status()
     response_body = response.json()
     print(f"Payload recebido: {response_body}")
-    if response_body["choices"]:
+    if response_body.get("choices"):
         assistant_message = response_body["choices"][0]["message"]["content"]
-        history.append({"role": "assistant", "content": assistant_message})
+        history.append(["", assistant_message])
     return history
-    
+
 def chatbot_submit(message, chat_history, system_message, max_tokens_val, temperature_val, top_p_val):
     """Submits the user message to the chatbot and updates the chat history."""
     print("Updating chatbot...")
 
-    # Atualiza o histórico do chat com a mensagem do usuário
-    chat_history.append([message, ""])  # Adiciona a mensagem do usuário com uma resposta vazia
+    # Adiciona a mensagem do usuário ao histórico
+    chat_history.append([message, ""])
 
     # Chama a API da NVIDIA para gerar uma resposta
-    chat_history = call_nvidia_api(chat_history, max_tokens_val, temperature_val, top_p_val)
+    chat_history = call_nvidia_api(chat_history, system_message, max_tokens_val, temperature_val, top_p_val)
 
     # Extrai apenas a mensagem do assistente da resposta
-    if chat_history and chat_history[-1][1]:  # Verifica se há uma resposta do assistente
+    if chat_history and chat_history[-1][1]:
         assistant_message = chat_history[-1][1]
     else:
         assistant_message = "Desculpe, ocorreu um erro ao gerar a resposta."