29 lines
771 B
Python
29 lines
771 B
Python
import asyncio
|
|
import io
|
|
from highlight_pdf import Highlighter
|
|
|
|
# PDF filename
|
|
pdf_filename = "example_pdf_document.pdf"
|
|
|
|
# Pages to consider (optional, can be None)
|
|
pages = [1, 2]
|
|
|
|
# Initialize the Highlighter
|
|
highlighter = Highlighter(
|
|
model='llama3.1',
|
|
comment=True # Enable comments to understand the context
|
|
)
|
|
|
|
# Define the main asynchronous function to highlight the PDF
|
|
async def main():
|
|
highlighted_pdf_buffer = await highlighter.highlight(
|
|
user_input=input('User input: '),
|
|
pdf_filename=pdf_filename,
|
|
)
|
|
|
|
# Save the highlighted PDF to a new file
|
|
with open("highlighted_example_pdf_document.pdf", "wb") as f:
|
|
f.write(highlighted_pdf_buffer.getbuffer())
|
|
|
|
# Run the main function using asyncio
|
|
asyncio.run(main()) |