aniket9909 commited on
Commit
65342e3
·
verified ·
1 Parent(s): 3de55ce

Update pdf_generator.py

Browse files
Files changed (1) hide show
  1. pdf_generator.py +5 -8
pdf_generator.py CHANGED
@@ -6,20 +6,17 @@ def generate_pdf(html_path, output_path="report.pdf"):
6
  browser = pw.chromium.launch()
7
  page = browser.new_page()
8
 
9
- # Absolute path fix for HuggingFace Docker
10
- abs_path = os.path.abspath(html_path)
11
- file_url = f"file:///{abs_path.replace(os.sep, '/')}"
12
 
13
- print("DEBUG: Loading HTML at ->", file_url)
14
-
15
- page.goto(file_url)
16
 
17
  page.pdf(
18
  path=output_path,
19
  format="A4",
20
- print_background=True
21
  )
22
 
23
  browser.close()
24
 
25
- return output_path
 
6
  browser = pw.chromium.launch()
7
  page = browser.new_page()
8
 
9
+ # Normalize path for Playwright
10
+ safe_path = html_path.replace("\\", "/")
 
11
 
12
+ page.goto(f"file:///{safe_path}")
 
 
13
 
14
  page.pdf(
15
  path=output_path,
16
  format="A4",
17
+ print_background=True,
18
  )
19
 
20
  browser.close()
21
 
22
+ return output_path