Update app.py

app.py

@@ -108,92 +108,120 @@ def index():
 
 @app.route("/download_pdf", methods=["POST"])
 def download_pdf():
-    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))
-
-    studio = [capital * ((1 + studio_ret / 100) ** (12 * ano)) for ano in anos]
-    franquia = [capital + (franquia_ret * ano) for ano in anos]
-    acoes = [capital * ((1 + acoes_ret / 100) ** ano) for ano in anos]
-    renda_fixa = [capital * ((1 + renda_fixa / 100) ** ano) for ano in anos]
-    ajustado_inflacao = [valor / ((1 + inflacao / 100) ** ano) for valor, ano in zip(studio, anos)]
-
+    # Recebe os dados do formulário
+    capital_inicial = float(request.form["capital"])
+    retorno_mensal_studio = float(request.form["studio_ret"])
+    valorizacao_anual = float(request.form["valorizacao"])
+    lucro_anual_franquia = float(request.form["franquia_ret"])
+    retorno_anual_acoes = float(request.form["acoes_ret"])
+    retorno_anual_renda_fixa = float(request.form["renda_fixa"])
+    inflacao_anual = float(request.form["inflacao"])
+
+    anos = list(range(1, 6))  # Simulação de 5 anos
+
+    # Calcula os valores ano a ano para cada investimento
+    valores_studio = [
+        capital_inicial * ((1 + retorno_mensal_studio / 100) ** (12 * ano))
+        for ano in anos
+    ]
+    valores_franquia = [
+        capital_inicial + (lucro_anual_franquia * ano)
+        for ano in anos
+    ]
+    valores_acoes = [
+        capital_inicial * ((1 + retorno_anual_acoes / 100) ** ano)
+        for ano in anos
+    ]
+    valores_renda_fixa = [
+        capital_inicial * ((1 + retorno_anual_renda_fixa / 100) ** ano)
+        for ano in anos
+    ]
+    valores_studio_ajustado_inflacao = [
+        valor / ((1 + inflacao_anual / 100) ** ano)
+        for valor, ano in zip(valores_studio, anos)
+    ]
+
+    # Monta DataFrame para tabela
     dados = {
         "Ano": anos,
-        "Studio": studio,
-        "Franquia": franquia,
-        "Ações": acoes,
-        "Renda Fixa": renda_fixa,
-        "Studio c/ Inflação": ajustado_inflacao,
+        "Studio": valores_studio,
+        "Franquia": valores_franquia,
+        "Ações": valores_acoes,
+        "Renda Fixa": valores_renda_fixa,
+        "Studio c/ Inflação": valores_studio_ajustado_inflacao,
     }
     df = pd.DataFrame(dados)
 
+    # Define investimento com maior retorno final (considerando valores nominais)
     investimentos_finais = {
-        "Studio": studio[-1],
-        "Franquia": franquia[-1],
-        "Ações": acoes[-1],
-        "Renda Fixa": renda_fixa[-1],
+        "Studio": valores_studio[-1],
+        "Franquia": valores_franquia[-1],
+        "Ações": valores_acoes[-1],
+        "Renda Fixa": valores_renda_fixa[-1],
     }
     investimento_mais_valorizado = max(investimentos_finais, key=investimentos_finais.get)
     valor_mais_alto = investimentos_finais[investimento_mais_valorizado]
 
-    # 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")
-    plt.plot(anos, acoes, label="Ações", marker="o")
-    plt.plot(anos, renda_fixa, label="Renda Fixa", marker="o")
-    plt.plot(anos, ajustado_inflacao, label="Studio c/ Inflação", linestyle="--", marker="o")
-    plt.title("Simulação de Investimentos")
+    # Gerar gráfico comparativo (linha)
+    plt.figure(figsize=(8,5))
+    plt.plot(anos, valores_studio, label="Studio")
+    plt.plot(anos, valores_franquia, label="Franquia")
+    plt.plot(anos, valores_acoes, label="Ações")
+    plt.plot(anos, valores_renda_fixa, label="Renda Fixa")
+    plt.plot(anos, valores_studio_ajustado_inflacao, label="Studio c/ Inflação", linestyle='--')
+    plt.title("Comparação dos Investimentos")
     plt.xlabel("Ano")
-    plt.ylabel("Valor Projetado (R$)")
+    plt.ylabel("Valor (R$)")
     plt.legend()
     plt.grid(True)
     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()
+    # Salvar gráfico em base64 para usar no template PDF
+    img_buffer = io.BytesIO()
+    plt.savefig(img_buffer, format='png')
+    img_buffer.seek(0)
+    grafico_base64 = base64.b64encode(img_buffer.getvalue()).decode('utf-8')
     plt.close()
 
-    tabela_html = df.to_html(index=False, classes="", float_format="R${:,.2f}".format)
-
-    # Renderizar template simplificado para PDF
-    rendered_html = render_template(
+    # Criar tabela HTML estilizada para o relatório
+    tabela = df.style.format({
+        "Studio": "R$ {:,.2f}",
+        "Franquia": "R$ {:,.2f}",
+        "Ações": "R$ {:,.2f}",
+        "Renda Fixa": "R$ {:,.2f}",
+        "Studio c/ Infla�o": "R$ {:,.2f}",
+    }).hide_index().set_table_styles([
+        {'selector': 'th', 'props': [('background-color', '#007bff'), ('color', 'white'), ('text-align', 'center')]},
+        {'selector': 'td', 'props': [('text-align', 'right'), ('padding', '6px 8px')]},
+        {'selector': 'table', 'props': [('border-collapse', 'collapse'), ('width', '100%')]},
+        {'selector': 'th, td', 'props': [('border', '1px solid #000')]},
+    ]).render()
+
+    # Renderiza template PDF
+    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,
+        capital=capital_inicial,
+        studio_ret=retorno_mensal_studio,
+        valorizacao=valorizacao_anual,
+        franquia_ret=lucro_anual_franquia,
+        acoes_ret=retorno_anual_acoes,
+        renda_fixa=retorno_anual_renda_fixa,
+        inflacao=inflacao_anual,
+        anos=anos,
+        tabela=tabela,
         grafico=grafico_base64,
-        tabela=tabela_html,
         investimento_mais_valorizado=investimento_mais_valorizado,
         valor_mais_alto=valor_mais_alto,
     )
 
+    # Gera PDF usando xhtml2pdf
     pdf = io.BytesIO()
-    pisa_status = pisa.CreatePDF(rendered_html, dest=pdf)
-    pdf.seek(0)
-
+    pisa_status = pisa.CreatePDF(html, dest=pdf)
     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")
-
-
-
-
+    pdf.seek(0)
+    return send_file(pdf, mimetype="application/pdf", as_attachment=True, download_name="relatorio_simulacao.pdf")