Update app.py

app.py

@@ -1,89 +1,70 @@
-import os
-os.environ['MPLCONFIGDIR'] = '/tmp/matplotlib-config'
-
-from flask import Flask, render_template, request
-import matplotlib.pyplot as plt
+from flask import Flask, render_template, request, send_file
 import io
 import base64
+import matplotlib.pyplot as plt
 import pandas as pd
 
 app = Flask(__name__)
 
 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({
+    # Cria DataFrame com dados simulados para 10 anos
+    anos = list(range(1, 11))
+    dados = {
         '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],
-    })
-
+        '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)
     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()
-
+def gerar_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.xlabel('Ano')
+    plt.ylabel('Retorno (R$)')
+    plt.title('Simulação de Retorno de Investimentos')
+    plt.legend()
+    plt.grid(True)
+    
     buf = io.BytesIO()
-    plt.savefig(buf, format='png')
-    plt.close(fig)
+    plt.savefig(buf, format='png', bbox_inches='tight')
+    plt.close()
     buf.seek(0)
-    img_png = base64.b64encode(buf.getvalue()).decode('utf-8')
-    return img_png
+    img_base64 = base64.b64encode(buf.read()).decode('utf-8')
+    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'])
@@ -92,21 +73,26 @@ def index():
         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', ''))
+        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')
 
 @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'])
@@ -114,8 +100,31 @@ 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,
-
+    
+    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'
+    )
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860, debug=True)