Update app.py

app.py

@@ -32,7 +32,7 @@ def user(message, history):
 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": item["role"], "content": item["content"]} for item in history])
+ messages.extend(history) # history já deve ser uma lista de dicionários com as chaves "role" e "content"
 
  payload = {
  "messages": messages,
@@ -41,7 +41,7 @@ def call_nvidia_api(history, system_message, max_tokens, temperature, top_p):
  "max_tokens": max_tokens,
  "stream": False
  }
- print(f"Payload enviado: {payload}")
+ print(f"Payload enviado: {json.dumps(payload, indent=2)}")
  session = requests.Session()
  response = session.post(INVOKE_URL, headers=headers, json=payload)
  while response.status_code == 202:
@@ -50,7 +50,7 @@ def call_nvidia_api(history, system_message, max_tokens, temperature, top_p):
  response = session.get(fetch_url, headers=headers)
  response.raise_for_status()
  response_body = response.json()
- print(f"Payload recebido: {response_body}")
+ print(f"Payload recebido: {json.dumps(response_body, indent=2)}")
  if response_body.get("choices"):
  assistant_message = response_body["choices"][0]["message"]["content"]
  history.append({"role": "assistant", "content": assistant_message})
@@ -61,13 +61,13 @@ def chatbot_submit(message, chat_history, system_message, max_tokens_val, temper
  print("Updating chatbot...")
 
  # Adiciona a mensagem do usuário ao histórico
- chat_history = user(message, chat_history)
+ chat_history.append({"role": "user", "content": message})
 
  # Chama a API da NVIDIA para gerar uma resposta
  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]["role"] == "assistant":
+ if chat_history and "content" in chat_history[-1]:
  assistant_message = chat_history[-1]["content"]
  else:
  assistant_message = "Desculpe, ocorreu um erro ao gerar a resposta."