Update app.py

app.py

@@ -7,19 +7,12 @@ 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 = capital
-    cap_imovel = capital
-    cap_franquia = capital
-    cap_acoes = capital
-    cap_renda_fixa = capital
+    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
@@ -44,13 +37,16 @@ def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, r
         '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']
+
     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.set_title('Comparação de Investimentos (valores corrigidos pela inflação)', fontsize=14, fontweight='bold')
     ax.set_xlabel('Ano')
     ax.set_ylabel('Valor (R$)')
@@ -64,70 +60,63 @@ def plotar_grafico(df):
     img.seek(0)
     return base64.b64encode(img.read()).decode()
 
+# Página inicial com formulário e simulação
 @app.route('/', methods=['GET', 'POST'])
 def index():
     if request.method == 'POST':
         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)
+            # 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)
             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,
-                                   capital=capital, studio_ret=studio_ret,
-                                   valorizacao=valorizacao, franquia_ret=franquia_ret,
-                                   acoes_ret=acoes_ret, renda_fixa=renda_fixa, inflacao=inflacao)
+
+            return render_template("index.html", grafico=grafico, tabela=tabela, **dados)
+
         except Exception as e:
             return f"Erro: {e}"
+
     return render_template("index.html")
 
+# Geração e download do PDF
 @app.route('/download_pdf', methods=['POST'])
 def download_pdf():
     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)
+        dados = {k: float(request.form[k]) for k in [
+            'capital', 'studio_ret', 'valorizacao',
+            'franquia_ret', 'acoes_ret', 'renda_fixa', 'inflacao'
+        ]}
+
+        df = gerar_simulacao(**dados)
         grafico = plotar_grafico(df)
 
-        ultimo_ano = df['Ano'].iloc[-1]
         valores_finais = df.iloc[-1, 1:]
-        nome_mais_alto = valores_finais.idxmax()
+        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),
-                               capital=capital,
-                               studio_ret=studio_ret,
-                               valorizacao=valorizacao,
-                               franquia_ret=franquia_ret,
-                               acoes_ret=acoes_ret,
-                               renda_fixa=renda_fixa,
-                               inflacao=inflacao,
-                               investimento_mais_valorizado=nome_mais_alto,
+                               investimento_mais_valorizado=nome_mais_valorizado,
                                valor_mais_alto=valor_mais_alto_str,
-                               df=df)
+                               df=df,
+                               **dados)
 
         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: {e}"
+        return f"Erro ao gerar PDF: {e}"
 
+# Execução local
 if __name__ == '__main__':
     app.run(debug=True, port=7860)
+