Update app.py

app.py

@@ -2,115 +2,187 @@
 import os
 os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
 os.makedirs('/tmp/fontconfig', exist_ok=True)
-from flask import Flask, request, render_template, send_file
-import io
-from xhtml2pdf import pisa
-import pandas as pd
+from flask import Flask, render_template, request, send_file
 import matplotlib.pyplot as plt
+import pandas as pd
+import io
 import base64
+from matplotlib.ticker import FuncFormatter
+from xhtml2pdf import pisa
+from flask import make_response
+from jinja2 import Template
 
 app = Flask(__name__)
 
-@app.route('/', methods=['GET', 'POST'])
+@app.route("/", methods=["GET", "POST"])
 def index():
-    if request.method == 'POST':
-        # Extrair dados do formulário
-        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'])
-
-        # Simulação fictícia
-        anos = list(range(1, 6))
-        resultados = {
-            "Studio": [capital * (1 + studio_ret/100)**(12 * ano/12) for ano in anos],
-            "Imóvel": [capital * (1 + valorizacao/100)**ano for ano in anos],
-            "Franquia": [capital + franquia_ret * ano for ano in anos],
-            "Ações": [capital * (1 + acoes_ret/100)**ano for ano in anos],
-            "Renda Fixa": [capital * (1 + renda_fixa/100)**ano for ano in anos]
-        }
+    if request.method == "POST":
+        # Recebe dados do formulário
+        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"])
 
-        df = pd.DataFrame(resultados, index=[f'{ano}º ano' for ano in anos])
-        tabela = df.to_html(classes='table', float_format='R$ {:,.2f}'.format)
+        anos = list(range(1, 6))
 
-        investimento_mais_valorizado = df.iloc[-1].idxmax()
-        valor_mais_alto = df.iloc[-1].max()
+        # Cálculos
+        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)]
+
+        # Tabela com pandas
+        dados = {
+            "Ano": anos,
+            "Studio": studio,
+            "Franquia": franquia,
+            "Ações": acoes,
+            "Renda Fixa": renda_fixa,
+            "Studio c/ Inflação": ajustado_inflacao,
+        }
+        df = pd.DataFrame(dados)
+
+        # Maior retorno
+        investimentos_finais = {
+            "Studio": studio[-1],
+            "Franquia": franquia[-1],
+            "Ações": acoes[-1],
+            "Renda Fixa": renda_fixa[-1],
+        }
+        investimento_mais_valorizado = max(investimentos_finais, key=investimentos_finais.get)
+        valor_mais_alto = investimentos_finais[investimento_mais_valorizado]
+
+        # Gerar gráfico
+        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")
+        plt.xlabel("Ano")
+        plt.ylabel("Valor Projetado (R$)")
+        plt.legend()
+        plt.grid(True)
+
+        def formatar_moeda(x, _):
+            return f"R${x:,.0f}".replace(",", ".")
+
+        plt.gca().yaxis.set_major_formatter(FuncFormatter(formatar_moeda))
+        plt.tight_layout()
 
-        # Gráfico
-        fig, ax = plt.subplots()
-        df.plot(ax=ax)
-        ax.set_ylabel("Valor (R$)")
-        ax.set_title("Comparativo de Investimentos")
+        buf = io.BytesIO()
+        plt.savefig(buf, format="png")
+        buf.seek(0)
+        grafico_base64 = base64.b64encode(buf.getvalue()).decode("utf-8")
+        buf.close()
+        plt.close()
+
+        # Tabela formatada para HTML
+        tabela_html = df.to_html(index=False, classes="table table-striped", float_format="R${:,.2f}".format)
+
+        return 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=False
+        )
 
-        img = io.BytesIO()
-        plt.tight_layout()
-        plt.savefig(img, format='png')
-        img.seek(0)
-        grafico = base64.b64encode(img.getvalue()).decode()
-
-        return render_template("index.html",
-                               tabela=tabela,
-                               grafico=grafico,
-                               investimento_mais_valorizado=investimento_mais_valorizado,
-                               valor_mais_alto=valor_mais_alto,
-                               capital=capital,
-                               studio_ret=studio_ret,
-                               valorizacao=valorizacao,
-                               franquia_ret=franquia_ret,
-                               acoes_ret=acoes_ret,
-                               renda_fixa=renda_fixa,
-                               inflacao=inflacao,
-                               gerar_pdf=False)
     return render_template("index.html")
 
-@app.route('/download_pdf', methods=['POST'])
+@app.route("/download_pdf", methods=["POST"])
 def download_pdf():
-    # Reutilize os mesmos dados do formulário para refazer a simulação
-    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'])
+    from flask import render_template_string
+
+    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))
-    resultados = {
-        "Studio": [capital * (1 + studio_ret/100)**(12 * ano/12) for ano in anos],
-        "Imóvel": [capital * (1 + valorizacao/100)**ano for ano in anos],
-        "Franquia": [capital + franquia_ret * ano for ano in anos],
-        "Ações": [capital * (1 + acoes_ret/100)**ano for ano in anos],
-        "Renda Fixa": [capital * (1 + renda_fixa/100)**ano for ano in anos]
+
+    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)]
+
+    dados = {
+        "Ano": anos,
+        "Studio": studio,
+        "Franquia": franquia,
+        "Ações": acoes,
+        "Renda Fixa": renda_fixa,
+        "Studio c/ Inflação": ajustado_inflacao,
     }
+    df = pd.DataFrame(dados)
 
-    df = pd.DataFrame(resultados, index=[f'{ano}º ano' for ano in anos])
-    tabela = df.to_html(classes='table', float_format='R$ {:,.2f}'.format)
-
-    investimento_mais_valorizado = df.iloc[-1].idxmax()
-    valor_mais_alto = df.iloc[-1].max()
-
-    html = render_template("index.html",
-                           tabela=tabela,
-                           grafico="",
-                           investimento_mais_valorizado=investimento_mais_valorizado,
-                           valor_mais_alto=valor_mais_alto,
-                           capital=capital,
-                           studio_ret=studio_ret,
-                           valorizacao=valorizacao,
-                           franquia_ret=franquia_ret,
-                           acoes_ret=acoes_ret,
-                           renda_fixa=renda_fixa,
-                           inflacao=inflacao,
-                           gerar_pdf=True)
+    investimentos_finais = {
+        "Studio": studio[-1],
+        "Franquia": franquia[-1],
+        "Ações": acoes[-1],
+        "Renda Fixa": renda_fixa[-1],
+    }
+    investimento_mais_valorizado = max(investimentos_finais, key=investimentos_finais.get)
+    valor_mais_alto = investimentos_finais[investimento_mais_valorizado]
+
+    # Gráfico
+    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")
+    plt.xlabel("Ano")
+    plt.ylabel("Valor Projetado (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()
+    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)
 
     pdf = io.BytesIO()
-    pisa.CreatePDF(io.StringIO(html), dest=pdf)
+    pisa.CreatePDF(io.StringIO(rendered_html), dest=pdf)
     pdf.seek(0)
-
-    return send_file(pdf, download_name="relatorio_simulacao.pdf", as_attachment=True)
+    return send_file(pdf, as_attachment=True, download_name="relatorio_simulacao.pdf", mimetype="application/pdf")