Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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:
|