Update app.py

app.py

@@ -22,21 +22,10 @@ def clear_chat():
     chat_history_state.value = []
     chatbot.textbox.value = ""
 
-def user(message, history):
-    """Updates the chat history with the user message."""
-    print(f"User message: {message}")
-    history = history or []
-    history.append({"role": "user", "content": message})
-    return 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}]
-    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]})
+    messages.extend([{"role": "user", "content": h[0]} for h in history])
 
     payload = {
         "messages": messages,
@@ -45,6 +34,7 @@ def call_nvidia_api(history, system_message, max_tokens, temperature, top_p):
         "max_tokens": max_tokens,
         "stream": False
     }
+    print(f"Payload enviado: {payload}")
     session = requests.Session()
     response = session.post(INVOKE_URL, headers=headers, json=payload)
     while response.status_code == 202:
@@ -53,26 +43,27 @@ 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}")
     if response_body.get("choices"):
         assistant_message = response_body["choices"][0]["message"]["content"]
-        history.append(["", assistant_message])  # Adiciona a nova mensagem do assistente ao histórico
-    return history
+        return assistant_message
+    else:
+        return "Desculpe, ocorreu um erro ao gerar a resposta."
 
 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."""
-    # Adiciona a mensagem do usuário ao histórico
-    updated_history = chat_history + [[message, ""]]
+    print("Updating chatbot...")
+
+    # Adiciona a mensagem do usuário ao histórico para exibição
+    chat_history.append([message, ""])
 
     # Chama a API da NVIDIA para gerar uma resposta
-    updated_history = call_nvidia_api(updated_history, system_message, max_tokens_val, temperature_val, top_p_val)
+    assistant_message = call_nvidia_api(chat_history, system_message, max_tokens_val, temperature_val, top_p_val)
 
-    # 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."
+    # Atualiza o histórico com a resposta do assistente
+    chat_history[-1][1] = assistant_message
 
-    return assistant_message, updated_history
+    return assistant_message, chat_history
 
 chat_history_state = gr.State([])
 system_msg = gr.Textbox(BASE_SYSTEM_MESSAGE, label="System Message", placeholder="System prompt.", lines=5)