100 lines
11 KiB
Plaintext
100 lines
11 KiB
Plaintext
|
§
|
|||
|
|
Æ•bhõ ã óˆ — d dl Z d dlZd dlZd dlmZ d dlZd dlmZmZm Z m
|
|||
|
|
Z
|
|||
|
|
d dl Z d dl Z d dl
|
|||
|
|
Z
|
|||
|
|
d dlmZ G d„ d¦ « ZdS )é N)ÚImage)ÚDictÚAnyÚListÚOptional)Údatetimec óì — e Zd Zd„ Zdededeeef fd„Zdedeeef fd„Zdedefd„Z d edeeef fd
|
|||
|
|
„Z
|
|||
|
|
d edefd „Z d edeeef fd
|
|||
|
|
„Z dedeeef fd„Z
|
|||
|
|
dededefd„ZdS )ÚDocumentProcessorc ó\ — t j t j ¬¦ « | _ d| _ d S )N)Úapi_keyz)meta-llama/llama-4-scout-17b-16e-instruct)ÚgroqÚGroqÚconfigÚ GROQ_API_KEYÚclientÚmodel)Úselfs ú0/Users/user/mkd/quickbooks/document_processor.pyÚ__init__zDocumentProcessor.__init__
|
|||
|
|
s$ € Ý”iÔ(;Ð<Ñ<Ô<ˆŒ Ø@ˆŒ
|
|||
|
|
ˆ
|
|||
|
|
ˆ
|
|||
|
|
ó Ú file_pathÚ file_typeÚreturnc ƒ óD K — | ¦ « dv r| |¦ « ƒ d{V —†S | ¦ « dk r| |¦ « ƒ d{V —†S t d|› �¦ « ‚# t $ r}dt |¦ « icY d}~S d}~ww xY w)z.Process uploaded file and extract receipt data)ÚjpgÚjpegÚpngÚgifÚbmpNÚpdfzUnsupported file type: Úerror)ÚlowerÚ_process_imageÚ _process_pdfÚ
|
|||
|
|
ValueErrorÚ ExceptionÚstr)r r r Úes r Ú process_filez DocumentProcessor.process_file sØ è è € ð %Ø�ŠÑ Ô Ð$HÐHÐHØ !× 0Ò 0°Ñ ;Ô ;Ð;Ð;Ð;Ð;Ð;Ð;Ð;Ø— ’ Ñ"Ô" eÒ+Ð+Ø !× .Ò .¨yÑ 9Ô 9Ð9Ð9Ð9Ð9Ð9Ð9Ð9å Ð!F¸9Ð!FÐ!FÑGÔGÐGøÝð %ð %ð %Ø�S ™VœVÐ$Ð $Ð $Ð $Ð $Ð $Ð $øøøøð %øøøs( „0A: µ2A: Á(A: Á:
|
|||
|
|
B ÂBÂB ÂB Ú
|
|||
|
|
image_pathc ƒ ó„ K — | |¦ « }d}| j j j dd|dœddd|› �idœgd œg| j d
|
|||
|
|
d ¬ ¦ « }|j d
|
|||
|
|
j j ¦ « }|
|
|||
|
|
|¦ « S # t $ r }ddt |¦ « › �icY d}~S d}~ww xY w)z)Extract data from image using Groq visiona^
|
|||
|
|
Analyze this receipt image and extract the following information in JSON format:
|
|||
|
|
{
|
|||
|
|
"vendor": "Store/company name",
|
|||
|
|
"total_amount": 0.00,
|
|||
|
|
"tax_amount": 0.00,
|
|||
|
|
"date": "YYYY-MM-DD",
|
|||
|
|
"category": "Food/Transport/Office/Other",
|
|||
|
|
"confidence": 0.95
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Rules:
|
|||
|
|
- Extract vendor name as it appears on receipt
|
|||
|
|
- Total amount should be the final total including tax
|
|||
|
|
- Tax amount is separate tax line if available
|
|||
|
|
- Date should be the date on the receipt
|
|||
|
|
- Categorize based on vendor type (Starbucks=Food, Shell=Transport, etc.)
|
|||
|
|
- Confidence score 0-1 based on how clear the receipt is
|
|||
|
|
|
|||
|
|
Return only valid JSON.
|
|||
|
|
ÚuserÚtext)Útyper- Ú image_urlÚurlzdata:image/jpeg;base64,)r. r/ ©ÚroleÚcontentéô çš™™™™™¹?)Úmessagesr Ú
|
|||
|
|
max_tokensÚ temperaturer r! zImage processing error: N)
|
|||
|
|
Ú
|
|||
|
|
_encode_imager ÚchatÚ completionsÚcreater ÚchoicesÚmessager3 ÚstripÚ_parse_extraction_resultr& r' )r r* Ú base64_imageÚpromptÚresponseÚ result_textr( s r r# z DocumentProcessor._process_image s è è € ð5 Bà ×-Ò-¨jÑ9Ô9ˆLðˆFð. ”{Ô'Ô3×:Ò:ð !'à%+°VÐ <Ð <à(3à$)Ð+SÀ\Ð+SÐ+Sð."ð ð ð$ð ð ð
|
|||
|
|
ð ”jØ Ø ð# ;ñ ô ˆHð* #Ô*¨1Ô-Ô5Ô=×CÒCÑEÔEˆKØ×0Ò0°Ñ=Ô=Ð =øåð Bð Bð BØÐ @½¸A¹¼Ð @Ð @ÐAÐ AÐ AÐ AÐ AÐ AÐ Aøøøøð Bøøøs „BB Â
|
|||
|
|
B?Â!B:Â4B?Â:B?c óÈ — t |d¦ « 5 }t j | ¦ « ¦ « d¦ « cddd¦ « S # 1 swxY w Y dS )z Encode image to base64 stringÚrbzutf-8N)ÚopenÚbase64Ú b64encodeÚreadÚdecode)r r* Ú
|
|||
|
|
image_files r r9 z DocumentProcessor._encode_imageV s¬ € å
|
|||
|
|
�*˜dÑ
|
|||
|
|
#Ô
|
|||
|
|
#ð G zÝÔ# J§O¢OÑ$5Ô$5Ñ6Ô6×=Ò=¸gÑFÔFð Gð Gð Gð Gñ Gô Gð Gð Gð Gð Gð Gð Gøøøð Gð Gð Gð Gð Gð Gs ‘9AÁAÁ AÚpdf_pathc ƒ ó² K — | |¦ « }| |¦ « S # t $ r }ddt |¦ « › �icY d}~S d}~ww xY w)z2Extract data from PDF by converting to image firstr! zPDF processing error: N)Ú_extract_text_from_pdfÚ_process_text_contentr& r' )r rM Ú text_contentr( s r r$ z DocumentProcessor._process_pdf[ s è è € ð @à ×6Ò6°xÑ@Ô@ˆLØ×-Ò-¨lÑ;Ô;Ð ;øåð @ð @ð @ØÐ >µc¸!±f´fÐ >Ð >Ð?Ð ?Ð ?Ð ?Ð ?Ð ?Ð ?øøøøð @øøøs „). ®
|
|||
|
|
A¸AÁ AÁAc ó — t |d¦ « 5 }t j |¦ « }d}|j D ] }|| ¦ « dz z
|
|||
|
|
}Œ |cddd¦ « S # 1 swxY w Y dS # t
|
|||
|
|
$ r }Y d}~dS d}~ww xY w)zExtract text from PDFrF Ú ú
|
|||
|
|
N)rG ÚPyPDF2Ú PdfReaderÚpagesÚ extract_textr& )r rM ÚfileÚ
|
|||
|
|
pdf_readerr- Úpager( s r rO z(DocumentProcessor._extract_text_from_pdfe sí € ð Ý�h Ñ%Ô%ð
|
|||
|
|
¨Ý #Ô -¨dÑ 3Ô 3�
|
|||
|
|
Ø�Ø &Ô ,ð 7ð 7�DؘD× -Ò -Ñ /Ô /°$Ñ 6Ñ6�D�DØð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ñ
|
|||
|
|
ô
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
øøøð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
ð
|
|||
|
|
øõ ð ð ð Ø�2�2�2�2�2øøøøð øøøs4 ‚A( ’<AÁ A( ÁA Á A( Á"A Á#A( Á(
|
|||
|
|
A=Á8A=rQ c óD — d|› d�}| j j j | j d|dœgdd¬¦ « }|j d j j ¦ « }| |¦ « S # t $ r }d d
|
|||
|
|
t |¦ « › �icY d }~S d }~ww xY w) z3Process text content using Groq (fallback for PDFs)z�
|
|||
|
|
Analyze this receipt text and extract the following information in JSON format:
|
|||
|
|
|
|||
|
|
Receipt Text:
|
|||
|
|
aê
|
|||
|
|
|
|||
|
|
Extract:
|
|||
|
|
{
|
|||
|
|
"vendor": "Store/company name",
|
|||
|
|
"total_amount": 0.00,
|
|||
|
|
"tax_amount": 0.00,
|
|||
|
|
"date": "YYYY-MM-DD",
|
|||
|
|
"category": "Food/Transport/Office/Other",
|
|||
|
|
"confidence": 0.95
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Rules:
|
|||
|
|
- Extract vendor name as it appears on receipt
|
|||
|
|
- Total amount should be the final total including tax
|
|||
|
|
- Tax amount is separate tax line if available
|
|||
|
|
- Date should be the date on the receipt
|
|||
|
|
- Categorize based on vendor type
|
|||
|
|
- Confidence score 0-1 based on clarity
|
|||
|
|
|
|||
|
|
Return only valid JSON.
|