Update app.py

app.py

@@ -19,7 +19,7 @@ def configurar_sessao(modelo):
     if 'modelo_atual' not in st.session_state or st.session_state['modelo_atual'] != modelo:
         st.session_state['modelo_atual'] = modelo
         st.session_state['mensagens'] = [
-            {"role": "system", "content": "Você é um assistente de investimentos inteligente, treinado para ajudar os usuários a gerenciar e otimizar seus portfólios financeiros."
+            {"role": "system", "content": "Você é um bot para investimentos, treinado para ajudar os usuários a gerenciar e otimizar seus portfólios financeiros."
                                           "Seu objetivo é fornecer análises de mercado, sugestões de investimentos personalizadas e educar os usuários sobre estratégias financeiras."
                                           "Você pode responder a perguntas sobre ações, fundos, criptomoedas, imóveis e outros tipos de investimentos."
                                           "Além disso, você oferece insights sobre tendências de mercado e ajuda os usuários a tomar decisões informadas com base no perfil de risco deles."},
@@ -72,10 +72,11 @@ def gerar_resposta(label, opcoes_classificacao):
         )
     else:
         novo_pront = (
-            "Esse é o acessor de investimentos.\n"
+            "Esse é seu acessor de investimentos.\n"
 			"- Você possui o acessor: \n"
 			"- João da Silva \n"
 			"- Tel: (11) 9 1111-1111 \n"
+			"- e-mail: [email protected] \n"
 			"- Empresa: Investimentos-SA \n"
         )
     
@@ -83,7 +84,7 @@ def gerar_resposta(label, opcoes_classificacao):
     return novo_pront
 
 # Função para processar a conversa e exibir no Streamlit
-def processar_conversa(pergunta_usuario, url, headers, tokenizer, mensagens, token_modelo):
+def processar_conversa(url, headers, tokenizer, mensagens, token_modelo):
     template = tokenizer.apply_chat_template(mensagens, tokenize=False, add_generation_prompt=True)
     json = {
         'inputs': template,
@@ -135,7 +136,7 @@ def main():
             mensagens.append({"role": "system", "content": novo_pront})
         
         mensagens.append({'role': 'user', 'content': pergunta_usuario})
-        mensagens = processar_conversa(pergunta_usuario, url, headers, tokenizer, mensagens, token_modelo)
+        mensagens = processar_conversa(url, headers, tokenizer, mensagens, token_modelo)
         exibir_mensagens(area_chat, mensagens)
     
     print(mensagens)