Update app.py

app.py

@@ -59,7 +59,7 @@ def call_nvidia_api(history, max_tokens, temperature, top_p):
     
     return history
 
-def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetition_penalty):
+def chat(history, system_message, max_tokens, temperature, top_p):
     print("Starting chat...")
     updated_history = call_nvidia_api(history, max_tokens, temperature, top_p)
     return updated_history, ""
@@ -100,13 +100,13 @@ with gr.Blocks() as demo:
     chat_history_state = gr.State([])
 
     # Ajuste na definição da função update_chatbot para aceitar o valor atualizado do system_msg
-    def update_chatbot(message, chat_history, system_message):  # Remova o valor padrão aqui
+    def update_chatbot(message, chat_history, system_message, max_tokens, temperature, top_p):
         print("Updating chatbot...")
         if not chat_history or (chat_history and chat_history[-1]["role"] != "user"):
             chat_history = user(message, chat_history, system_message if not chat_history else None)
         else:
             chat_history = user(message, chat_history)
-        chat_history, _ = chat(chat_history, system_message, max_tokens.value, temperature.value, top_p.value, 40, 1.1)
+        chat_history, _ = chat(chat_history, system_message, max_tokens, temperature, top_p)
     
         formatted_chat_history = []
         for user_msg, assistant_msg in zip([msg["content"].strip() for msg in chat_history if msg["role"] == "user"],
@@ -116,10 +116,9 @@ with gr.Blocks() as demo:
     
         return formatted_chat_history, chat_history, ""
     
-    # Ajuste na chamada do botão submit para incluir system_msg como um input
     submit.click(
         fn=update_chatbot,
-        inputs=[message, chat_history_state, system_msg],  # Inclua system_msg aqui
+        inputs=[message, chat_history_state, system_msg, max_tokens, temperature, top_p],
         outputs=[chatbot, chat_history_state, message]
     )