Update app.py

app.py

@@ -8,17 +8,17 @@ from matplotlib.ticker import FuncFormatter
 from xhtml2pdf import pisa
 from datetime import datetime
 import openai
-import traceback
 import logging
 
 app = Flask(__name__)
 
-# Configurar logger para console
+# Configuração de logger
 logging.basicConfig(level=logging.DEBUG)
 
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
 def formatar_brl(valor):
+    """Formata número float em string moeda BRL."""
     return f"R$ {valor:,.2f}".replace(",", "X").replace(".", ",").replace("X", ".")
 
 def gerar_analise(investimentos_finais, capital, patrimonio_studio_final):
@@ -35,25 +35,29 @@ def gerar_analise(investimentos_finais, capital, patrimonio_studio_final):
     return texto
 
 def gerar_analise_ia(investimentos_finais, capital, patrimonio_studio_final):
-    prompt = f"""
-    Faça uma análise profissional e detalhada em linguagem clara e acessível sobre os seguintes dados financeiros:
-    - Capital inicial: R$ {capital:,.2f}
-    - Valor patrimonial final do Studio: R$ {patrimonio_studio_final:,.2f}
-    - Investimentos finais após 5 anos:
-    """
+    prompt = (
+        f"Faça uma análise profissional e detalhada em linguagem clara e acessível sobre os seguintes dados financeiros:\n\n"
+        f"<strong>Capital inicial:</strong> R$ {capital:,.2f}\n"
+        f"<strong>Valor patrimonial final do Studio:</strong> R$ {patrimonio_studio_final:,.2f}\n"
+        f"<strong>Investimentos finais após 5 anos:</strong>\n"
+    )
     for nome, valor in investimentos_finais.items():
-        prompt += f"    - {nome}: R$ {valor:,.2f}\n"
+        prompt += f"- {nome}: R$ {valor:,.2f}\n"
 
-    prompt += """
-    Destaque qual foi o melhor investimento, o retorno percentual sobre o capital, e comente sobre possíveis estratégias e perfis de risco. Seja objetivo, claro e com tom consultivo.
-    """
+    prompt += (
+        "\nDestaque usando <strong>negrito</strong> qual foi o melhor investimento e o retorno percentual sobre o capital. "
+        "Comente também sobre possíveis estratégias e perfis de risco. Use quebras de linha para organizar as ideias. "
+        "Seja objetivo, claro e com tom consultivo."
+    )
 
     resposta = openai.ChatCompletion.create(
         model="gpt-4o",
         messages=[
             {"role": "system", "content": "Você é um analista financeiro experiente."},
             {"role": "user", "content": prompt}
-        ]
+        ],
+        temperature=0.7,
+        max_tokens=400  # reduzido para melhorar tempo de resposta
     )
     return resposta.choices[0].message.content
 
@@ -64,6 +68,7 @@ def render_pdf(template_src, context_dict):
     if not pisa_status.err:
         result.seek(0)
         return result
+    logging.error(f"Erro pisa.CreatePDF: {pisa_status.err}")
     return None
 
 def safe_float(valor, nome, default=0.0):
@@ -75,11 +80,58 @@ def safe_float(valor, nome, default=0.0):
         logging.error(f"Erro convertendo {nome} com valor '{valor}': {e}")
         return default
 
+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]
+    renda_acumulada_studio = [capital * (((1 + studio_ret / 100) ** (12 * ano)) - 1) for ano in anos]
+    studio_total = [p + r for p, r in zip(patrimonio_studio, renda_acumulada_studio)]
+
+    franquia = [capital + (franquia_ret * ano) for ano in anos]
+    acoes = [capital * ((1 + acoes_ret / 100) ** ano) for ano in anos]
+    renda_fixa_valores = [capital * ((1 + renda_fixa / 100) ** ano) for ano in anos]
+
+    dados = {
+        "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],
+    }
+    return dados, investimentos_finais
+
+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")
+    plt.plot(anos, franquia, label="Franquia", marker="o")
+    plt.plot(anos, acoes, label="Ações", marker="o")
+    plt.plot(anos, renda_fixa_valores, label="Renda Fixa", marker="o")
+    plt.title("Projeção de Investimentos (5 anos)")
+    plt.xlabel("Ano")
+    plt.ylabel("Valor (R$)")
+    plt.legend()
+    plt.grid(True)
+    plt.gca().yaxis.set_major_formatter(FuncFormatter(lambda x, _: f"R${x:,.0f}".replace(",", ".")))
+    plt.tight_layout()
+
+    buf = io.BytesIO()
+    plt.savefig(buf, format="png")
+    buf.seek(0)
+    grafico_base64 = base64.b64encode(buf.getvalue()).decode("utf-8")
+    buf.close()
+    plt.close()
+    return grafico_base64
+
 @app.route("/", methods=["GET", "POST"])
 def index():
     if request.method == "POST":
         try:
-            # Log dos dados recebidos
             logging.debug(f"Dados recebidos no POST: {request.form}")
 
             capital = safe_float(request.form.get("capital"), "capital")
@@ -90,59 +142,23 @@ def index():
             renda_fixa = safe_float(request.form.get("renda_fixa"), "renda_fixa")
             inflacao = safe_float(request.form.get("inflacao"), "inflacao")
 
-            anos = list(range(1, 6))
+            dados, investimentos_finais = calcular_projecoes(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa)
 
-            patrimonio_studio = [capital * ((1 + valorizacao / 100) ** ano) for ano in anos]
-            renda_acumulada_studio = [capital * (((1 + studio_ret / 100) ** (12 * ano)) - 1) for ano in anos]
-            studio_total = [p + r for p, r in zip(patrimonio_studio, renda_acumulada_studio)]
-
-            franquia = [capital + (franquia_ret * ano) for ano in anos]
-            acoes = [capital * ((1 + acoes_ret / 100) ** ano) for ano in anos]
-            renda_fixa_valores = [capital * ((1 + renda_fixa / 100) ** ano) for ano in anos]
-
-            dados = {
-                "Ano": anos,
-                "Studio (Patrimônio + Renda)": studio_total,
-                "Franquia": franquia,
-                "Ações": acoes,
-                "Renda Fixa": renda_fixa_valores
-            }
             df = pd.DataFrame(dados)
-
-            investimentos_finais = {
-                "Studio": studio_total[-1],
-                "Franquia": franquia[-1],
-                "Ações": acoes[-1],
-                "Renda Fixa": renda_fixa_valores[-1],
-            }
-
-            plt.figure(figsize=(8, 5))
-            plt.plot(anos, studio_total, label="Studio (Patrimônio + Renda)", marker="o")
-            plt.plot(anos, franquia, label="Franquia", marker="o")
-            plt.plot(anos, acoes, label="Ações", marker="o")
-            plt.plot(anos, renda_fixa_valores, label="Renda Fixa", marker="o")
-            plt.title("Projeção de Investimentos (5 anos)")
-            plt.xlabel("Ano")
-            plt.ylabel("Valor (R$)")
-            plt.legend()
-            plt.grid(True)
-            plt.gca().yaxis.set_major_formatter(FuncFormatter(lambda x, _: f"R${x:,.0f}".replace(",", ".")))
-            plt.tight_layout()
-
-            buf = io.BytesIO()
-            plt.savefig(buf, format="png")
-            buf.seek(0)
-            grafico_base64 = base64.b64encode(buf.getvalue()).decode("utf-8")
-            buf.close()
-            plt.close()
-
             df_formatado = df.copy()
             for col in df.columns:
                 if col != "Ano":
                     df_formatado[col] = df_formatado[col].apply(formatar_brl)
             tabela = df_formatado.to_html(index=False, classes="table table-striped table-sm text-end", border=0)
 
-            analise_final = gerar_analise(investimentos_finais, capital, patrimonio_studio[-1])
+            anos = dados["Ano"]
+            studio_total = dados["Studio (Patrimônio + Renda)"]
+            franquia = dados["Franquia"]
+            acoes = dados["Ações"]
+            renda_fixa_valores = dados["Renda Fixa"]
+
+            grafico_base64 = gerar_grafico(anos, studio_total, franquia, acoes, renda_fixa_valores)
+            analise_final = gerar_analise(investimentos_finais, capital, studio_total[-1])
 
             return render_template(
                 "index.html",
@@ -171,56 +187,27 @@ def gerar_pdf():
         renda_fixa = safe_float(request.form.get("renda_fixa"), "renda_fixa")
         inflacao = safe_float(request.form.get("inflacao"), "inflacao")
 
-        anos = list(range(1, 6))
-
-        patrimonio_studio = [capital * ((1 + valorizacao / 100) ** ano) for ano in anos]
-        renda_acumulada_studio = [capital * (((1 + studio_ret / 100) ** (12 * ano)) - 1) for ano in anos]
-        studio_total = [p + r for p, r in zip(patrimonio_studio, renda_acumulada_studio)]
-        franquia = [capital + (franquia_ret * ano) for ano in anos]
-        acoes = [capital * ((1 + acoes_ret / 100) ** ano) for ano in anos]
-        renda_fixa_valores = [capital * ((1 + renda_fixa / 100) ** ano) for ano in anos]
-
-        dados = {
-            "Ano": anos,
-            "Studio (Patrimônio + Renda)": studio_total,
-            "Franquia": franquia,
-            "Ações": acoes,
-            "Renda Fixa": renda_fixa_valores
-        }
-        df = pd.DataFrame(dados)
-
-        investimentos_finais = {
-            "Studio": studio_total[-1],
-            "Franquia": franquia[-1],
-            "Ações": acoes[-1],
-            "Renda Fixa": renda_fixa_valores[-1],
-        }
-
-        buf = io.BytesIO()
-        plt.figure(figsize=(8, 5))
-        plt.plot(anos, studio_total, label="Studio (Patrimônio + Renda)", marker="o")
-        plt.plot(anos, franquia, label="Franquia", marker="o")
-        plt.plot(anos, acoes, label="Ações", marker="o")
-        plt.plot(anos, renda_fixa_valores, label="Renda Fixa", marker="o")
-        plt.title("Projeção de Investimentos (5 anos)")
-        plt.xlabel("Ano")
-        plt.ylabel("Valor (R$)")
-        plt.legend()
-        plt.grid(True)
-        plt.tight_layout()
-        plt.savefig(buf, format="png")
-        buf.seek(0)
-        grafico_base64 = base64.b64encode(buf.read()).decode("utf-8")
-        buf.close()
-        plt.close()
+        dados, investimentos_finais = calcular_projecoes(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, 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)
 
-        analise_final_ia = gerar_analise_ia(investimentos_finais, capital, patrimonio_studio[-1])
+        anos = dados["Ano"]
+        studio_total = dados["Studio (Patrimônio + Renda)"]
+
+        grafico_base64 = gerar_grafico(
+            anos,
+            studio_total,
+            dados["Franquia"],
+            dados["Ações"],
+            dados["Renda Fixa"],
+        )
+
+        analise_final_ia = gerar_analise_ia(investimentos_finais, capital, studio_total[-1])
 
         context = {
             "capital": capital,