conversacional_invest / chatbot_respostas.py
Persano's picture
Create chatbot_respostas.py
d602bc7 verified
import pandas as pd
from modelo_huggingface import modelo
# Função que gera resposta com base na pergunta
def gerar_resposta(pergunta):
# Aqui você pode colocar lógica para verificar os tópicos e escolher qual resposta gerar.
# Se a pergunta for sobre financiamento:
if "financiamento" in pergunta.lower():
return "O financiamento imobiliário pode ser feito de diferentes maneiras, dependendo do seu perfil de crédito."
# Se for sobre ROI:
elif "ROI" in pergunta.lower():
return calcular_roi()
# Caso contrário, utilizar o modelo do Hugging Face para gerar uma resposta geral
return modelo(pergunta)
def calcular_roi():
# Carregar dados financeiros para calcular ROI
tabela_roi = pd.read_csv('dados/tabelas_roi.csv')
# Exemplo de cálculo (aqui você pode expandir para algo mais detalhado)
roi = tabela_roi['lucro'].mean() / tabela_roi['investimento'].mean()
return f"O ROI médio do seu investimento seria {roi:.2f}."