Update app.py

app.py

@@ -2,9 +2,14 @@ import gradio as gr
 from fpdf import FPDF
 import os
 import tempfile
+import matplotlib.pyplot as plt
+import numpy as np
 
 # Função para calcular o ROI
-def calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros):
+def calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, desconto_aluguel):
+    # Desconto sobre o aluguel (custos + administração + impostos)
+    aluguel_mensal_com_desconto = aluguel_mensal * (1 - desconto_aluguel / 100)
+    
     # Calcular valor total do financiamento
     valor_financiado = valor_imovel - investimento_proprio
     
@@ -13,15 +18,20 @@ def calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos,
     numero_parcelas = tempo_anos * 12
     parcela_mensal = valor_financiado * (juros_mensal * (1 + juros_mensal) ** numero_parcelas) / ((1 + juros_mensal) ** numero_parcelas - 1)
     
-    # Calcular ROI
+    # Calcular ROI para o primeiro ano
+    total_pago_1_ano = parcela_mensal * 12 + investimento_proprio
+    lucro_1_ano = aluguel_mensal_com_desconto * 12 - total_pago_1_ano
+    roi_1_ano = (lucro_1_ano / total_pago_1_ano) * 100
+
+    # Calcular ROI geral
     total_pago = parcela_mensal * numero_parcelas + investimento_proprio
-    lucro = aluguel_mensal * numero_parcelas
+    lucro = aluguel_mensal_com_desconto * numero_parcelas
     roi = (lucro - total_pago) / total_pago * 100
 
-    return roi, parcela_mensal, lucro
+    return roi_1_ano, roi, parcela_mensal, lucro_1_ano, lucro
 
 # Função para gerar o PDF
-def gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, roi, parcela_mensal, lucro):
+def gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, roi_1_ano, roi, parcela_mensal, lucro_1_ano, lucro):
     # Criar um PDF com o cálculo do ROI
     pdf = FPDF()
     pdf.add_page()
@@ -37,8 +47,10 @@ def gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, ta
     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"ROI do 1º Ano: {roi_1_ano:.2f}%", ln=True)
+    pdf.cell(200, 10, txt=f"ROI Geral: {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 do 1º Ano: R$ {lucro_1_ano:,.2f}", ln=True)
     pdf.cell(200, 10, txt=f"Lucro Total: R$ {lucro:,.2f}", ln=True)
     
     # Salvar o PDF em um arquivo temporário
@@ -48,23 +60,38 @@ def gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, ta
 
     return pdf_file_path
 
-# Função para o botão "Gerar ROI"
-def calcular_gerar_roi(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)
+# Função para gerar o gráfico
+def gerar_grafico(roi_1_ano, roi):
+    # Gerar gráfico
+    anos = [1, 20]
+    roi_values = [roi_1_ano, roi]
     
-    # Exibir resultado
-    return f"ROI Calculado: {roi:.2f}%\nParcela Mensal: R$ {parcela_mensal:,.2f}\nLucro Total: R$ {lucro:,.2f}"
+    plt.figure(figsize=(6, 4))
+    plt.bar(anos, roi_values, color='blue', alpha=0.7)
+    plt.title('ROI ao Longo do Tempo')
+    plt.xlabel('Ano')
+    plt.ylabel('ROI (%)')
+    plt.xticks(anos, ['1º Ano', 'Geral'])
+    
+    # Salvar gráfico em um arquivo temporário
+    temp_dir = tempfile.gettempdir()
+    grafico_path = os.path.join(temp_dir, "grafico_roi.png")
+    plt.savefig(grafico_path)
+    
+    return grafico_path
 
-# Função para o botão "Gerar PDF"
-def gerar_pdf_arquivo(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros):
+# Função para o botão "Gerar ROI e PDF"
+def calcular_gerar_roi_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, desconto_aluguel):
     # Calcular ROI
-    roi, parcela_mensal, lucro = calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros)
+    roi_1_ano, roi, parcela_mensal, lucro_1_ano, lucro = calcular_roi(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, desconto_aluguel)
     
     # Gerar o PDF com os dados
-    pdf_path = gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, roi, parcela_mensal, lucro)
+    pdf_path = gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, roi_1_ano, roi, parcela_mensal, lucro_1_ano, lucro)
+    
+    # Gerar gráfico
+    grafico_path = gerar_grafico(roi_1_ano, roi)
     
-    return pdf_path
+    return grafico_path, pdf_path
 
 # Interface Gradio com gr.Blocks
 with gr.Blocks() as iface:
@@ -74,25 +101,27 @@ with gr.Blocks() as iface:
     aluguel_mensal = gr.Number(label="Aluguel Mensal (R$)", value=2500)
     tempo_anos = gr.Number(label="Tempo (anos)", value=20)
     taxa_juros = gr.Number(label="Taxa de Juros (%)", value=8)
-    
+    desconto_aluguel = gr.Number(label="Desconto sobre o Aluguel (%)", value=10)
+
     # Resultado do cálculo ROI
     roi_resultado = gr.Textbox(label="Resultado do Cálculo do ROI", lines=3)
     
     # Botões
-    gerar_roi_button = gr.Button("Gerar ROI")
-    gerar_pdf_button = gr.Button("Gerar PDF", elem_id="pdf_button")
+    gerar_button = gr.Button("Gerar ROI e PDF")
     
-    # Funções de callback para os botões
-    gerar_roi_button.click(fn=calcular_gerar_roi, inputs=[valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros], outputs=roi_resultado)
-    pdf_file_output = gerar_pdf_button.click(fn=gerar_pdf_arquivo, inputs=[valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros], outputs=gr.File(label="Baixar PDF"))
+    # Exibição do gráfico
+    grafico_output = gr.Image(label="Gráfico de ROI")
     
-    # Configuração do tema e título
-    iface.title = "Cálculo de ROI de Aluguel"
-    iface.description = "Insira os valores abaixo e calcule o retorno sobre investimento (ROI) de aluguel."
+    # Exibição do link de download
+    pdf_download_button = gr.File(label="Baixar PDF", visible=False)
     
-    # Mostrar link de download como botão após a geração do PDF
-    pdf_file_output.update(visible=True)
+    # Funções de callback para os botões
+    gerar_button.click(fn=calcular_gerar_roi_pdf, inputs=[valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros, desconto_aluguel], outputs=[grafico_output, pdf_download_button])
     
+    # Estilo e descrição
+    iface.title = "Cálculo de ROI de Aluguel"
+    iface.description = "Insira os valores abaixo para calcular o ROI do seu imóvel e gerar o PDF com os resultados."
+
 # Rodar a interface Gradio
 iface.launch(share=True)