Update app.py

app.py

@@ -1,126 +1,94 @@
-from flask import Flask, render_template, request
+from flask import Flask, request, jsonify
 import matplotlib.pyplot as plt
 import io
 import base64
-import csv
-from flask import send_file
 
 app = Flask(__name__)
 
-def calcular_retornos(capital, studio_valor_anual, studio_valoriza, franquia_valor_anual, acoes_retorno, renda_fixa_retorno):
-    anos = [2025, 2026, 2027, 2028, 2029]
-    resultados = {
-        'studio': [],
-        'franquia': [],
-        'acoes': [],
-        'renda_fixa': []
-    }
-
-    valor_studio = capital
-    valor_franquia = capital
-    valor_acoes = capital
-    valor_rf = capital
-
-    for _ in anos:
-        valor_studio += studio_valor_anual
-        valor_studio *= (1 + studio_valoriza / 100)
-        resultados['studio'].append(round(valor_studio, 2))
-
-        valor_franquia += franquia_valor_anual
-        resultados['franquia'].append(round(valor_franquia, 2))
-
-        valor_acoes *= (1 + acoes_retorno / 100)
-        resultados['acoes'].append(round(valor_acoes, 2))
-
-        valor_rf *= (1 + renda_fixa_retorno / 100)
-        resultados['renda_fixa'].append(round(valor_rf, 2))
-
-    return anos, resultados
-
-@app.route('/', methods=['GET', 'POST'])
+@app.route('/')
 def index():
-    grafico = None
-    resultados = {}
-    anos = []
-    params = {
-        'capital': 400000,
-        'studio_valor_anual': 49440,
-        'studio_valoriza': 6,
-        'franquia_valor_anual': 20000,
-        'acoes_retorno': 10,
-        'renda_fixa_retorno': 9,
-    }
-    if request.method == 'POST':
-        params['capital'] = float(request.form.get('capital', params['capital']))
-        params['studio_valor_anual'] = float(request.form.get('studio_valor_anual', params['studio_valor_anual']))
-        params['studio_valoriza'] = float(request.form.get('studio_valoriza', params['studio_valoriza']))
-        params['franquia_valor_anual'] = float(request.form.get('franquia_valor_anual', params['franquia_valor_anual']))
-        params['acoes_retorno'] = float(request.form.get('acoes_retorno', params['acoes_retorno']))
-        params['renda_fixa_retorno'] = float(request.form.get('renda_fixa_retorno', params['renda_fixa_retorno']))
-
-        anos, resultados = calcular_retornos(
-            params['capital'],
-            params['studio_valor_anual'],
-            params['studio_valoriza'],
-            params['franquia_valor_anual'],
-            params['acoes_retorno'],
-            params['renda_fixa_retorno']
-        )
-
-        plt.figure(figsize=(8, 5))
-        for tipo, valores in resultados.items():
-            plt.plot(anos, valores, marker='o', label=tipo.title())
-
-        plt.title("Comparativo de Investimentos (2025-2029)", fontsize=14)
-        plt.xlabel("Ano")
-        plt.ylabel("Valor acumulado (R$)")
-        plt.grid(True, linestyle='--', alpha=0.7)
-        plt.legend()
-
+    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)
-        grafico = base64.b64encode(buf.getvalue()).decode('utf8')
-        plt.close()
-
-    return render_template('index.html', grafico=grafico, resultados=resultados, anos=anos, params=params)
-
-def gerar_csv(resultados, anos):
-    buf = io.StringIO()
-    writer = csv.writer(buf)
-    header = ["Ano"] + list(resultados.keys())
-    writer.writerow(header)
-    for i in range(len(anos)):
-        row = [anos[i]] + [resultados[tipo][i] for tipo in resultados]
-        writer.writerow(row)
-    buf.seek(0)
-    return io.BytesIO(buf.read().encode('utf-8'))
-
-@app.route('/download_csv', methods=['POST'])
-def download_csv():
-    params = {
-        'capital': float(request.form.get('capital', 400000)),
-        'studio_valor_anual': float(request.form.get('studio_valor_anual', 49440)),
-        'studio_valoriza': float(request.form.get('studio_valoriza', 6)),
-        'franquia_valor_anual': float(request.form.get('franquia_valor_anual', 20000)),
-        'acoes_retorno': float(request.form.get('acoes_retorno', 10)),
-        'renda_fixa_retorno': float(request.form.get('renda_fixa_retorno', 9)),
-    }
-
-    anos, resultados = calcular_retornos(
-        params['capital'],
-        params['studio_valor_anual'],
-        params['studio_valoriza'],
-        params['franquia_valor_anual'],
-        params['acoes_retorno'],
-        params['renda_fixa_retorno']
-    )
-
-    csv_file = gerar_csv(resultados, anos)
-    return send_file(csv_file, mimetype='text/csv', as_attachment=True, download_name='simulacao_investimentos.csv')
-
-if __name__ == '__main__':
- 
-
-    app.run(host='0.0.0.0', port=7860, debug=True)
+        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)
+