Update app.py

app.py

@@ -2,88 +2,19 @@
 import os
 os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
 os.makedirs('/tmp/fontconfig', exist_ok=True)
-
-from flask import Flask, render_template, request, send_file, make_response
-import matplotlib.pyplot as plt
-import pandas as pd
+from flask import Flask, request, render_template, send_file
 import io
+from xhtml2pdf import pisa
+import pandas as pd
+import matplotlib.pyplot as plt
 import base64
-from weasyprint import HTML
-from datetime import datetime
 
 app = Flask(__name__)
 
-def formata_real(valor):
-    return f"R$ {valor:,.2f}".replace(',', 'v').replace('.', ',').replace('v', '.')
-
-def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, anos=10, inflacao=4.5):
-    data = []
-    cap_studio = capital
-    cap_acoes = capital
-    cap_renda = capital
-    cap_franquia = capital
-
-    for ano in range(anos + 1):
-        if ano == 0:
-            studio_valor = cap_studio
-            acoes_valor = cap_acoes
-            renda_valor = cap_renda
-            franquia_valor = cap_franquia
-        else:
-            renda_aluguel = cap_studio * (studio_ret / 100) * 12
-            cap_studio += renda_aluguel
-            cap_studio *= (1 + valorizacao / 100)
-
-            cap_franquia += franquia_ret
-
-            cap_acoes *= (1 + acoes_ret / 100)
-            cap_renda *= (1 + renda_fixa / 100)
-
-            cap_studio /= (1 + inflacao / 100)
-            cap_acoes /= (1 + inflacao / 100)
-            cap_renda /= (1 + inflacao / 100)
-            cap_franquia /= (1 + inflacao / 100)
-
-            studio_valor = cap_studio
-            franquia_valor = cap_franquia
-            acoes_valor = cap_acoes
-            renda_valor = cap_renda
-
-        data.append({
-            "Ano": ano,
-            "Studio (R$)": round(studio_valor, 2),
-            "Franquia (R$)": round(franquia_valor, 2),
-            "Ações (R$)": round(acoes_valor, 2),
-            "Renda Fixa (R$)": round(renda_valor, 2),
-        })
-
-    return pd.DataFrame(data)
-
-def plotar_grafico(df):
-    plt.style.use('ggplot')
-    fig, ax = plt.subplots(figsize=(10, 6))
-    cores = ['#2c3e50', '#16a085', '#2980b9', '#8e44ad']
-
-    for i, col in enumerate(df.columns[1:]):
-        ax.plot(df['Ano'], df[col], label=col, color=cores[i], linewidth=2)
-
-    ax.set_title('Comparação de Investimentos (valores corrigidos pela inflação)', fontsize=16, fontweight='bold')
-    ax.set_xlabel('Ano', fontsize=14)
-    ax.set_ylabel('Valor (R$)', fontsize=14)
-    ax.legend(fontsize=12)
-    ax.grid(True, linestyle='--', alpha=0.7)
-    plt.tight_layout()
-
-    img = io.BytesIO()
-    plt.savefig(img, format='png', dpi=150)
-    plt.close(fig)
-    img.seek(0)
-    img_base64 = base64.b64encode(img.read()).decode()
-    return img_base64
-
-@app.route('/download_pdf', methods=['POST'])
-def download_pdf():
-    try:
+@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'])
@@ -92,38 +23,95 @@ def download_pdf():
         renda_fixa = float(request.form['renda_fixa'])
         inflacao = float(request.form['inflacao'])
 
-        df = gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao=inflacao)
-        grafico = plotar_grafico(df)
+        # 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]
+        }
+
+        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()
+
+        # Gráfico
+        fig, ax = plt.subplots()
+        df.plot(ax=ax)
+        ax.set_ylabel("Valor (R$)")
+        ax.set_title("Comparativo de Investimentos")
+
+        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")
 
-        valores_finais = df.iloc[-1, 1:]
-        investimento_mais_valorizado = valores_finais.idxmax()
-        valor_mais_alto = valores_finais.max()
-
-        ano_atual = datetime.now().year
-
-        # Formata os valores no DataFrame antes de converter para HTML
-        df_formatado = df.copy()
-        for col in df.columns[1:]:
-            df_formatado[col] = df_formatado[col].apply(formata_real)
-
-        tabela_html = df_formatado.to_html(classes="table table-striped table-hover", index=False)
-
-        html_content = render_template('relatorio.html',
-                                       tabela=tabela_html,
-                                       grafico=grafico,
-                                       capital=formata_real(capital),
-                                       investimento_mais_valorizado=investimento_mais_valorizado,
-                                       valor_mais_alto=formata_real(valor_mais_alto),
-                                       year=ano_atual)
-
-        pdf = HTML(string=html_content).write_pdf()
+@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'])
+
+    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]
+    }
+
+    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)
+
+    pdf = io.BytesIO()
+    pisa.CreatePDF(io.StringIO(html), dest=pdf)
+    pdf.seek(0)
+
+    return send_file(pdf, download_name="relatorio_simulacao.pdf", as_attachment=True)
 
-        response = make_response(pdf)
-        response.headers['Content-Type'] = 'application/pdf'
-        response.headers['Content-Disposition'] = 'attachment; filename=relatorio_simulacao.pdf'
-        return response
-    except Exception as e:
-        return f"Erro: {e}"
 
 
 if __name__ == '__main__':