Update app.py

app.py

@@ -6,8 +6,9 @@ import gradio as gr
 from gradio import components
 from gradio import Interface
 import xlsxwriter
+from reportlab.lib.units import inch
 from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph
+from reportlab.platypus import SimpleDocTemplate, Spacer, Image, Frame, PageTemplate, BaseDocTemplate, Table, Paragraph, NextPageTemplate, PageBreak
 from reportlab.lib.styles import getSampleStyleSheet
 import shutil
 import os
@@ -17,24 +18,35 @@ 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)
+   
     
         #-----------------#
 
@@ -899,7 +911,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)
         
         #-----------------#