Update app.py

app.py

@@ -63,14 +63,14 @@ def chatbot_submit(message, chat_history, system_message, max_tokens_val, temper
  print("Updating chatbot...")
 
  # Atualiza o histórico do chat com a mensagem do usuário
- chat_history.append({"role": "user", "content": message})
+ chat_history.append([message, ""]) # Adiciona a mensagem do usuário com uma resposta vazia
 
  # Chama a API da NVIDIA para gerar uma resposta
  chat_history = call_nvidia_api(chat_history, 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":
- assistant_message = chat_history[-1]["content"]
+ if chat_history and chat_history[-1][1]: # Verifica se há uma resposta do assistente
+ assistant_message = chat_history[-1][1]
  else:
  assistant_message = "Desculpe, ocorreu um erro ao gerar a resposta."
 
@@ -82,11 +82,11 @@ max_tokens = gr.Slider(20, 1024, label="Max Tokens", step=20, value=1024)
 temperature = gr.Slider(0.0, 1.0, label="Temperature", step=0.1, value=0.2)
 top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.7)
 with gr.Blocks() as demo:
+ chat_history_state = gr.State([])
  chatbot = gr.ChatInterface(
  fn=chatbot_submit,
  additional_inputs=[system_msg, max_tokens, temperature, top_p],
  title="LLAMA 70B Free Demo",
- state=chat_history_state, 
  description="""
  <div style="text-align: center; font-size: 1.5em; margin-bottom: 20px;">
  <strong>Explore the Capabilities of LLAMA 2 70B</strong>