Update app.py
Browse files
app.py
CHANGED
|
@@ -99,25 +99,27 @@ with gr.Blocks() as demo:
|
|
| 99 |
top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.95, 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.value, temperature.value, top_p.value, 40, 1.1)
|
| 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: # Verify if either message is not empty
|
| 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],
|
| 121 |
outputs=[chatbot, chat_history_state, message]
|
| 122 |
)
|
| 123 |
|
|
|
|
| 99 |
top_p = gr.Slider(0.0, 1.0, label="Top P", step=0.05, value=0.95, interactive=True)
|
| 100 |
chat_history_state = gr.State([])
|
| 101 |
|
| 102 |
+
# Ajuste na definição da função update_chatbot para aceitar o valor atualizado do system_msg
|
| 103 |
+
def update_chatbot(message, chat_history, system_message): # Remova o valor padrão 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.value, temperature.value, top_p.value, 40, 1.1)
|
| 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: # Verify if either message is not empty
|
| 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 system_msg como um input
|
| 120 |
submit.click(
|
| 121 |
fn=update_chatbot,
|
| 122 |
+
inputs=[message, chat_history_state, system_msg], # Inclua system_msg aqui
|
| 123 |
outputs=[chatbot, chat_history_state, message]
|
| 124 |
)
|
| 125 |
|