Update app.py

app.py

@@ -1,94 +1,121 @@
-from flask import Flask, request, jsonify
+import os
+os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib-config'
+
+from flask import Flask, render_template, request
 import matplotlib.pyplot as plt
 import io
 import base64
+import pandas as pd
 
 app = Flask(__name__)
 
-@app.route('/')
+def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao):
+    # Simulação de 10 anos
+    anos = list(range(11))
+    studio_valor = []
+    franquia_valor = []
+    acoes_valor = []
+    renda_fixa_valor = []
+
+    cap_studio = capital
+    cap_franquia = capital
+    cap_acoes = capital
+    cap_rf = capital
+
+    for ano in anos:
+        # Studio (rendimento mensal *12 + valorização anual)
+        rent_studio = cap_studio * (studio_ret/100) * 12
+        val_studio = cap_studio * (valorizacao/100)
+        cap_studio += rent_studio + val_studio
+
+        # Franquia (lucro fixo anual)
+        cap_franquia += franquia_ret
+
+        # Ações (rendimento anual percentual)
+        cap_acoes += cap_acoes * (acoes_ret/100)
+
+        # Renda fixa (rendimento anual percentual)
+        cap_rf += cap_rf * (renda_fixa/100)
+
+        # Ajuste pela inflação
+        cap_studio /= (1 + inflacao/100)
+        cap_franquia /= (1 + inflacao/100)
+        cap_acoes /= (1 + inflacao/100)
+        cap_rf /= (1 + inflacao/100)
+
+        # Armazena valores ajustados
+        studio_valor.append(cap_studio)
+        franquia_valor.append(cap_franquia)
+        acoes_valor.append(cap_acoes)
+        renda_fixa_valor.append(cap_rf)
+
+    # Monta DataFrame para excel e gráfico
+    df = pd.DataFrame({
+        'Ano': anos,
+        'Studio (R$)': [round(v,2) for v in studio_valor],
+        'Franquia (R$)': [round(v,2) for v in franquia_valor],
+        'Ações (R$)': [round(v,2) for v in acoes_valor],
+        'Renda Fixa (R$)': [round(v,2) for v in renda_fixa_valor],
+    })
+
+    return df
+
+def criar_grafico(df):
+    plt.style.use('seaborn-darkgrid')
+    fig, ax = plt.subplots(figsize=(10,6))
+    ax.plot(df['Ano'], df['Studio (R$)'], label='Studio')
+    ax.plot(df['Ano'], df['Franquia (R$)'], label='Franquia')
+    ax.plot(df['Ano'], df['Ações (R$)'], label='Ações')
+    ax.plot(df['Ano'], df['Renda Fixa (R$)'], label='Renda Fixa')
+    ax.set_title('Simulação de Investimentos - Valor Ajustado pela Inflação')
+    ax.set_xlabel('Ano')
+    ax.set_ylabel('Valor acumulado (R$)')
+    ax.legend()
+    plt.tight_layout()
+
+    buf = io.BytesIO()
+    plt.savefig(buf, format='png')
+    plt.close(fig)
+    buf.seek(0)
+    img_png = base64.b64encode(buf.getvalue()).decode('utf-8')
+    return img_png
+
+@app.route('/', methods=['GET', 'POST'])
 def index():
-    return "Backend para o simulador está rodando."
-
-@app.route('/simular', methods=['POST'])
-def simular():
-    try:
-        # Recebe os dados do formulário
-        capital = float(request.form.get('capital', 0))
-        studio_ret = float(request.form.get('studio_ret', 0)) / 100
-        valorizacao = float(request.form.get('valorizacao', 0)) / 100
-        franquia_ret = float(request.form.get('franquia_ret', 0))
-        acoes_ret = float(request.form.get('acoes_ret', 0)) / 100
-        renda_fixa = float(request.form.get('renda_fixa', 0)) / 100
-        inflacao = float(request.form.get('inflacao', 0)) / 100
-
-        anos = 10
-
-        # Simulação simples: calcular valor acumulado após 10 anos para cada investimento
-
-        # Studio: retorno mensal aplicado com juros compostos
-        studio_acumulado = capital * ((1 + studio_ret) ** (12 * anos))
-
-        # Imóvel: valorização anual composta + franquia lucro fixo anual
-        valor_imovel = capital * ((1 + valorizacao) ** anos)
-        lucro_franquia_total = franquia_ret * anos
-        imovel_acumulado = valor_imovel + lucro_franquia_total
-
-        # Ações: retorno anual composto
-        acoes_acumulado = capital * ((1 + acoes_ret) ** anos)
-
-        # Renda fixa: retorno anual composto
-        renda_fixa_acumulado = capital * ((1 + renda_fixa) ** anos)
-
-        # Inflação acumulada
-        inflacao_acumulada = (1 + inflacao) ** anos
-
-        # Ajustando valores pela inflação (poder de compra atual)
-        studio_real = studio_acumulado / inflacao_acumulada
-        imovel_real = imovel_acumulado / inflacao_acumulada
-        acoes_real = acoes_acumulado / inflacao_acumulada
-        renda_fixa_real = renda_fixa_acumulado / inflacao_acumulada
-
-        # Texto explicativo
-        texto = f"""
-        Simulação para {anos} anos com capital inicial de R$ {capital:,.2f}:
-
-        - Studio (retorno mensal de {studio_ret*100:.2f}%): R$ {studio_real:,.2f} (valor corrigido pela inflação)
-        - Imóvel (valorização anual de {valorizacao*100:.2f}% + lucro franquia anual R$ {franquia_ret:,.2f}): R$ {imovel_real:,.2f}
-        - Ações (retorno anual de {acoes_ret*100:.2f}%): R$ {acoes_real:,.2f}
-        - Renda Fixa (retorno anual de {renda_fixa*100:.2f}%): R$ {renda_fixa_real:,.2f}
-
-        Nota: Todos os valores estão ajustados pela inflação média anual de {inflacao*100:.2f}%.
-        """
-
-        # Gerar gráfico comparativo
-        labels = ['Studio', 'Imóvel', 'Ações', 'Renda Fixa']
-        valores = [studio_real, imovel_real, acoes_real, renda_fixa_real]
-
-        fig, ax = plt.subplots(figsize=(8,5))
-        bars = ax.bar(labels, valores, color=['#00c9a7', '#007cf0', '#ffa500', '#dd1144'])
-        ax.set_title(f'Valor acumulado corrigido pela inflação em {anos} anos', fontsize=14, pad=15)
-        ax.set_ylabel('Valor (R$)')
-        ax.grid(axis='y', linestyle='--', alpha=0.6)
-
-        # Mostrar valores em cima das barras
-        for bar in bars:
-            yval = bar.get_height()
-            ax.text(bar.get_x() + bar.get_width()/2, yval + max(valores)*0.02, f'R$ {yval:,.0f}', ha='center', fontsize=11)
-
-        # Salvar gráfico em PNG na memória
-        buf = io.BytesIO()
-        plt.tight_layout()
-        plt.savefig(buf, format='png')
-        plt.close(fig)
-        buf.seek(0)
-        img_base64 = base64.b64encode(buf.read()).decode('utf-8')
-
-        return jsonify({'texto': texto.strip(), 'graficoBase64': img_base64})
-
-    except Exception as e:
-        return jsonify({'error': str(e)}), 400
-
-if __name__ == "__main__":
-    app.run(debug=True)
+    grafico = None
+    df = None
+    if request.method == 'POST':
+        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)
+        grafico = criar_grafico(df)
+
+    return render_template('index.html', grafico=grafico, capital=request.form.get('capital', ''),
+                           studio_ret=request.form.get('studio_ret', ''),
+                           valorizacao=request.form.get('valorizacao', ''),
+                           franquia_ret=request.form.get('franquia_ret', ''),
+                           acoes_ret=request.form.get('acoes_ret', ''),
+                           renda_fixa=request.form.get('renda_fixa', ''),
+                           inflacao=request.form.get('inflacao', ''))
+
+@app.route('/download_excel', methods=['POST'])
+def download_excel():
+    from flask import send_file
+    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,
+