tosin2013 commited on
Commit
7315647
·
verified ·
1 Parent(s): e87b617

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -8
app.py CHANGED
@@ -8,6 +8,7 @@ import base64
8
  import torch
9
  from fpdf import FPDF
10
  import os
 
11
 
12
  # Preload models
13
  models = {
@@ -51,33 +52,43 @@ def highlight_relevant_text(context, start, end):
51
  )
52
  return highlighted_text
53
 
 
 
 
 
 
 
 
 
 
54
  def generate_pdf_report(question, answer, score, score_explanation, score_chart, highlighted_context):
55
  pdf = FPDF()
56
  pdf.add_page()
57
 
58
- # Add the Unicode font (e.g., DejaVuSans)
59
- pdf.add_font("DejaVu", "", os.path.join("fonts", "DejaVuSans.ttf"), uni=True)
60
- pdf.set_font("DejaVu", size=12)
 
61
 
62
  pdf.multi_cell(0, 10, f"Question: {question}")
63
  pdf.ln()
64
 
65
- pdf.set_font("DejaVu", size=12)
66
  pdf.multi_cell(0, 10, f"Answer: {answer}")
67
  pdf.ln()
68
 
69
- pdf.set_font("DejaVu", size=12)
70
  pdf.multi_cell(0, 10, f"Confidence Score: {score}")
71
  pdf.ln()
72
 
73
- pdf.set_font("DejaVu", size=12)
74
  pdf.multi_cell(0, 10, f"Score Explanation: {score_explanation}")
75
  pdf.ln()
76
 
77
- pdf.set_font("DejaVu", size=12)
78
  pdf.multi_cell(0, 10, "Highlighted Context:")
79
  pdf.ln()
80
- pdf.set_font("DejaVu", size=10)
81
  pdf.multi_cell(0, 10, highlighted_context)
82
  pdf.ln()
83
 
 
8
  import torch
9
  from fpdf import FPDF
10
  import os
11
+ import glob
12
 
13
  # Preload models
14
  models = {
 
52
  )
53
  return highlighted_text
54
 
55
+ def find_system_font():
56
+ # Search common directories for TrueType fonts
57
+ font_dirs = ["/usr/share/fonts", "/usr/local/share/fonts"]
58
+ for font_dir in font_dirs:
59
+ ttf_files = glob.glob(os.path.join(font_dir, "**/*.ttf"), recursive=True)
60
+ if ttf_files:
61
+ return ttf_files[0] # Return the first found font
62
+ raise FileNotFoundError("No TTF font file found in system font directories.")
63
+
64
  def generate_pdf_report(question, answer, score, score_explanation, score_chart, highlighted_context):
65
  pdf = FPDF()
66
  pdf.add_page()
67
 
68
+ # Find a system font
69
+ font_path = find_system_font()
70
+ pdf.add_font("SystemFont", "", font_path) # Use the found system font
71
+ pdf.set_font("SystemFont", size=12)
72
 
73
  pdf.multi_cell(0, 10, f"Question: {question}")
74
  pdf.ln()
75
 
76
+ pdf.set_font("SystemFont", size=12)
77
  pdf.multi_cell(0, 10, f"Answer: {answer}")
78
  pdf.ln()
79
 
80
+ pdf.set_font("SystemFont", size=12)
81
  pdf.multi_cell(0, 10, f"Confidence Score: {score}")
82
  pdf.ln()
83
 
84
+ pdf.set_font("SystemFont", size=12)
85
  pdf.multi_cell(0, 10, f"Score Explanation: {score_explanation}")
86
  pdf.ln()
87
 
88
+ pdf.set_font("SystemFont", size=12)
89
  pdf.multi_cell(0, 10, "Highlighted Context:")
90
  pdf.ln()
91
+ pdf.set_font("SystemFont", size=10)
92
  pdf.multi_cell(0, 10, highlighted_context)
93
  pdf.ln()
94