Update app.py

app.py

@@ -1,72 +1,93 @@
 import gradio as gr
-from weasyprint import HTML
-from datetime import datetime
 import tempfile
+import os
+from fpdf import FPDF
+
+def format_brl(value):
+    return f"R$ {value:,.2f}".replace(",", "X").replace(".", ",").replace("X", ".")
 
 def calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto):
     try:
         valor_imovel = float(valor_imovel)
         investimento_proprio = float(investimento_proprio)
         aluguel_mensal = float(aluguel_mensal)
-        anos = int(anos)
+        anos = float(anos)
         taxa_selic = float(taxa_selic) / 100
         desconto = float(desconto) / 100
-    except ValueError:
-        return "Erro: Verifique os dados de entrada."
-
-    aluguel_liquido = aluguel_mensal * (1 - desconto)
-    retorno_total = aluguel_liquido * 12 * anos
-    roi = ((retorno_total - investimento_proprio) / investimento_proprio) * 100
-
-    resultado = (
-        f"Valor do Imóvel: R$ {valor_imovel:,.2f}\n"
-        f"Investimento Próprio: R$ {investimento_proprio:,.2f}\n"
-        f"Aluguel Mensal (líquido): R$ {aluguel_liquido:,.2f}\n"
-        f"Tempo: {anos} anos\n"
-        f"Taxa Selic: {taxa_selic * 100:.2f}%\n"
-        f"ROI estimado: {roi:.2f}%"
-    )
-    return resultado
+
+        aluguel_liquido = aluguel_mensal * (1 - desconto)
+        retorno_total = aluguel_liquido * 12 * anos
+        roi = (retorno_total / investimento_proprio) * 100 if investimento_proprio > 0 else 0
+
+        valor_imovel_fmt = format_brl(valor_imovel)
+        investimento_proprio_fmt = format_brl(investimento_proprio)
+        aluguel_liquido_fmt = format_brl(aluguel_liquido)
+        retorno_fmt = format_brl(retorno_total)
+
+        resultado = (
+            f"🧮 **Cálculo de ROI Imobiliário**\n\n"
+            f"📌 Valor do Imóvel: {valor_imovel_fmt}\n"
+            f"📌 Investimento Próprio: {investimento_proprio_fmt}\n"
+            f"📌 Aluguel Mensal (líquido): {aluguel_liquido_fmt}\n"
+            f"📌 Retorno em {anos:.1f} anos: {retorno_fmt}\n"
+            f"📌 Taxa Selic: {taxa_selic * 100:.2f}%\n"
+            f"✅ ROI estimado: **{roi:.2f}%**"
+        )
+
+        return resultado, ""
+    except Exception as e:
+        return "Erro no cálculo: " + str(e), ""
 
 def gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto):
-    texto = calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto).replace("\n", "<br>")
-    html = f"<h2>Simulação de ROI</h2><p>{texto}</p>"
+    resultado, _ = calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto)
 
-    data = datetime.now().strftime("%d-%m-%Y")
-    filename = f"/tmp/roi-simulacao-{data}.pdf"
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    for linha in resultado.split('\n'):
+        pdf.cell(200, 10, txt=linha.strip(), ln=True)
 
-    HTML(string=html).write_pdf(filename)
-    return filename
+    temp_dir = tempfile.mkdtemp()
+    pdf_path = os.path.join(temp_dir, "roi_calculo.pdf")
+    pdf.output(pdf_path)
+
+    return pdf_path
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 📊 Simulador de ROI de Aluguel com Geração de PDF")
-    with gr.Row():
-        valor_imovel = gr.Textbox(label="Valor do imóvel (R$)", placeholder="Ex: 500000")
-        investimento_proprio = gr.Textbox(label="Investimento próprio (R$)", placeholder="Ex: 100000")
+    gr.Markdown("# 📊 Simulador de ROI de Aluguel com PDF")
     with gr.Row():
-        aluguel_mensal = gr.Textbox(label="Aluguel mensal (R$)", placeholder="Ex: 2500")
-        anos = gr.Textbox(label="Duração (anos)", placeholder="Ex: 5")
+        valor_imovel = gr.Textbox(label="Valor do Imóvel (R$)", placeholder="Ex: 300000")
+        investimento_proprio = gr.Textbox(label="Investimento Próprio (R$)", placeholder="Ex: 100000")
     with gr.Row():
-        taxa_selic = gr.Textbox(label="Taxa SELIC (%)", placeholder="Ex: 10.75")
+        aluguel_mensal = gr.Textbox(label="Aluguel Mensal Bruto (R$)", placeholder="Ex: 2000")
         desconto = gr.Textbox(label="Desconto sobre o aluguel (%)", placeholder="Ex: 10")
     with gr.Row():
-        botao_calcular = gr.Button("📈 Calcular ROI")
-        botao_pdf = gr.Button("📄 Gerar PDF")
-    resultado = gr.Textbox(label="Resultado da Simulação", lines=6)
-    arquivo_pdf = gr.File(label="Download do PDF")
+        anos = gr.Textbox(label="Tempo (em anos)", placeholder="Ex: 5")
+        taxa_selic = gr.Textbox(label="Taxa Selic (%)", placeholder="Ex: 10.75")
+
+    calcular_btn = gr.Button("🔍 Calcular ROI")
+    gerar_pdf_btn = gr.Button("📄 Gerar PDF do Resultado")
+    resultado_output = gr.Markdown()
+    pdf_output = gr.File(label="⬇️ Baixar PDF", interactive=True, visible=False)
 
-    botao_calcular.click(
-        calcular_roi,
+    calcular_btn.click(
+        fn=calcular_roi,
         inputs=[valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto],
-        outputs=resultado
+        outputs=[resultado_output, pdf_output]
     )
-    botao_pdf.click(
-        gerar_pdf,
+
+    gerar_pdf_btn.click(
+        fn=gerar_pdf,
         inputs=[valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto],
-        outputs=arquivo_pdf
+        outputs=pdf_output
+    ).then(
+        lambda: gr.update(visible=True),
+        inputs=[],
+        outputs=pdf_output
     )
 
 demo.launch(server_port=7860)
 
 
 
+