Update app.py

app.py

@@ -1,92 +1,84 @@
 import gradio as gr
-import tempfile
-import os
 from fpdf import FPDF
+import os
+import tempfile
 
-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 = float(anos)
-        taxa_selic = float(taxa_selic) / 100
-        desconto = float(desconto) / 100
-
-        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):
-    resultado, _ = calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto)
-
+# Função para calcular o ROI
+def calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros):
+    # Calcular valor total do financiamento
+    valor_financiado = valor_imovel - investimento_proprio
+    
+    # Calcular parcelas mensais do financiamento
+    juros_mensal = (taxa_juros / 100) / 12
+    numero_parcelas = tempo_anos * 12
+    parcela_mensal = valor_financiado * (juros_mensal * (1 + juros_mensal) ** numero_parcelas) / ((1 + juros_mensal) ** numero_parcelas - 1)
+    
+    # Calcular ROI
+    total_pago = parcela_mensal * numero_parcelas + investimento_proprio
+    lucro = aluguel_mensal * numero_parcelas
+    roi = (lucro - total_pago) / total_pago * 100
+
+    return roi, parcela_mensal, lucro
+
+# Função para gerar o PDF
+def gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, roi, parcela_mensal, lucro):
+    # Criar um PDF com o cálculo do ROI
     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)
-
-    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 PDF")
-    with gr.Row():
-        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():
-        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():
-        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)
-
-    calcular_btn.click(
-        fn=calcular_roi,
-        inputs=[valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto],
-        outputs=[resultado_output, pdf_output]
-    )
-
-    gerar_pdf_btn.click(
-        fn=gerar_pdf,
-        inputs=[valor_imovel, investimento_proprio, aluguel_mensal, anos, taxa_selic, desconto],
-        outputs=pdf_output
-    ).then(
-        lambda: gr.update(visible=True),
-        inputs=[],
-        outputs=pdf_output
-    )
+    
+    # Título
+    pdf.cell(200, 10, txt="Cálculo de ROI de Aluguel", ln=True, align='C')
+    pdf.ln(10)
+    
+    # Adicionar os dados no PDF
+    pdf.cell(200, 10, txt=f"Valor do Imóvel: R$ {valor_imovel:,.2f}", ln=True)
+    pdf.cell(200, 10, txt=f"Investimento Próprio: R$ {investimento_proprio:,.2f}", ln=True)
+    pdf.cell(200, 10, txt=f"Aluguel Mensal: R$ {aluguel_mensal:,.2f}", ln=True)
+    pdf.cell(200, 10, txt=f"Tempo (anos): {tempo_anos} anos", ln=True)
+    pdf.cell(200, 10, txt=f"Taxa de Juros: {taxa_juros}%", ln=True)
+    pdf.cell(200, 10, txt=f"ROI Calculado: {roi:.2f}%", ln=True)
+    pdf.cell(200, 10, txt=f"Parcela Mensal do Financiamento: R$ {parcela_mensal:,.2f}", ln=True)
+    pdf.cell(200, 10, txt=f"Lucro Total: R$ {lucro:,.2f}", ln=True)
+    
+    # Salvar o PDF em um arquivo temporário
+    temp_dir = tempfile.gettempdir()
+    pdf_file_path = os.path.join(temp_dir, "resultado_roi.pdf")
+    pdf.output(pdf_file_path)
+
+    return pdf_file_path
+
+# Interface Gradio
+def interface(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros):
+    # Calcular ROI
+    roi, parcela_mensal, lucro = calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros)
+    
+    # Gerar PDF
+    pdf_path = gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, roi, parcela_mensal, lucro)
+    
+    # Retornar o link de download
+    return f"Aqui está o seu cálculo de ROI. Clique no link abaixo para baixar o PDF:", pdf_path
+
+# Definir a interface do Gradio
+iface = gr.Interface(
+    fn=interface,
+    inputs=[
+        gr.Number(label="Valor do Imóvel (R$)", value=500000),
+        gr.Number(label="Investimento Próprio (R$)", value=100000),
+        gr.Number(label="Aluguel Mensal (R$)", value=2500),
+        gr.Number(label="Tempo (anos)", value=20),
+        gr.Number(label="Taxa de Juros (%)", value=8)
+    ],
+    outputs=[
+        gr.Textbox(label="Resultado do Cálculo do ROI", lines=3),
+        gr.File(label="Baixar PDF")
+    ],
+    live=True
+)
+
+# Rodar a interface Gradio
+iface.launch(share=True)
 
-demo.launch(server_port=7860)