Update app.py (#8)

- Update app.py (7cf314f1d3eeab6bb66f85a24291c249f0b463a8)


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

app.py

@@ -17,23 +17,34 @@ import ast
         #-----------------#
     
 #função para relatórios em pdf
-def save_results_to_pdf(results_formatados, intervalo_confiança, valores_finais):
+def save_results_to_pdf(results_formatados, intervalo_confiança, valores_finais, image):
     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)
+    story = [
+        Image('avaliase.jpg', width=3.33 * inch, height=1.28 * inch),
+        Spacer(1, 36),
+        Paragraph('Relatório', styles['Heading1']),
+        Spacer(1, 36),
+        Paragraph('Resultados Estatísticos', styles['Heading2'])]
+    
+    lines = results_formatados.splitlines()
+    # Create a paragraph for each line
+    for line in lines:
+        story.append(Paragraph(line, styles['Normal']))
+    story.append(Spacer(1, 36))
+    story.append(Paragraph('Intervalo de Confiança', styles['Heading2']))
+    lines = intervalo_confiança.splitlines()
+    for line in lines:
+        story.append(Paragraph(line, styles['Normal']))
+    story.append(Spacer(1, 36))
+    story.append(Paragraph('Valores Finais', styles['Heading2']))
+    lines = valores_finais.splitlines()
+    for line in lines:
+        story.append(Paragraph(line, styles['Normal']))
+    story.append(Spacer(1, 36))
+    story.append(Image('scatter_plot.png', width=4 * inch, height=4 * inch))
+    
+    doc.build(story)
     
     
         #-----------------#
@@ -903,7 +914,7 @@ def avaliacao_imovel(planilha, num_linhas_desejadas=3, finalidade='Defina o tipo
 
         #-----------------#
 
-        save_results_to_pdf(resultados_formatados, intervalo_confiança, valores_finais)
+        save_results_to_pdf(resultados_formatados, intervalo_confiança, valores_finais, 'scatter_plot.png')
         result.to_excel(output_file, index=False)
         
         #-----------------#