Update app.py

app.py

@@ -22,10 +22,13 @@ def clear_chat():
     chat_history_state.value = []
     chatbot.textbox.value = ""
 
-def call_nvidia_api(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": api_history,
+        "messages": messages,
         "temperature": temperature,
         "top_p": top_p,
         "max_tokens": max_tokens,
@@ -47,24 +50,20 @@ def call_nvidia_api(api_history, max_tokens, temperature, top_p):
     else:
         return "Desculpe, ocorreu um erro ao gerar a resposta."
 
-def chatbot_submit(message, chat_history, api_history, system_message, max_tokens_val, temperature_val, top_p_val):
-    """Submits the user message and updates both histories."""
-    # Update Gradio history
-    chat_history.append([message, ""])
+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...")
 
-    # Update API history
-    api_history.append({"role": "user", "content": message})
+    # Adiciona a mensagem do usuário ao histórico para exibição
+    chat_history.append([message, ""])
 
-    # Call NVIDIA API
-    assistant_message = call_nvidia_api(api_history, max_tokens_val, temperature_val, top_p_val)
+    # Chama a API da NVIDIA para gerar uma resposta
+    assistant_message = call_nvidia_api(chat_history, system_message, max_tokens_val, temperature_val, top_p_val)
 
-    # Update Gradio history with response
+    # Atualiza o histórico com a resposta do assistente
     chat_history[-1][1] = assistant_message
 
-    # Update API history with response
-    api_history.append({"role": "assistant", "content": assistant_message})
-
-    return assistant_message, chat_history, api_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)