Update app.py

app.py

@@ -174,43 +174,82 @@ def index():
 
     return render_template("index.html")
 
+
+
+
+
+
 @app.route("/gerar-pdf", methods=["POST"])
 def gerar_pdf():
     try:
-        logging.debug(f"Dados recebidos para gerar PDF: {request.form}")
-
-        capital = safe_float(request.form.get("capital"), "capital")
-        studio_ret = safe_float(request.form.get("studio_ret"), "studio_ret")
-        valorizacao = safe_float(request.form.get("valorizacao"), "valorizacao")
-        franquia_ret = safe_float(request.form.get("franquia_ret"), "franquia_ret")
-        acoes_ret = safe_float(request.form.get("acoes_ret"), "acoes_ret")
-        renda_fixa = safe_float(request.form.get("renda_fixa"), "renda_fixa")
-        inflacao = safe_float(request.form.get("inflacao"), "inflacao")
-
-        dados, investimentos_finais = calcular_projecoes(capital, studio_ret, valorizacao, franquia_ret, acoes_ret, renda_fixa)
+        # --- Coleta e conversão dos dados do formulário
+        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"])  # Ainda não usado
+
+        anos = list(range(1, 6))
+
+        # --- Cálculo das projeções de investimento
+        patrimonio_studio = [capital * ((1 + valorizacao / 100) ** ano) for ano in anos]
+        renda_acumulada_studio = [capital * (((1 + studio_ret / 100) ** (12 * ano)) - 1) for ano in anos]
+        studio_total = [p + r for p, r in zip(patrimonio_studio, renda_acumulada_studio)]
+        patrimonio_final = patrimonio_studio[-1]
+
+        franquia = [capital + (franquia_ret * ano) for ano in anos]
+        acoes = [capital * ((1 + acoes_ret / 100) ** ano) for ano in anos]
+        renda_fixa_valores = [capital * ((1 + renda_fixa / 100) ** ano) for ano in anos]
+
+        # --- Monta o DataFrame
+        df = pd.DataFrame({
+            "Ano": anos,
+            "Studio (Patrimônio + Renda)": studio_total,
+            "Franquia": franquia,
+            "Ações": acoes,
+            "Renda Fixa": renda_fixa_valores
+        })
+
+        # --- Dicionário final de investimentos
+        investimentos_finais = {
+            "Studio": studio_total[-1],
+            "Franquia": franquia[-1],
+            "Ações": acoes[-1],
+            "Renda Fixa": renda_fixa_valores[-1],
+        }
 
-        df = pd.DataFrame(dados)
+        # --- Geração do gráfico
+        buf = io.BytesIO()
+        plt.figure(figsize=(8, 5))
+        plt.plot(anos, studio_total, label="Studio (Patrimônio + Renda)", marker="o")
+        plt.plot(anos, franquia, label="Franquia", marker="o")
+        plt.plot(anos, acoes, label="Ações", marker="o")
+        plt.plot(anos, renda_fixa_valores, label="Renda Fixa", marker="o")
+        plt.title("Projeção de Investimentos (5 anos)")
+        plt.xlabel("Ano")
+        plt.ylabel("Valor (R$)")
+        plt.legend()
+        plt.grid(True)
+        plt.tight_layout()
+        plt.savefig(buf, format="png")
+        buf.seek(0)
+        grafico_base64 = base64.b64encode(buf.read()).decode("utf-8")
+        buf.close()
+        plt.close()
+
+        # --- Formata a tabela
         df_formatado = df.copy()
         for col in df.columns:
             if col != "Ano":
                 df_formatado[col] = df_formatado[col].apply(formatar_brl)
         tabela_html = df_formatado.to_html(index=False, classes="tabela", border=0)
 
-        anos = dados["Ano"]
-        studio_total = dados["Studio (Patrimônio + Renda)"]
-
-        grafico_base64 = gerar_grafico(
-            anos,
-            studio_total,
-            dados["Franquia"],
-            dados["Ações"],
-            dados["Renda Fixa"],
-        )
-
-       
-        analise_final_ia = gerar_analise_ia(investimentos_finais, capital, patrimonio_studio[-1])
-        analise_final_ia = analise_final_ia.replace("\n", "<br>")
+        # --- Geração da análise com IA
+        analise_final_ia = gerar_analise_ia(investimentos_finais, capital, patrimonio_final)
 
+        # --- Contexto para o PDF
         context = {
             "capital": capital,
             "grafico": grafico_base64,
@@ -219,16 +258,18 @@ def gerar_pdf():
             "data_hoje": datetime.now().strftime("São Paulo, %d de %B de %Y")
         }
 
+        # --- Renderiza e retorna o PDF
         pdf = render_pdf("relatorio.html", context)
         if pdf:
             return send_file(pdf, mimetype="application/pdf", as_attachment=True, download_name="relatorio_investimento.pdf")
         else:
-            logging.error("Erro ao gerar PDF: pisa.CreatePDF retornou erro")
             return "Erro ao gerar PDF", 500
 
-    except Exception:
-        logging.error("Erro na rota /gerar-pdf", exc_info=True)
-        return "Erro interno no servidor ao gerar PDF. Verifique os logs.", 500
+    except Exception as e:
+        app.logger.error("Erro na rota /gerar-pdf", exc_info=True)
+        return "Erro interno ao gerar o PDF", 500
+
+
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=int(os.environ.get("PORT", 7860)), debug=True)