Update app.py

app.py

@@ -48,20 +48,27 @@ def gerar_pdf(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, ta
 
     return pdf_file_path
 
-# Interface Gradio
-def interface(valor_imovel, investimento_proprio, aluguel_mensal, tempo_anos, taxa_juros):
+# 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)
+    
+    # Exibir resultado
+    return f"ROI Calculado: {roi:.2f}%\nParcela Mensal: R$ {parcela_mensal:,.2f}\nLucro Total: R$ {lucro:,.2f}"
+
+# Função para o botão "Gerar PDF"
+def gerar_pdf_arquivo(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
+    # Gerar o PDF com os dados
     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
+    return pdf_path
 
-# Definir a interface do Gradio com estilos
+# Interface Gradio
 iface = gr.Interface(
-    fn=interface,
+    fn=calcular_gerar_roi,
     inputs=[
         gr.Number(label="Valor do Imóvel (R$)", value=500000),
         gr.Number(label="Investimento Próprio (R$)", value=100000),
@@ -71,35 +78,26 @@ iface = gr.Interface(
     ],
     outputs=[
         gr.Textbox(label="Resultado do Cálculo do ROI", lines=3),
-        gr.File(label="Baixar PDF", elem_id="download-pdf")
     ],
     live=True,
     title="Cálculo de ROI de Aluguel",
-    description="Insira os valores abaixo e calcule o retorno sobre investimento (ROI) de aluguel.",
-    css="""
-        .gr-button {
-            background-color: #4CAF50;
-            color: white;
-            font-size: 16px;
-            border-radius: 5px;
-        }
-        .gr-button:focus {
-            background-color: #45a049;
-        }
-        #download-pdf {
-            display: inline-block;
-            margin-top: 20px;
-            padding: 10px;
-            background-color: #007bff;
-            color: white;
-            font-size: 16px;
-            border-radius: 5px;
-            text-align: center;
-        }
-        #download-pdf:hover {
-            background-color: #0056b3;
-        }
-    """
+    description="Insira os valores abaixo e calcule o retorno sobre investimento (ROI) de aluguel."
+)
+
+# Adicionando o botão de "Gerar PDF"
+iface.add_button(
+    "Gerar PDF",
+    fn=gerar_pdf_arquivo,
+    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.File(label="Baixar PDF")
+    ]
 )
 
 # Rodar a interface Gradio
@@ -107,4 +105,3 @@ iface.launch(share=True)
 
 
 
-