o i@stddlmZmZddlmZddlmZddlZddlZddl Z ddl Z eZ ddZ ddZ dd d Zd d ZdS))requestjsonify) chatgpt_call)PromptsNc CszdtjvrtddddfWStjd}t|dkr'tddddfWSg}|D]}|js1q+|j}d|vrz:t |}d }|j D]}| }|rS||d 7}qE| re| d | |jd n|dt|}| |Wq+ty} z(td |jdt| td |jdt| dddfWYd} ~ WSd} ~ wwd|vrz t|}| |Wq+ty} z(td|jdt| td|jdt| dddfWYd} ~ WSd} ~ wwtd|ddddfWS|s tddddfWSt|} t| t|dddfWStyF} ztdt| tdt| dddfWYd} ~ Sd} ~ ww)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(P/Users/harjaapdhillon/Desktop/newidea/medvista_backend/Services/report_parser.pyparse_medical_reports s             r*c Csz0t|}t}tj||dd|d|}t| d}t ||j }d||j dWSt yE}z t dt |d}~ww) 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}z'|jjjdddd d d d d ddd|idgd gdd}|jdjjWSty\}zt d|dt |d|dt |dWYd}~Sd}~ww)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) s z!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