Update app.py

app.py

@@ -31,17 +31,17 @@ def gerar_analise(investimentos_finais, capital, patrimonio_studio_final):
 
 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:
-"""
+    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:
+    """
     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.
-"""
+    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.
+    """
 
     resposta = openai.ChatCompletion.create(
         model="gpt-4o",
@@ -55,26 +55,24 @@ Destaque qual foi o melhor investimento, o retorno percentual sobre o capital, e
 def render_pdf(template_src, context_dict):
     html = render_template(template_src, **context_dict)
     result = io.BytesIO()
-    # Importante: usar io.StringIO para passar a string HTML para pisa
-    pisa_status = pisa.CreatePDF(io.StringIO(html), dest=result)
-    if not pisa_status.err:
-        result.seek(0)
-        return result
-    return None
+    # Passa o HTML diretamente para CreatePDF (mais confiável)
+    pisa_status = pisa.CreatePDF(src=html, dest=result)
+    if pisa_status.err:
+        print(f"[ERRO] Pisa CreatePDF falhou com código: {pisa_status.err}")
+        return None
+    result.seek(0)
+    return result
 
 @app.route("/", methods=["GET", "POST"])
 def index():
     if request.method == "POST":
-        try:
-            capital = float(request.form["capital"])
-            studio_ret = float(request.form["studio_ret"])
-            valorizacao = float(request.form["valorizacao"])
-            franquia_ret = float(request.form["franquia_ret"])
-            acoes_ret = float(request.form["acoes_ret"])
-            renda_fixa = float(request.form["renda_fixa"])
-            inflacao = float(request.form["inflacao"])
-        except (KeyError, ValueError):
-            return render_template("index.html", erro="Por favor, preencha todos os campos corretamente.")
+        capital = float(request.form["capital"])
+        studio_ret = float(request.form["studio_ret"])
+        valorizacao = float(request.form["valorizacao"])
+        franquia_ret = float(request.form["franquia_ret"])
+        acoes_ret = float(request.form["acoes_ret"])
+        renda_fixa = float(request.form["renda_fixa"])
+        inflacao = float(request.form["inflacao"])
 
         anos = list(range(1, 6))
 
@@ -126,7 +124,7 @@ def index():
         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="tabela", border=0)
+        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])
 
@@ -142,16 +140,13 @@ def index():
 
 @app.route("/gerar-pdf", methods=["POST"])
 def gerar_pdf():
-    try:
-        capital = float(request.form["capital"])
-        studio_ret = float(request.form["studio_ret"])
-        valorizacao = float(request.form["valorizacao"])
-        franquia_ret = float(request.form["franquia_ret"])
-        acoes_ret = float(request.form["acoes_ret"])
-        renda_fixa = float(request.form["renda_fixa"])
-        inflacao = float(request.form["inflacao"])
-    except (KeyError, ValueError):
-        return "Erro: dados inválidos.", 400
+    capital = float(request.form["capital"])
+    studio_ret = float(request.form["studio_ret"])
+    valorizacao = float(request.form["valorizacao"])
+    franquia_ret = float(request.form["franquia_ret"])
+    acoes_ret = float(request.form["acoes_ret"])
+    renda_fixa = float(request.form["renda_fixa"])
+    inflacao = float(request.form["inflacao"])
 
     anos = list(range(1, 6))
 
@@ -178,7 +173,6 @@ def gerar_pdf():
         "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")
@@ -191,6 +185,8 @@ def gerar_pdf():
     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.read()).decode("utf-8")
@@ -209,7 +205,7 @@ def gerar_pdf():
         "capital": capital,
         "grafico": grafico_base64,
         "tabela": tabela_html,
-        "analise_final": analise_final_ia.replace("\n", "<br>"),
+        "analise_final": analise_final_ia,
         "data_hoje": datetime.now().strftime("São Paulo, %d de %B de %Y")
     }