Update app.py

app.py

@@ -72,7 +72,13 @@ def update_chatbot(message, chat_history, system_message, max_tokens, temperatur
         chat_history = user(message, chat_history)
     chat_history, _ = call_nvidia_api(chat_history, max_tokens, temperature, top_p)
     return chat_history
-    
+
+system_msg = gr.Textbox(BASE_SYSTEM_MESSAGE, label="System Message", placeholder="System prompt.", lines=5)
+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)
+
+
 # Gradio interface setup
 with gr.Blocks() as demo:
     with gr.Row():
@@ -96,23 +102,16 @@ with gr.Blocks() as demo:
 """
     gr.Markdown(description)
     chatbox = gr.Textbox(label="What do you want to chat about?", placeholder="Ask me anything.", lines=3)
-    system_msg = gr.Textbox(BASE_SYSTEM_MESSAGE, label="System Message", placeholder="System prompt.", lines=5)
-    max_tokens = gr.Slider(20, 1024, label="Max Tokens", step=20, value=1024, interactive=True)
-    temperature = gr.Slider(0.0, 1.0, label="Temperature", step=0.1, value=0.2, interactive=True)
-    top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.7, interactive=True)
     chat_history_state = gr.State([])
 
-    additional_inputs = [system_msg, max_tokens, temperature, top_p]
-
     chatbot = gr.ChatInterface(
-        fn=update_chatbot,
-        inputs=[chatbox, chat_history_state],
-        outputs=chat_history_state,
-        additional_inputs=additional_inputs,
+        fn=lambda message, history: update_chatbot(message, history, system_msg.value, max_tokens.value, temperature.value, top_p.value),
+        chatbot=None,  # Use o componente padrão do Chatbot
+        textbox=chatbox,  # Personalize o componente Textbox se necessário
+        additional_inputs=[system_msg, max_tokens, temperature, top_p],  # Componentes adicionais
         title="LLAMA 2 70B Chatbot",
         submit_btn="Submit",
         clear_btn="🗑️ Clear",
-        textbox=chatbox,
     )
 
     chatbot.clear(