Update app.py

app.py

@@ -31,8 +31,12 @@ def user(message, history):
 
 def call_nvidia_api(history, system_message, max_tokens, temperature, top_p):
  """Calls the NVIDIA API to generate a response."""
+ # Transforma o histórico de listas de listas para o formato esperado pela API
  messages = [{"role": "system", "content": system_message}]
- messages.extend(history) # history já deve ser uma lista de dicionários com as chaves "role" e "content"
+ for msg in history:
+ messages.append({"role": "user", "content": msg[0]})
+ if msg[1]: # Garante que não adicionamos mensagens vazias do assistente
+ messages.append({"role": "assistant", "content": msg[1]})
 
  payload = {
  "messages": messages,
@@ -41,7 +45,6 @@ def call_nvidia_api(history, system_message, max_tokens, temperature, top_p):
  "max_tokens": max_tokens,
  "stream": False
  }
- 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,29 +53,26 @@ 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: {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})
+ history.append(["", assistant_message]) # Adiciona a nova mensagem do assistente ao histórico
  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...")
-
  # Adiciona a mensagem do usuário ao histórico
- chat_history.append({"role": "user", "content": message})
+ updated_history = chat_history + [[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)
+ updated_history = call_nvidia_api(updated_history, system_message, max_tokens_val, temperature_val, top_p_val)
 
- # Extrai apenas a mensagem do assistente da resposta
- if chat_history and "content" in chat_history[-1]:
- assistant_message = chat_history[-1]["content"]
+ # A última entrada no histórico atualizado contém a última mensagem do usuário e a resposta do assistente
+ if updated_history and updated_history[-1][1]:
+ assistant_message = updated_history[-1][1]
  else:
  assistant_message = "Desculpe, ocorreu um erro ao gerar a resposta."
 
- return assistant_message, chat_history
+ return assistant_message, updated_history
 
 chat_history_state = gr.State([])
 system_msg = gr.Textbox(BASE_SYSTEM_MESSAGE, label="System Message", placeholder="System prompt.", lines=5)