Update app.py

app.py

@@ -1,5 +1,7 @@
 import os
 
+import os
+
 # Corrige problema fontconfig (necessário para WeasyPrint em alguns ambientes)
 os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
 os.makedirs('/tmp/fontconfig', exist_ok=True)
@@ -11,7 +13,6 @@ import io
 import base64
 from weasyprint import HTML
 
-
 app = Flask(__name__)
 
 def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, anos=10, inflacao=4.5):
@@ -28,7 +29,6 @@ def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, r
             renda_valor = cap_renda
             franquia_valor = cap_franquia
         else:
-            # Renda de aluguel anual (studio_ret é mensal)
             renda_aluguel = cap_studio * (studio_ret / 100) * 12
             cap_studio += renda_aluguel
             cap_studio *= (1 + valorizacao / 100)
@@ -38,7 +38,6 @@ def gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, r
             cap_acoes *= (1 + acoes_ret / 100)
             cap_renda *= (1 + renda_fixa / 100)
 
-            # Ajusta pela inflação anual
             cap_studio /= (1 + inflacao / 100)
             cap_acoes /= (1 + inflacao / 100)
             cap_renda /= (1 + inflacao / 100)
@@ -85,6 +84,8 @@ def plotar_grafico(df):
 def index():
     grafico = None
     df = None
+    gerar_pdf = False  # Flag para template
+
     if request.method == 'POST':
         try:
             capital = float(request.form['capital'])
@@ -100,10 +101,11 @@ def index():
 
             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)
+                                   acoes_ret=acoes_ret, renda_fixa=renda_fixa, inflacao=inflacao,
+                                   gerar_pdf=gerar_pdf)
         except Exception as e:
             return f"Erro: {e}"
-    return render_template('index.html')
+    return render_template('index.html', gerar_pdf=gerar_pdf)
 
 @app.route('/download_excel', methods=['POST'])
 def download_excel():
@@ -143,12 +145,11 @@ def download_pdf():
         df = gerar_simulacao(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa, inflacao=inflacao)
         grafico = plotar_grafico(df)
 
-        # Calcula o investimento mais valorizado para o destaque do relatório
-        valores_finais = df.iloc[-1, 1:]  # pega a última linha, exceto a coluna 'Ano'
+        valores_finais = df.iloc[-1, 1:]
         investimento_mais_valorizado = valores_finais.idxmax()
         valor_mais_alto = valores_finais.max()
 
-        html_content = render_template('relatorio.html',
+        html_content = render_template('index.html',
                                        tabela=df.to_html(classes="table table-striped table-hover", index=False),
                                        grafico=grafico,
                                        capital=capital,
@@ -158,9 +159,10 @@ def download_pdf():
                                        acoes_ret=acoes_ret,
                                        renda_fixa=renda_fixa,
                                        inflacao=inflacao,
-                                       df=df,
                                        investimento_mais_valorizado=investimento_mais_valorizado,
-                                       valor_mais_alto=valor_mais_alto)
+                                       valor_mais_alto=valor_mais_alto,
+                                       gerar_pdf=True)  # Passa a flag para ocultar botão no PDF
+
         pdf = HTML(string=html_content).write_pdf()
 
         response = make_response(pdf)
@@ -172,3 +174,4 @@ def download_pdf():
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)
+