Update app.py

app.py

@@ -1,61 +1,69 @@
 import os
-os.environ['FONTCONFIG_CACHE'] = '/tmp/fontconfig_cache'
-os.makedirs('/tmp/fontconfig_cache', exist_ok=True)
 
+# Corrige problema fontconfig
+os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
+os.makedirs('/tmp/fontconfig', exist_ok=True)
 
-
-from flask import Flask, render_template, request, make_response
-import pandas as pd
+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__)
 
-# Função principal de simulação
-def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao=0, anos=10):
-    anos_lista = list(range(1, anos + 1))
-    studio, imovel, franquia, acoes, renda_fixa_arr = [], [], [], [], []
-
-    cap_studio = cap_imovel = cap_franquia = cap_acoes = cap_renda_fixa = capital
-
-    for ano in anos_lista:
-        inf_cum = (1 + inflacao / 100) ** ano
-        cap_studio *= (1 + studio_ret / 100)
-        cap_imovel *= (1 + valorizacao / 100)
-        cap_franquia += franquia_ret
-        cap_acoes *= (1 + acoes_ret / 100)
-        cap_renda_fixa *= (1 + renda_fixa / 100)
-
-        studio.append(cap_studio / inf_cum)
-        imovel.append(cap_imovel / inf_cum)
-        franquia.append(cap_franquia / inf_cum)
-        acoes.append(cap_acoes / inf_cum)
-        renda_fixa_arr.append(cap_renda_fixa / inf_cum)
-
-    return pd.DataFrame({
-        'Ano': anos_lista,
-        'Studio': studio,
-        'Imovel': imovel,
-        'Franquia': franquia,
-        'Acoes': acoes,
-        'Renda Fixa': renda_fixa_arr
-    })
-
-# Geração do gráfico como imagem base64
-def plotar_grafico(df):
-    plt.style.use('ggplot')
-    fig, ax = plt.subplots(figsize=(10, 6))
-    cores = ['#2c3e50', '#16a085', '#2980b9', '#8e44ad', '#d35400']
+def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, anos=10, inflacao=4.5):
+    data = []
+    cap_studio = capital
+    cap_acoes = capital
+    cap_renda = capital
+    cap_franquia = capital
+    for ano in range(anos+1):
+        if ano == 0:
+            studio_valor = cap_studio
+            acoes_valor = cap_acoes
+            renda_valor = cap_renda
+            franquia_valor = cap_franquia
+        else:
+            renda_aluguel = cap_studio * (studio_ret/100) * 12
+            cap_studio += renda_aluguel
+            cap_studio *= (1 + valorizacao/100)
+
+            cap_franquia += franquia_ret
+
+            cap_acoes *= (1 + acoes_ret/100)
+            cap_renda *= (1 + renda_fixa/100)
+
+            # Corrige inflação
+            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),
+        })
+    return pd.DataFrame(data)
 
+def plotar_grafico(df):
+    plt.style.use('ggplot')  # estilo moderno simples
+    fig, ax = plt.subplots(figsize=(10,6))
+    cores = ['#2c3e50', '#16a085', '#2980b9', '#8e44ad']
     for i, col in enumerate(df.columns[1:]):
-        estilo = '--' if col.lower() == 'acoes' else '-'
-        ax.plot(df['Ano'], df[col], label=col, color=cores[i], linewidth=2, linestyle=estilo)
-
+        ax.plot(df['Ano'], df[col], label=col, color=cores[i], linewidth=2)
     ax.set_title('Comparação de Investimentos (valores corrigidos pela inflação)', fontsize=14, fontweight='bold')
-    ax.set_xlabel('Ano')
-    ax.set_ylabel('Valor (R$)')
+    ax.set_xlabel('Ano', fontsize=12)
+    ax.set_ylabel('Valor (R$)', fontsize=12)
     ax.legend()
     ax.grid(True, linestyle='--', alpha=0.7)
     plt.tight_layout()
@@ -64,65 +72,90 @@ def plotar_grafico(df):
     plt.savefig(img, format='png', dpi=150)
     plt.close(fig)
     img.seek(0)
-    return base64.b64encode(img.read()).decode()
+    img_base64 = base64.b64encode(img.read()).decode()
+    return img_base64
 
-# Página inicial com formulário e simulação
 @app.route('/', methods=['GET', 'POST'])
 def index():
+    grafico = None
+    df = None
     if request.method == 'POST':
         try:
-            # Coleta dos dados do formulário
-            dados = {k: float(request.form[k]) for k in [
-                'capital', 'studio_ret', 'valorizacao',
-                'franquia_ret', 'acoes_ret', 'renda_fixa', 'inflacao'
-            ]}
-
-            df = gerar_simulacao(**dados)
+            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)
-            tabela = df.to_html(classes="table table-striped table-hover", index=False)
-
-            return render_template("index.html", grafico=grafico, tabela=tabela, **dados)
 
+            return render_template('index.html', grafico=grafico, tabela=df.to_html(classes="table table-striped table-hover", 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: {e}"
+    return render_template('index.html')
 
-    return render_template("index.html")
+@app.route('/download_excel', methods=['POST'])
+def download_excel():
+    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}"
 
-# Geração e download do PDF
 @app.route('/download_pdf', methods=['POST'])
 def download_pdf():
     try:
-        dados = {k: float(request.form[k]) for k in [
-            'capital', 'studio_ret', 'valorizacao',
-            'franquia_ret', 'acoes_ret', 'renda_fixa', 'inflacao'
-        ]}
-
-        df = gerar_simulacao(**dados)
+        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)
 
-        valores_finais = df.iloc[-1, 1:]
-        nome_mais_valorizado = valores_finais.idxmax()
-        valor_mais_alto = valores_finais.max()
-        valor_mais_alto_str = f"{valor_mais_alto:,.2f}".replace(',', 'X').replace('.', ',').replace('X', '.')
-
-        html = render_template("relatorio.html",
-                               grafico=grafico,
-                               tabela=df.to_html(classes="table table-striped table-hover", index=False),
-                               investimento_mais_valorizado=nome_mais_valorizado,
-                               valor_mais_alto=valor_mais_alto_str,
-                               df=df,
-                               **dados)
+        html_content = render_template('relatorio.html',
+                                       tabela=df.to_html(classes="table table-striped table-hover", 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()
 
-        pdf = HTML(string=html).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 ao gerar PDF: {e}"
+        return f"Erro: {e}"
 
-# Execução local
 if __name__ == '__main__':
-    app.run(debug=True, port=7860)
+    app.run(host='0.0.0.0', port=7860, debug=True)