import os
os.environ['FONTCONFIG_PATH'] = '/tmp/fontconfig'
os.makedirs('/tmp/fontconfig', exist_ok=True)

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 matplotlib.ticker import FuncFormatter
from xhtml2pdf import pisa
from datetime import datetime

app = Flask(__name__)

def formatar_brl(valor):
    return f"R$ {valor:,.2f}".replace(",", "X").replace(".", ",").replace("X", ".")

def gerar_analise(investimentos_finais, capital):
    melhor = max(investimentos_finais, key=investimentos_finais.get)
    valor_melhor = investimentos_finais[melhor]
    retorno_pct = ((valor_melhor - capital) / capital) * 100
    analise = f"""
    Após análise dos cenários projetados para 5 anos, o investimento <strong>{melhor}</strong> apresenta o melhor desempenho,
    com um valor final estimado de <strong>{formatar_brl(valor_melhor)}</strong>, equivalente a um retorno de <strong>{retorno_pct:.1f}%</strong>.
    """
    return analise

def gerar_consideracoes_finais(investimentos_finais, capital):
    melhor = max(investimentos_finais, key=investimentos_finais.get)
    valor_melhor = investimentos_finais[melhor]
    retorno_pct = ((valor_melhor - capital) / capital) * 100

    texto = f"""
    <p>Após análise dos cenários projetados para 5 anos, o investimento <strong>{melhor}</strong> apresenta o melhor desempenho,
    com um valor final estimado de <strong>{formatar_brl(valor_melhor)}</strong>, equivalente a um retorno significativo sobre o capital inicial.</p>
    <p><strong>Valor Patrimonial:</strong> corresponde ao valor contábil do investimento, conforme normas contábeis e práticas de investidores profissionais,
    representando o patrimônio acumulado e a valorização dos ativos ao longo do tempo.</p>
    """
    return texto

@app.route("/", methods=["GET", "POST"])
def index():
    if request.method == "POST":
        # Captura os 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"])

        anos = list(range(1, 6))

        # Calcula evolução Studio
        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)]

        # Outras opções
        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 dataframe para tabela e relatório
        dados = {
            "Ano": anos,
            "Studio (Patrimônio + Renda)": studio_total,
            "Franquia": franquia,
            "Ações": acoes,
            "Renda Fixa": renda_fixa_valores
        }
        df = pd.DataFrame(dados)

        investimentos_finais = {
            "Studio": studio_total[-1],
            "Franquia": franquia[-1],
            "Ações": acoes[-1],
            "Renda Fixa": renda_fixa_valores[-1],
        }

        investimento_mais_valorizado = max(investimentos_finais, key=investimentos_finais.get)
        valor_mais_alto = investimentos_finais[investimento_mais_valorizado]

        # Formatar dataframe para exibir valores em BRL
        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="table table-striped table-sm", border=0)

        analise_final = gerar_analise(investimentos_finais, capital)
        consideracoes = gerar_consideracoes_finais(investimentos_finais, capital)

        # Gerar gráfico base64 para exibir na página HTML
        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")
        plt.xlabel("Ano")
        plt.ylabel("Valor (R$)")
        plt.legend()
        plt.grid(True)
        plt.gca().yaxis.set_major_formatter(FuncFormatter(lambda x, _: f"R${x:,.0f}".replace(",", ".")))
        plt.tight_layout()
        buf = io.BytesIO()
        plt.savefig(buf, format="png")
        buf.seek(0)
        grafico_base64 = base64.b64encode(buf.getvalue()).decode("utf-8")
        buf.close()
        plt.close()

        return render_template(
            "index.html",
            capital=capital,
            studio_ret=studio_ret,
            valorizacao=valorizacao,
            franquia_ret=franquia_ret,
            acoes_ret=acoes_ret,
            renda_fixa=renda_fixa,
            inflacao=inflacao,
            anos=anos,
            tabela=tabela_html,
            grafico=grafico_base64,
            analise_final=analise_final,
            consideracoes=consideracoes,
            investimento_mais_valorizado=investimento_mais_valorizado,
            valor_mais_alto=valor_mais_alto
        )

    return render_template("index.html")


@app.route("/download_pdf", methods=["POST"])
def download_pdf():
    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"])

    anos = list(range(1, 6))

    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)]

    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]

    dados = {
        "Ano": anos,
        "Studio (Patrimônio + Renda)": studio_total,
        "Franquia": franquia,
        "Ações": acoes,
        "Renda Fixa": renda_fixa_valores
    }
    df = pd.DataFrame(dados)

    investimentos_finais = {
        "Studio": studio_total[-1],
        "Franquia": franquia[-1],
        "Ações": acoes[-1],
        "Renda Fixa": renda_fixa_valores[-1],
    }
    investimento_mais_valorizado = max(investimentos_finais, key=investimentos_finais.get)
    valor_mais_alto = investimentos_finais[investimento_mais_valorizado]

    # Gerar gráfico PNG
    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.gca().yaxis.set_major_formatter(FuncFormatter(lambda x, _: f"R${x:,.0f}".replace(",", ".")))
    plt.tight_layout()

    img_buf = io.BytesIO()
    plt.savefig(img_buf, format='png')
    img_buf.seek(0)
    img_base64 = base64.b64encode(img_buf.getvalue()).decode("utf-8")
    img_buf.close()
    plt.close()

    # Formatar dataframe para relatório PDF (com 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, border=1, classes="table")

    analise = gerar_analise(investimentos_finais, capital)
    consideracoes = gerar_consideracoes_finais(investimentos_finais, capital)

    html = f"""
    <html>
    <head>
    <meta charset="utf-8" />
    <style>
        body {{
            font-family: Arial, sans-serif;
            margin: 40px;
            font-size: 12pt;
            color: #333;
        }}
        h1 {{
            text-align: center;
            color: #2c3e50;
        }}
        h2 {{
            color: #2c3e50;
            border-bottom: 1px solid #ccc;
            padding-bottom: 5px;
        }}
        p {{
            text-align: justify;
            line-height: 1.4;
        }}
        .analise {{
            background: #e8f5e9;
            border-left: 6px solid #2e7d32;
            padding: 10px;
            margin: 15px 0;
            font-size: 11pt;
        }}
        img {{
            display: block;
            margin: 20px auto;
            max-width: 100%;
            height: auto;
        }}
        table {{
            width: 100%;
            border-collapse: collapse;
            margin-top: 20px;
            font-size: 11pt;
        }}
        th, td {{
            border: 1px solid #ccc;
            padding: 8px;
            text-align: right;
        }}
        th {{
            background-color: #f0f0f0;
            color: #2c3e50;
        }}
        th:first-child, td:first-child {{
            text-align: center;
        }}
    </style>
    </head>
    <body>
        <h1>Relatório de Simulação de Investimentos</h1>
        <p><strong>Data da Simulação:</strong> {datetime.today().strftime('%d/%m/%Y')}</p>
        <p><strong>Capital Inicial:</strong> {formatar_brl(capital)}</p>

        <div class="analise">{analise}</div>
        <div class="analise">{consideracoes}</div>

        <h2>Gráfico de Evolução dos Investimentos</h2>
        <img src="data:image/png;base64,{img_base64}" alt="Gráfico de Investimentos" />

        <h2>Evolução Ano a Ano</h2>
        {tabela_html}
    </body>
    </html>
    """

    pdf_buffer = io.BytesIO()
    pisa_status = pisa.CreatePDF(io.StringIO(html), dest=pdf_buffer)
    if pisa_status.err:
        return "Erro ao gerar PDF", 500

    pdf_buffer.seek(0)
    return send_file(
        pdf_buffer,
        mimetype="application/pdf",
        as_attachment=True,
        download_name="relatorio_investimentos.pdf"
    )


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=7860, debug=True)