lasse/pdf-highlighter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor single_pdf.py to remove unused variables and prompt user for input

Browse Source
This commit is contained in:
lasseedfast 2024-10-08 10:53:26 +02:00
parent c845d327b6
commit e53659fe7d
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/single_pdf.py
View File

@ -2,12 +2,6 @@ import asyncio
import io import io
from highlight_pdf import Highlighter from highlight_pdf import Highlighter
# User input/question
user_input = "What are the main findings?"
# Answer received from LLM based on text in a PDF
llm_answer = "The main findings are that the treatment was effective in 70% of cases."
# PDF filename # PDF filename
pdf_filename = "example_pdf_document.pdf" pdf_filename = "example_pdf_document.pdf"
@ -23,8 +17,8 @@ highlighter = Highlighter(
# Define the main asynchronous function to highlight the PDF # Define the main asynchronous function to highlight the PDF
async def main(): async def main():
highlighted_pdf_buffer = await highlighter.highlight( highlighted_pdf_buffer = await highlighter.highlight(
user_input=user_input, user_input=input('User input: '),
data=[{"text": llm_answer, "pdf_filename": pdf_filename, "pages": pages}] pdf_filename=pdf_filename,
) )
# Save the highlighted PDF to a new file # Save the highlighted PDF to a new file