Update app.py

app.py

@@ -103,10 +103,11 @@ def index():
 
     return render_template("index.html")
 
+
+
+
 @app.route("/download_pdf", methods=["POST"])
 def download_pdf():
-    from flask import render_template_string
-
     capital = float(request.form["capital"])
     studio_ret = float(request.form["studio_ret"])
     valorizacao = float(request.form["valorizacao"])
@@ -142,7 +143,7 @@ def download_pdf():
     investimento_mais_valorizado = max(investimentos_finais, key=investimentos_finais.get)
     valor_mais_alto = investimentos_finais[investimento_mais_valorizado]
 
-    # Gráfico
+    # Gerar gráfico (base64)
     plt.figure(figsize=(10, 6))
     plt.plot(anos, studio, label="Studio", marker="o")
     plt.plot(anos, franquia, label="Franquia", marker="o")
@@ -163,31 +164,35 @@ def download_pdf():
     buf.close()
     plt.close()
 
-    tabela_html = df.to_html(index=False, classes="table table-striped", float_format="R${:,.2f}".format)
-
-    rendered_html = render_template("index.html",
-                                    capital=capital,
-                                    studio_ret=studio_ret,
-                                    valorizacao=valorizacao,
-                                    franquia_ret=franquia_ret,
-                                    acoes_ret=acoes_ret,
-                                    renda_fixa=renda_fixa,
-                                    inflacao=inflacao,
-                                    grafico=grafico_base64,
-                                    tabela=tabela_html,
-                                    investimento_mais_valorizado=investimento_mais_valorizado,
-                                    valor_mais_alto=valor_mais_alto,
-                                    gerar_pdf=True)
+    tabela_html = df.to_html(index=False, classes="", float_format="R${:,.2f}".format)
+
+    # Renderizar template simplificado para PDF
+    rendered_html = render_template(
+        "relatorio_pdf.html",
+        capital=capital,
+        studio_ret=studio_ret,
+        valorizacao=valorizacao,
+        franquia_ret=franquia_ret,
+        acoes_ret=acoes_ret,
+        renda_fixa=renda_fixa,
+        inflacao=inflacao,
+        grafico=grafico_base64,
+        tabela=tabela_html,
+        investimento_mais_valorizado=investimento_mais_valorizado,
+        valor_mais_alto=valor_mais_alto,
+    )
 
     pdf = io.BytesIO()
-    pisa.CreatePDF(io.StringIO(rendered_html), dest=pdf)
+    pisa_status = pisa.CreatePDF(rendered_html, dest=pdf)
     pdf.seek(0)
+
+    if pisa_status.err:
+        return "Erro ao gerar PDF", 500
+
     return send_file(pdf, as_attachment=True, download_name="relatorio_simulacao.pdf", mimetype="application/pdf")
 
 
 
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=True)
 
 
 
@@ -196,3 +201,7 @@ if __name__ == '__main__':
 
 
 
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True)
+