Update app.py (#2)

- Update app.py (e493d5c0041f5c57340d2559f04ab79edd6fb8a7)


Co-authored-by: Fernando Schwartzer <fschwartzer@users.noreply.huggingface.co>

app.py

@@ -5,6 +5,33 @@ import gradio as gr
 from gradio import components
 from gradio import Interface
 import xlsxwriter
+from reportlab.lib.pagesizes import letter
+from reportlab.platypus import SimpleDocTemplate, Paragraph
+from reportlab.lib.styles import getSampleStyleSheet
+import shutil
+
+# Function to save results in a PDF file
+def save_results_to_pdf(results_formatados, intervalo_confiança, valores_finais):
+    doc = SimpleDocTemplate("resultados.pdf", pagesize=letter)
+    styles = getSampleStyleSheet()
+    
+    # Create a list of elements to include in the PDF
+    elements = []
+    
+    # Add the formatted results to the PDF
+    formatted_results = Paragraph(results_formatados, styles["Normal"])
+    elements.append(formatted_results)
+    
+    # Add the intervalo de confianca to the PDF
+    confianca = Paragraph(intervalo_confiança, styles["Normal"])
+    elements.append(confianca)
+    
+    # Add the valores calculados to the PDF
+    calculados = Paragraph(valores_finais, styles["Normal"])
+    elements.append(calculados)
+    
+    # Build the PDF
+    doc.build(elements)
 
 def renderizar_dataframe(df):
     try:
@@ -444,8 +471,12 @@ def avaliacao_imovel(planilha, num_linhas_desejadas=10):
 
         #-----------------#
 
+        save_results_to_pdf(resultados_formatados, intervalo_confiança, valores_finais)
+        
+        #-----------------#
+
         # Retorna tanto a planilha quanto os resultados formatados
-        return 'relatório.xlsx', result_render, resultados_formatados, intervalo_confiança, valores_finais
+        return 'relatório.xlsx', 'resultados.pdf', result_render, resultados_formatados, intervalo_confiança, valores_finais
         
     
 # Interface do Gradio com input como arquivo XLS ou XLSX
@@ -457,6 +488,7 @@ interface = gr.Interface(
     ],
     outputs=[
         gr.components.File(label="Download planilha"),
+        gr.components.File(label="Download Relatório em PDF"),
         gr.outputs.HTML(label="Resultado Renderizado"),
         gr.components.Textbox(label="Resultados estatísticos"),
         gr.components.Textbox(label="Intervalo de confiança de 80%"),
@@ -471,4 +503,4 @@ interface = gr.Interface(
 
 # Executar o aplicativo Gradio
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(debug=True)