Update app.py

app.py

@@ -1,95 +1,120 @@
 from flask import Flask, render_template, request, send_file
-import io
-import base64
 import matplotlib.pyplot as plt
 import pandas as pd
+import io
+import base64
 
 app = Flask(__name__)
 
-def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao):
-    # Cria DataFrame com dados simulados para 10 anos
-    anos = list(range(1, 11))
-    dados = {
-        'Ano': anos,
-        'Studio_Rentabilidade_Bruta': [],
-        'Studio_Valorizacao': [],
-        'Franquia_Lucro_Anual': [],
-        'Acoes_Retorno': [],
-        'Renda_Fixa_Retorno': [],
-    }
-    
-    capital_studio = capital
-    capital_acoes = capital
-    capital_rf = capital
-    
-    for ano in anos:
-        # Studio: rentabilidade mensal bruta acumulada no ano + valorização
-        rent_mes = studio_ret / 100
-        rent_anual = ((1 + rent_mes) ** 12 - 1) * capital_studio
-        val_ano = (valorizacao / 100) * capital_studio
-        dados['Studio_Rentabilidade_Bruta'].append(rent_anual)
-        dados['Studio_Valorizacao'].append(val_ano)
-        
-        # Franquia: valor fixo anual informado (pode ser ajustado depois)
-        dados['Franquia_Lucro_Anual'].append(franquia_ret)
-        
-        # Ações e RF: acumulam juros compostos
-        capital_acoes = capital_acoes * (1 + (acoes_ret / 100))
-        capital_rf = capital_rf * (1 + (renda_fixa / 100))
-        
-        dados['Acoes_Retorno'].append(capital_acoes - capital)
-        dados['Renda_Fixa_Retorno'].append(capital_rf - capital)
-    
-    df = pd.DataFrame(dados)
+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
+    })
+
     return df
 
-def gerar_grafico(df):
+def plotar_grafico(df):
     plt.figure(figsize=(10,6))
-    plt.plot(df['Ano'], df['Studio_Rentabilidade_Bruta'], label='Studio - Renda Bruta')
-    plt.plot(df['Ano'], df['Studio_Valorizacao'], label='Studio - Valorização')
-    plt.plot(df['Ano'], df['Franquia_Lucro_Anual'], label='Franquia - Lucro Anual')
-    plt.plot(df['Ano'], df['Acoes_Retorno'], label='Ações - Retorno Acumulado')
-    plt.plot(df['Ano'], df['Renda_Fixa_Retorno'], label='Renda Fixa - Retorno Acumulado')
+    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('Retorno (R$)')
-    plt.title('Simulação de Retorno de Investimentos')
+    plt.ylabel('Valor acumulado (R$)')
+    plt.grid(True, alpha=0.3)
     plt.legend()
-    plt.grid(True)
-    
+    plt.tight_layout()
+
     buf = io.BytesIO()
-    plt.savefig(buf, format='png', bbox_inches='tight')
+    plt.savefig(buf, format='png', transparent=True)
     plt.close()
     buf.seek(0)
-    img_base64 = base64.b64encode(buf.read()).decode('utf-8')
+    img_base64 = base64.b64encode(buf.getvalue()).decode('utf8')
     return img_base64
 
 @app.route('/', methods=['GET', 'POST'])
 def index():
+    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 = gerar_grafico(df)
-        
-        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
-        )
-    else:
-        return render_template('index.html')
+        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)
+            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)
+        except Exception as e:
+            return f"Erro ao processar dados: {e}", 400
+
+    return render_template('index.html', grafico=None)
 
 @app.route('/download_excel', methods=['POST'])
 def download_excel():
@@ -100,29 +125,19 @@ def download_excel():
     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
-    )
-    
+
+    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')
-        writer.save()
     output.seek(0)
-    
-    return send_file(
-        output,
-        download_name="simulacao_investimentos.xlsx",
-        as_attachment=True,
-        mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
-    )
+
+    return send_file(output,
+                     download_name="simulacao_investimentos.xlsx",
+                     as_attachment=True,
+                     mimetype='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
+
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)