Marthee commited on
Commit
11ce43f
·
verified ·
1 Parent(s): e0bb42f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -99,7 +99,7 @@ def identify_headers_with_openrouter(pdf_path, model,LLM_prompt, pages_to_check=
99
  model=str(model)
100
  toc_pages = get_toc_page_numbers(doc)
101
  lines_for_prompt = []
102
- print(model,LLM_prompt)
103
  # Collect text lines from pages (skip TOC pages)
104
  for pno in range(len(doc)):
105
  if pages_to_check and pno not in pages_to_check:
@@ -128,7 +128,7 @@ def identify_headers_with_openrouter(pdf_path, model,LLM_prompt, pages_to_check=
128
  return []
129
 
130
  prompt = (
131
- LLM_prompt.join(lines_for_prompt)
132
  )
133
 
134
  if not api_key:
 
99
  model=str(model)
100
  toc_pages = get_toc_page_numbers(doc)
101
  lines_for_prompt = []
102
+
103
  # Collect text lines from pages (skip TOC pages)
104
  for pno in range(len(doc)):
105
  if pages_to_check and pno not in pages_to_check:
 
128
  return []
129
 
130
  prompt = (
131
+ LLM_prompt + "\n\nLines:\n" + "\n".join(lines_for_prompt)
132
  )
133
 
134
  if not api_key: