Update app.py

app.py

@@ -1,93 +1,87 @@
-from flask import Flask, render_template, request, send_file
+from flask import Flask, render_template, request, send_file, make_response
 import matplotlib.pyplot as plt
 import pandas as pd
 import io
 import base64
+from weasyprint import HTML
 
 app = Flask(__name__)
 
-def gerar_simulacao(capital, studio_ret_mensal, valorizacao_anual, franquia_ret_anual,
-                   acoes_ret_anual, renda_fixa_anual, inflacao_anual, anos=10):
-    """
-    Gera DataFrame com projeção de investimentos ao longo dos anos.
-    - studio_ret_mensal em % (ex: 1.0)
-    - valorizacao_anual em % (ex: 6.0)
-    - franquia_ret_anual em R$ (lucro fixo anual)
-    - acoes_ret_anual em % (ex: 10.0)
-    - renda_fixa_anual em % (ex: 9.0)
-    - inflacao_anual em %
-    """
-
-    # Inicializa listas para armazenar os valores ano a ano
-    anos_lista = list(range(1, anos + 1))
-    studio_valor = []
-    franquia_valor = []
-    acoes_valor = []
-    renda_fixa_valor = []
-
-    # Capital inicial para cada investimento
-    studio_capital = capital
-    franquia_capital = capital
-    acoes_capital = capital
-    renda_fixa_capital = capital
-
-    # Retornos mensais para o studio (convertendo % para decimal)
-    studio_ret_mensal /= 100
-    valorizacao_anual /= 100
-    acoes_ret_anual /= 100
-    renda_fixa_anual /= 100
-    inflacao_anual /= 100
-
-    for ano in anos_lista:
-        # Studio:
-        # 1) Rentabilidade mensal capitalizada 12 meses
-        renda_mensal = studio_capital * studio_ret_mensal
-        renda_anual = renda_mensal * 12
-        # 2) Valorização anual do imóvel aplicada ao capital investido
-        studio_capital = (studio_capital + renda_anual) * (1 + valorizacao_anual)
-        studio_valor.append(studio_capital)
-
-        # Franquia:
-        # Considera lucro anual fixo + reinvestimento (simplificação)
-        franquia_capital = (franquia_capital + franquia_ret_anual) * (1 + inflacao_anual * 0.2)  # pequena valorização pela inflação
-        franquia_valor.append(franquia_capital)
-
-        # Ações:
-        acoes_capital = acoes_capital * (1 + acoes_ret_anual)
-        acoes_valor.append(acoes_capital)
-
-        # Renda Fixa:
-        renda_fixa_capital = renda_fixa_capital * (1 + renda_fixa_anual)
-        renda_fixa_valor.append(renda_fixa_capital)
-
-    df = pd.DataFrame({
-        'Ano': anos_lista,
-        'Studio (R$)': studio_valor,
-        'Franquia (R$)': franquia_valor,
-        'Ações (R$)': acoes_valor,
-        'Renda Fixa (R$)': renda_fixa_valor
-    })
-
+def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, anos=10, inflacao=4.5):
+    # Dados anuais: simulação simples para 10 anos
+    # capital: valor inicial
+    # studio_ret: retorno mensal % do aluguel short stay
+    # valorizacao: valorização anual do imóvel %
+    # franquia_ret: lucro anual fixo da franquia (em R$)
+    # acoes_ret: retorno anual em ações %
+    # renda_fixa: retorno anual em renda fixa %
+    # inflacao: inflação anual %
+
+    data = []
+    cap_studio = capital
+    cap_acoes = capital
+    cap_renda = capital
+    cap_franquia = capital
+    for ano in range(anos+1):
+        # Estimativas anuais
+        if ano == 0:
+            studio_valor = cap_studio
+            acoes_valor = cap_acoes
+            renda_valor = cap_renda
+            franquia_valor = cap_franquia
+        else:
+            # Studio: aluguel mensal 1% x 12 meses + valorização imóvel
+            renda_aluguel = cap_studio * (studio_ret/100) * 12
+            cap_studio += renda_aluguel
+            cap_studio *= (1 + valorizacao/100)
+
+            # Franquia: lucro fixo anual reinvestido
+            cap_franquia += franquia_ret
+
+            # Ações: cresce com taxa anual
+            cap_acoes *= (1 + acoes_ret/100)
+
+            # Renda fixa:
+            cap_renda *= (1 + renda_fixa/100)
+
+            # Corrige pela inflação para "real"
+            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),
+        })
+
+    df = pd.DataFrame(data)
     return df
 
 def plotar_grafico(df):
-    plt.figure(figsize=(10,6))
-    plt.plot(df['Ano'], df['Studio (R$)'], marker='o', label='Studio')
-    plt.plot(df['Ano'], df['Franquia (R$)'], marker='o', label='Franquia')
-    plt.plot(df['Ano'], df['Ações (R$)'], marker='o', label='Ações')
-    plt.plot(df['Ano'], df['Renda Fixa (R$)'], marker='o', label='Renda Fixa')
-    plt.title('Projeção de Investimentos ao Longo dos Anos')
-    plt.xlabel('Ano')
-    plt.ylabel('Valor acumulado (R$)')
-    plt.grid(True, alpha=0.3)
-    plt.legend()
+    plt.style.use('seaborn-darkgrid')
+    fig, ax = plt.subplots(figsize=(10,6))
+    for col in df.columns[1:]:
+        ax.plot(df['Ano'], df[col], label=col)
+    ax.set_title('Comparação de Investimentos (valores corrigidos pela inflação)')
+    ax.set_xlabel('Ano')
+    ax.set_ylabel('Valor (R$)')
+    ax.legend()
     plt.tight_layout()
 
-    buf = io.BytesIO()
-    plt.savefig(buf, format='png', transparent=True)
-    plt.close()
-    buf.seek(0)
-    img_base64 = base64.b64encode(buf.getvalue()).decode('utf8')
+    img = io.BytesIO()
+    plt.savefig(img, format='png')
+    plt.close(fig)
+    img.seek(0)
+    img_base64 = base64.b64encode(img.read()).decode()
     return img_base64
 
 @app.route('/', methods=['GET', 'POST'])
@@ -104,40 +98,71 @@ def index():
             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)
+            df = gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao=inflacao)
             grafico = plotar_grafico(df)
-
-            # Passa variáveis para renderizar no template
-            return render_template('index.html', grafico=grafico, capital=capital, studio_ret=studio_ret,
-                                   valorizacao=valorizacao, franquia_ret=franquia_ret, acoes_ret=acoes_ret,
-                                   renda_fixa=renda_fixa, inflacao=inflacao)
+            return render_template('index.html', grafico=grafico, tabela=df.to_html(classes="table table-striped", index=False),
+                                   capital=capital, studio_ret=studio_ret, valorizacao=valorizacao, franquia_ret=franquia_ret,
+                                   acoes_ret=acoes_ret, renda_fixa=renda_fixa, inflacao=inflacao)
         except Exception as e:
-            return f"Erro ao processar dados: {e}", 400
-
-    return render_template('index.html', grafico=None)
+            return f"Erro: {e}"
+    return render_template('index.html')
 
 @app.route('/download_excel', methods=['POST'])
 def download_excel():
-    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 = gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao)
-
-    output = io.BytesIO()
-    with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
-        df.to_excel(writer, index=False, sheet_name='Simulação')
-    output.seek(0)
-
-    return send_file(output,
-                     download_name="simulacao_investimentos.xlsx",
-                     as_attachment=True,
-                     mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
-
+    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'])
+
+        df = gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao=inflacao)
+
+        output = io.BytesIO()
+        with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
+            df.to_excel(writer, sheet_name='Simulação', index=False)
+        output.seek(0)
+        return send_file(output,
+                         as_attachment=True,
+                         download_name="simulacao_investimentos.xlsx",
+                         mimetype="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
+    except Exception as e:
+        return f"Erro: {e}"
+
+@app.route('/download_pdf', methods=['POST'])
+def download_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'])
+
+        df = gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao=inflacao)
+        grafico = plotar_grafico(df)
+
+        html_content = render_template('relatorio.html',
+                                       tabela=df.to_html(classes="table table-striped", index=False),
+                                       grafico=grafico,
+                                       capital=capital,
+                                       studio_ret=studio_ret,
+                                       valorizacao=valorizacao,
+                                       franquia_ret=franquia_ret,
+                                       acoes_ret=acoes_ret,
+                                       renda_fixa=renda_fixa,
+                                       inflacao=inflacao)
+        pdf = HTML(string=html_content).write_pdf()
+
+        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__':
     app.run(host='0.0.0.0', port=7860, debug=True)