Update app.py
Browse files
app.py
CHANGED
@@ -99,28 +99,31 @@ with gr.Blocks() as demo:
|
|
99 |
top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.7, interactive=True)
|
100 |
chat_history_state = gr.State([])
|
101 |
|
102 |
-
|
|
|
103 |
print("Updating chatbot...")
|
104 |
if not chat_history or (chat_history and chat_history[-1]["role"] != "user"):
|
105 |
chat_history = user(message, chat_history, system_message if not chat_history else None)
|
106 |
else:
|
107 |
chat_history = user(message, chat_history)
|
108 |
-
chat_history, _ = chat(chat_history, system_message, max_tokens
|
109 |
|
110 |
formatted_chat_history = []
|
111 |
for user_msg, assistant_msg in zip([msg["content"].strip() for msg in chat_history if msg["role"] == "user"],
|
112 |
[msg["content"].strip() for msg in chat_history if msg["role"] == "assistant"]):
|
113 |
-
if user_msg or assistant_msg:
|
114 |
formatted_chat_history.append([user_msg, assistant_msg])
|
115 |
|
116 |
return formatted_chat_history, chat_history, ""
|
117 |
-
|
|
|
118 |
submit.click(
|
119 |
fn=update_chatbot,
|
120 |
-
inputs=[message, chat_history_state, system_msg],
|
121 |
outputs=[chatbot, chat_history_state, message]
|
122 |
)
|
123 |
|
|
|
124 |
clear.click(
|
125 |
fn=clear_chat,
|
126 |
inputs=[chat_history_state, message],
|
|
|
99 |
top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.7, interactive=True)
|
100 |
chat_history_state = gr.State([])
|
101 |
|
102 |
+
# Ajuste na definição da função update_chatbot para aceitar os valores atualizados
|
103 |
+
def update_chatbot(message, chat_history, system_message, max_tokens, temperature, top_p): # Inclua os novos parâmetros aqui
|
104 |
print("Updating chatbot...")
|
105 |
if not chat_history or (chat_history and chat_history[-1]["role"] != "user"):
|
106 |
chat_history = user(message, chat_history, system_message if not chat_history else None)
|
107 |
else:
|
108 |
chat_history = user(message, chat_history)
|
109 |
+
chat_history, _ = chat(chat_history, system_message, max_tokens, temperature, top_p, 40, 1.1) # Use os parâmetros atualizados aqui
|
110 |
|
111 |
formatted_chat_history = []
|
112 |
for user_msg, assistant_msg in zip([msg["content"].strip() for msg in chat_history if msg["role"] == "user"],
|
113 |
[msg["content"].strip() for msg in chat_history if msg["role"] == "assistant"]):
|
114 |
+
if user_msg or assistant_msg:
|
115 |
formatted_chat_history.append([user_msg, assistant_msg])
|
116 |
|
117 |
return formatted_chat_history, chat_history, ""
|
118 |
+
|
119 |
+
# Ajuste na chamada do botão submit para incluir os novos inputs
|
120 |
submit.click(
|
121 |
fn=update_chatbot,
|
122 |
+
inputs=[message, chat_history_state, system_msg, max_tokens, temperature, top_p], # Inclua os novos inputs aqui
|
123 |
outputs=[chatbot, chat_history_state, message]
|
124 |
)
|
125 |
|
126 |
+
|
127 |
clear.click(
|
128 |
fn=clear_chat,
|
129 |
inputs=[chat_history_state, message],
|