Update app.py

app.py

@@ -63,7 +63,20 @@ def chat(history, system_message, max_tokens, temperature, top_p, top_k, repetit
     print("Starting chat...")
     updated_history = call_nvidia_api(history, max_tokens, temperature, top_p)
     return updated_history, ""
-
+    
+def update_chatbot(message, chat_history, system_message, max_tokens, temperature, top_p):
+    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 = call_nvidia_api(chat_history, max_tokens, temperature, top_p)
+
+    formatted_chat_history = []
+    for msg in chat_history:
+        if msg["role"] == "user" or msg["role"] == "assistant":
+            formatted_chat_history.append([msg["content"].strip()])
+
+    return formatted_chat_history, chat_history
 # Gradio interface setup
 with gr.Blocks() as demo:
     with gr.Row():
@@ -88,46 +101,32 @@ with gr.Blocks() as demo:
 <p> <strong>HF Created by:</strong> @artificialguybr (<a href="https://twitter.com/artificialguybr">Twitter</a>)</p>
 <p> <strong>Discover more:</strong> <a href="https://artificialguy.com">artificialguy.com</a></p>
 """
-    gr.Markdown(description)
-    chatbot = gr.Chatbot()
-    message = gr.Textbox(label="What do you want to chat about?", placeholder="Ask me anything.", lines=3)
-    submit = gr.Button(value="Send message")
-    clear = gr.Button(value="New topic")
-    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([])
-
-    # Ajuste na definição da função update_chatbot para aceitar os valores atualizados
-    def update_chatbot(message, chat_history, system_message, max_tokens, temperature, top_p):  # Inclua os novos parâmetros aqui
-        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, temperature, top_p, 40, 1.1)  # Use os parâmetros atualizados aqui
-    
-        formatted_chat_history = []
-        for user_msg, assistant_msg in zip([msg["content"].strip() for msg in chat_history if msg["role"] == "user"],
-                                        [msg["content"].strip() for msg in chat_history if msg["role"] == "assistant"]):
-            if user_msg or assistant_msg:
-                formatted_chat_history.append([user_msg, assistant_msg])
-    
-        return formatted_chat_history, chat_history, ""
-    
-    # Ajuste na chamada do botão submit para incluir os novos inputs
-    submit.click(
-        fn=update_chatbot,
-        inputs=[message, chat_history_state, system_msg, max_tokens, temperature, top_p],  # Inclua os novos inputs aqui
-        outputs=[chatbot, chat_history_state, message]
-    )
-
-
-    clear.click(
-        fn=clear_chat,
-        inputs=[chat_history_state, message],
-        outputs=[chat_history_state, message]
-    )
+            gr.Markdown(description)
+            chatbot = gr.Chatbot()
+            message = 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, visible=False)
+            max_tokens = gr.Slider(20, 1024, label="Max Tokens", step=20, value=1024, interactive=True, visible=False)
+            temperature = gr.Slider(0.0, 1.0, label="Temperature", step=0.1, value=0.2, interactive=True, visible=False)
+            top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.7, interactive=True, visible=False)
+            chat_history_state = gr.State([])
+
+            submit = gr.Button(value="Send message")
+            clear = gr.Button(value="New topic")
+
+            additional_inputs = gr.Accordion("Additional Inputs", open=False)
+            with additional_inputs:
+                gr.Row([system_msg, max_tokens, temperature, top_p])
+
+            submit.click(
+                fn=update_chatbot,
+                inputs=[message, chat_history_state, system_msg, max_tokens, temperature, top_p],
+                outputs=[chatbot, chat_history_state]
+            )
+
+            clear.click(
+                fn=clear_chat,
+                inputs=[chat_history_state],
+                outputs=[chatbot, chat_history_state]
+            )
 
 demo.launch()