U t75i@stddlmZmZddlmZddlmZddlZddlZddl Z ddl Z eZ ddZ ddZ dd d Zd d ZdS))requestjsonify) chatgpt_call)PromptsNc CszJdtjkr"tddddfWStjd}t|dkrNtddddfWSg}|D]}|jsdqV|j}d|kr\ztt |}d }|j D]}| }|r||d 7}q| r| d | |jd n|dt|}| |Wnltk rX} zLtd |jdt| td |jdt| dddfWYWSd} ~ XYnXqVd|krzt|}| |Wnltk r} zLtd|jdt| td|jdt| dddfWYWSd} ~ XYnXqVtd|ddddfWSqV|s,tddddfWSt|} t| t|dddfWStk r} z6tdt| tdt| dddfWYSd} ~ XYnXdS)z Accepts multiple files (PDF, images) and extracts medical data using AI Returns structured JSON - NO database storage (privacy-first) fileszNo files uploadedtrue)messageerrorirzNo files providedpdf Zpdf_texttypecontentfilenamezError processing PDF : iNimagezError processing image zUnsupported file type: z$. Only PDF and images are supported.z2No data could be extracted from the uploaded filesF)dataZfiles_processedr z Error in parse_medical_reports: zError parsing reports: )rrrgetlistlenr content_typelowerPyPDF2Z PdfReaderZpagesZ extract_textstripappendseekprocess_image_file Exceptionprintstr parse_with_ai) rextracted_datafileZ file_typeZ pdf_readertextpageZ page_textZ image_resulte parsed_datar(4/var/www/html/api-medvista/Services/report_parser.pyparse_medical_reports s        ""   r*c Csz`t|}t}tj||dd|d|}t| d}t ||j }d||j dWSt k r}zt dt |W5d}~XYnXdS) zj Process image file using imageio Reads image, converts to JPEG format, and encodes to base64 ZJPEG)formatrzutf-8rr z%Error processing image with imageio: N)imageioZimreadioBytesIOZimwriterreadbase64 b64encodedecodeparse_image_with_ai_visionrrr )r#rZ img_byte_arrZ image_bytes image_base64Zextracted_textr&r(r(r)ris   rr c Csddlm}|ddd}zN|jjjdddd d d d d ddd|idgd gdd}|jdjjWStk r}z6t d|dt |d|dt |dWYSd}~XYnXdS)za Use AI vision to extract text from medical images Works with DeepSeek vision models r)OpenAIz#sk-e108ff73c53144379d0b8a4aefbaf525zhttps://api.deepseek.com)api_keybase_urlz deepseek-chatsystemzYou are a medical document OCR expert. Extract ALL text and data from medical reports accurately, preserving the structure and format.)roleruserr$aNExtract all medical information from this report image. Include: - Test names and their values - Units and reference ranges - Doctor's notes and observations - Patient information (if visible) - Any abnormal findings or flags - Date of report - Laboratory name Format the output as clear, structured text, preserving the original layout.)rr$ image_urlurlzdata:image/jpeg;base64,)rr;i)modelmessages max_tokenszVision API failed for rz[Image data from z - vision processing failed: ]N) openair5chat completionscreatechoicesrrrrr )r4rr5clientresponser&r(r(r)r3s: #r3cCs2d}|dd|D}t|}t|dd}|S)z Send all extracted data to AI for structured parsing Combines text from all files and creates structured JSON output z ================================================================================ === NEW DOCUMENT === ================================================================================ c Ss.g|]&}d|dd|dd|dqS)u 📄 File: ru 📋 Type: ru 📝 Content: rr().0itemr(r(r) sz!parse_with_ai..i )r?)joinpromptsr*r)r" separatorZcombined_contentpromptr'r(r(r)r!s   r!)r )flaskrrServices.ai_workerrServices.promptsrrr0r,r-rLr*rr3r!r(r(r(r)s  ]# 8