Update app.py

app.py

@@ -15,21 +15,15 @@ from babel.numbers import format_currency
 
 app = Flask(__name__)
 
-# Configuração do logger
+# Logger
 logging.basicConfig(level=logging.DEBUG)
 
-# Configure sua chave da OpenAI via variável de ambiente
+# OpenAI key da variável de ambiente
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
-# Função para formatar valores monetários em BRL
 def formatar_brl(valor):
-    try:
-        return format_currency(valor, "BRL", locale="pt_BR")
-    except Exception:
-        logging.warning(f"Erro ao formatar valor monetário: {valor}")
-        return f"R$ {valor:,.2f}"
+    return format_currency(valor, "BRL", locale="pt_BR")
 
-# Função para converter para float com log e valor default
 def safe_float(valor, nome, default=0.0):
     try:
         f = float(valor)
@@ -39,7 +33,6 @@ def safe_float(valor, nome, default=0.0):
         logging.error(f"Erro convertendo {nome} com valor '{valor}': {e}")
         return default
 
-# Função que calcula projeções para os 5 anos
 def calcular_projecoes(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa):
     anos = list(range(1, 6))
     patrimonio_studio = [capital * ((1 + valorizacao / 100) ** ano) for ano in anos]
@@ -65,7 +58,6 @@ def calcular_projecoes(capital, studio_ret, valorizacao, franquia_ret, acoes_ret
     }
     return dados, investimentos_finais
 
-# Função para gerar gráfico base64
 def gerar_grafico(anos, studio_total, franquia, acoes, renda_fixa_valores):
     plt.figure(figsize=(8, 5))
     plt.plot(anos, studio_total, label="Studio (Patrimônio + Renda)", marker="o")
@@ -87,7 +79,6 @@ def gerar_grafico(anos, studio_total, franquia, acoes, renda_fixa_valores):
     grafico_base64 = base64.b64encode(buf.getvalue()).decode("utf-8")
     return grafico_base64
 
-# Função que gera o relatório com a API OpenAI
 def gerar_analise_ia(investimentos_finais, capital, patrimonio_studio_final):
     prompt = f"""
 Você é um analista financeiro experiente, e está redigindo um relatório para um cliente com capital inicial de R$ {capital:,.2f}. Após 5 anos, os resultados dos investimentos foram:
@@ -104,23 +95,18 @@ Escreva um relatório claro, didático e consultivo, com no máximo 10 linhas, c
 5. Sugestão de diversificação balanceada para perfil moderado.
 Use parágrafos e destaque em negrito os títulos dos tópicos. Não use markdown, apenas HTML.
 """
-    try:
-        resposta = openai.ChatCompletion.create(
-            model="gpt-4o",
-            messages=[
-                {"role": "system", "content": "Você é um analista financeiro experiente."},
-                {"role": "user", "content": prompt}
-            ],
-            max_tokens=500,
-            temperature=0.7
-        )
-        texto = resposta.choices[0].message.content
-        return texto
-    except Exception as e:
-        logging.error(f"Erro ao chamar API OpenAI: {e}")
-        return "<p>Erro ao gerar análise com IA.</p>"
+    resposta = openai.ChatCompletion.create(
+        model="gpt-4o",
+        messages=[
+            {"role": "system", "content": "Você é um analista financeiro experiente."},
+            {"role": "user", "content": prompt}
+        ],
+        max_tokens=500,
+        temperature=0.7
+    )
+    texto = resposta.choices[0].message.content
+    return texto
 
-# Função para gerar PDF a partir de template HTML
 def render_pdf(template_src, context_dict):
     html = render_template(template_src, **context_dict)
     result = io.BytesIO()
@@ -131,7 +117,6 @@ def render_pdf(template_src, context_dict):
     logging.error(f"Erro pisa.CreatePDF: {pisa_status.err}")
     return None
 
-# Rota principal com formulário
 @app.route("/", methods=["GET", "POST"])
 def index():
     if request.method == "POST":
@@ -176,72 +161,51 @@ def index():
             logging.error("Erro na rota /", exc_info=True)
             return "Erro interno no servidor. Verifique os logs.", 500
 
-    # GET simples
     return render_template("index.html")
 
-# Rota para gerar PDF com relatório
-@app.route("/gerar-pdf", methods=["POST"])
-def gerar_pdf():
+@app.route("/relatorio", methods=["POST"])
+def relatorio():
     try:
-        campos = ["capital", "studio_ret", "valorizacao", "franquia_ret", "acoes_ret", "renda_fixa", "inflacao"]
-        dados = {campo: safe_float(request.form.get(campo), campo) for campo in campos}
-
-        capital = dados["capital"]
-        anos = list(range(1, 6))
-
-        patrimonio_studio = [capital * ((1 + dados["valorizacao"] / 100) ** ano) for ano in anos]
-        renda_acumulada_studio = [capital * (((1 + dados["studio_ret"] / 100) ** (12 * ano)) - 1) for ano in anos]
-        studio_total = [p + r for p, r in zip(patrimonio_studio, renda_acumulada_studio)]
-        patrimonio_final = studio_total[-1]
-
-        franquia = [capital + (dados["franquia_ret"] * ano) for ano in anos]
-        acoes = [capital * ((1 + dados["acoes_ret"] / 100) ** ano) for ano in anos]
-        renda_fixa_valores = [capital * ((1 + dados["renda_fixa"] / 100) ** ano) for ano in anos]
-
-        df = pd.DataFrame({
-            "Ano": anos,
-            "Studio (Patrimônio + Renda)": studio_total,
-            "Franquia": franquia,
-            "Ações": acoes,
-            "Renda Fixa": renda_fixa_valores
-        })
-
-        investimentos_finais = {
-            "Studio": studio_total[-1],
-            "Franquia": franquia[-1],
-            "Ações": acoes[-1],
-            "Renda Fixa": renda_fixa_valores[-1]
-        }
+        logging.debug(f"Dados recebidos para PDF: {request.form}")
 
-        grafico_base64 = gerar_grafico(anos, studio_total, franquia, acoes, renda_fixa_valores)
+        capital = safe_float(request.form.get("capital"), "capital")
+        studio_ret = safe_float(request.form.get("studio_ret"), "studio_ret")
+        valorizacao = safe_float(request.form.get("valorizacao"), "valorizacao")
+        franquia_ret = safe_float(request.form.get("franquia_ret"), "franquia_ret")
+        acoes_ret = safe_float(request.form.get("acoes_ret"), "acoes_ret")
+        renda_fixa = safe_float(request.form.get("renda_fixa"), "renda_fixa")
+        inflacao = safe_float(request.form.get("inflacao"), "inflacao")
+
+        dados, investimentos_finais = calcular_projecoes(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa)
 
+        anos = dados["Ano"]
+        studio_total = dados["Studio (Patrimônio + Renda)"]
+        franquia = dados["Franquia"]
+        acoes = dados["Ações"]
+        renda_fixa_valores = dados["Renda Fixa"]
+
+        df = pd.DataFrame(dados)
         df_formatado = df.copy()
         for col in df.columns:
             if col != "Ano":
                 df_formatado[col] = df_formatado[col].apply(formatar_brl)
-        tabela_html = df_formatado.to_html(index=False, classes="tabela", border=0)
+        tabela_html = df_formatado.to_html(index=False, classes="table table-striped table-sm text-end", border=0)
+
+        grafico_base64 = gerar_grafico(anos, studio_total, franquia, acoes, renda_fixa_valores)
 
-        analise_final = gerar_analise_ia(investimentos_finais, capital, patrimonio_final)
+        analise_final = gerar_analise_ia(investimentos_finais, capital, studio_total[-1])
 
         data_formatada = format_date(datetime.now(), "d 'de' MMMM 'de' yyyy", locale="pt_BR")
         data_hoje = f"São Paulo, {data_formatada}"
 
-        capital_formatado = formatar_brl(capital)
-        studio_ret_formatado = f"{dados['studio_ret']:.2f} %"
-        valorizacao_formatado = f"{dados['valorizacao']:.2f} %"
-        franquia_ret_formatado = formatar_brl(dados["franquia_ret"])
-        acoes_ret_formatado = f"{dados['acoes_ret']:.2f} %"
-        renda_fixa_formatado = f"{dados['renda_fixa']:.2f} %"
-        inflacao_formatado = f"{dados['inflacao']:.2f} %"
-
         context = {
-            "capital_formatado": capital_formatado,
-            "studio_ret_formatado": studio_ret_formatado,
-            "valorizacao_formatado": valorizacao_formatado,
-            "franquia_ret_formatado": franquia_ret_formatado,
-            "acoes_ret_formatado": acoes_ret_formatado,
-            "renda_fixa_formatado": renda_fixa_formatado,
-            "inflacao_formatado": inflacao_formatado,
+            "capital_formatado": formatar_brl(capital),
+            "studio_ret_formatado": f"{studio_ret:.2f} %",
+            "valorizacao_formatado": f"{valorizacao:.2f} %",
+            "franquia_ret_formatado": formatar_brl(franquia_ret),
+            "acoes_ret_formatado": f"{acoes_ret:.2f} %",
+            "renda_fixa_formatado": f"{renda_fixa:.2f} %",
+            "inflacao_formatado": f"{inflacao:.2f} %",
             "grafico": grafico_base64,
             "tabela": tabela_html,
             "analise_final": analise_final,
@@ -254,9 +218,9 @@ def gerar_pdf():
         else:
             return "Erro ao gerar o PDF", 500
 
-    except Exception as e:
-        logging.error("Erro na rota /gerar-pdf", exc_info=True)
-        return f"Erro interno ao gerar o PDF:\n{str(e)}", 500
+    except Exception:
+        logging.error("Erro na rota /relatorio", exc_info=True)
+        return "Erro interno ao gerar o PDF. Verifique os logs.", 500
 
 
 if __name__ == "__main__":