o i9@sddlmZmZddlmZddlmZddlmZddl m Z ddl Z eZ ddZ d d Zd d Zd dZddZddZdS))requestjsonify)connect_to_database)Prompts) chatgpt_call)closingNcCstjdd}|r3zt|}td|Wnty2}ztd|d}WYd}~nd}~wwt}|sAtddddfS|j d d }z | d |f| }| d |f| }|d dkr| d|f| }i}|D] } | d|| d<qt| ||t|dddfWS| d|f| } | s||tddddfWS| d} | d} | d} | d}| d}| d}| d}| d}| d }| d!}| d"}| d#}| d$}| d%}| d&}| d'}d(d)|D}tj| | ||||||||||||||d*}t|}|D] }| d+| |||fq| d|f| }i}|D] } | d|| d<q3| ||t|dddfWSty}z%td,t|||td-t|dddfWYd}~Sd}~ww).z] Get diagnostics for a visit - supports report context from uploaded medical reports report_contextNz.Report context received for diagnostics visit zError parsing report_context: zDatabase connection errortruemessageerrorT dictionaryaa SELECT pc.possible_causes AS possible_causes FROM possible_causes pc RIGHT JOIN possible_causes_by_doctor pcd ON pc.visits_id = pcd.visits_id AND pc.id = pcd.possible_causes WHERE pc.visits_id = %s; zESELECT COUNT(*) AS total_count FROM diagnostics WHERE visits_id = %s; total_countrz/SELECT * FROM diagnostics WHERE visits_id = %s; diagnosticsidFdatar a SELECT p.id as patient_id, p.fullname AS patient_name, p.gender, p.age, p.fk_doctor, p.allergies, p.medications, v.reasonforvisit AS reasonForConsultation, v.height, v.weight, v.symptoms, v.glucoselevel AS glucoseLevel, v.bloodpressure AS bp, p.tobaccouse AS tobaccoUse, p.alcoholconsumption AS alcoholConsumption, p.dietandexercisehabits AS dietAndExerciseHabits FROM patients p JOIN visits v ON p.id = v.fk_patients WHERE v.id = %s; zPatient not found patient_id patient_namegenderage fk_doctor allergies medicationsreasonForConsultationheightweight glucoseLevelbp tobaccoUsesymptomsalcoholConsumptiondietAndExerciseHabitscSsg|]}|dqS)possible_causes).0itemr(r(N/Users/harjaapdhillon/Desktop/newidea/medvista_backend/Services/Diagnostics.py zsz#get_diagnostics..)rz INSERT INTO diagnostics (patient_id, visits_id, diagnostics, doctor_id) VALUES (%s, %s, %s, %s); zError in get_diagnostics: Internal server error: )rargsgetjsonloadsprint Exceptionrrcursorexecutefetchallfetchonecommitclosepromptsget_diagnosticsrstr)visit_idre connectionr4r'resultrowsfiltered_resultsrowaccountrrrr doctor_idrrrrr r!r"r#patientSymptomsr%r&possibleCausesByDoctorpromptrZ diagnosticr(r(r+r; s      (r;c Cs0t}|d}|d}|d}|d}gd}|D]}||vs(||s6td|ddd fSqt}zzt|}|d |||fd d |D} t || } | slt d |d|d|dnR|d|f|d|f|d|f|d|f|d|f|dd|f|dd|f|dd|f|dd|f|D] } |d|||| fq| Wdn1swYWn1t y} z$| t dt| td| ddd fWYd} ~ W|Sd} ~ wwW|n|wtd!d"dd#fS)$Nr=rErr)rr=rErzMissing required field: r r z SELECT diagnostics FROM diagnostics_by_doctor WHERE visits_id = %s AND doctor_id = %s AND patient_id = %s cSsh|]}t|dqS)r)int)r)rCr(r(r+ sz/update_diagnostics_by_doctor..z+All diagnostics already exist for visit_id z , doctor_id z , patient_id .z6DELETE FROM diagnostics_by_doctor WHERE visits_id = %sz6DELETE FROM medications_by_doctor WHERE visits_id = %sz,DELETE FROM medications WHERE visits_id = %sz/DELETE FROM diet_by_doctor WHERE visits_id = %sz%DELETE FROM diet WHERE visits_id = %sz0UPDATE visits SET diagnostics = %s WHERE id = %sz0UPDATE visits SET medications = %s WHERE id = %srz)UPDATE visits SET diet = %s WHERE id = %sz.UPDATE visits SET food_type = %s WHERE id = %svegz INSERT INTO diagnostics_by_doctor (visits_id, doctor_id, patient_id, diagnostics) VALUES (%s, %s, %s, %s) z'Error in update_diagnostics_by_doctor: zInternal server error r z+Diagnostics by doctor updated successfully.false)rget_jsonr/rrrr4r5r6setr2r8r3rollbackr<r9) rr=rErrrequired_fieldsfieldr?r4Zexisting_diagnosticsZnew_diagnosticsZ diagnostic_idr>r(r(r+update_diagnostics_by_doctors       '$ rVc Cs|stddigdddfSt}z]z't|jdd}d}|||f|}Wdn1s3wYWn/tyh}z#td t|tdd |igddd fWYd}~W| Sd}~wwW| n| w|stdd igd ddfSt|d ddfS)Nr zMissing visit_idr rrITrar SELECT d.id AS diagnostic_id, d.patient_id, d.visits_id, d.diagnostics AS diagnostic, d.created_at AS diagnostic_created_at, d.updated_at AS diagnostic_updated_at, d_d.id AS diagnostic_by_doctor_id, d_d.doctor_id, d_d.diagnostics AS doctor_diagnostic, d_d.created_at AS doctor_diagnostic_created_at, d_d.updated_at AS doctor_diagnostic_updated_at FROM diagnostics d INNER JOIN diagnostics_by_doctor d_d ON d.id = d_d.diagnostics AND d.visits_id = d_d.visits_id WHERE d.visits_id = %s z#Error in get_diagnostics_by_visit: r-r z No data foundrOrr) rrrr4r5r6r3r2r<r9)r=r?r4queryr@r>r(r(r+get_diagnostics_by_visits* * rXc CsFz{t}|}tj}|d}|d}|d}|d}|r%|r%|s/tddddfWSd }|||||f|}|rV|| | td ddd fWSd } || ||||f| |j } | | td | dddfWSt y} zt dt| tddt| ddfWYd} ~ Sd} ~ ww)Nr visits_idrrE4patient_id, visits_id, and diagnostics are required.r r rIz{ SELECT id FROM diagnostics WHERE patient_id = %s AND visits_id = %s AND diagnostics = %s zdiagnostics already exists.iz INSERT INTO diagnostics (patient_id, visits_id, diagnostics, doctor_id) VALUES (%s, %s, %s, %s) z!diagnostics created successfully.rO)r Zdiagnostics_idr rPzError in create_diagnostic: z*Error occurred while creating diagnostics.r r detailsr )rr4rr0r/rr5r7r6r9r8 lastrowidr3r2r<) r?r4rrrYrrE check_queryexisting_record insert_queryZnew_diagnostics_idr>r(r(r+create_diagnostics<     $rac Csz=t}|}d}|||f||jdkr,||tddddfWS||tdddd fWStyd}ztd t |td dt |d d fWYd}~Sd}~ww)Nz%DELETE FROM diagnostics WHERE id = %srzdiagnostics not found.r r rz!diagnostics deleted successfully.rOrzError in delete_diagnostic: z'Error occurred while diagnostics cause.r[r ) rr4r5r8rowcountr9rr3r2r<)rr?r4 delete_queryr>r(r(r+delete_diagnosticKs$ $rdc Cs zht}|}tj}|d}|d}|d}|d}|r%|r%|s/tddddfWSd }|||||||f||jd krW| | td ddd fWS| | td dddfWSt y} zt dt | tddt | ddfWYd} ~ Sd} ~ ww)NrrYrrErZr r rIz UPDATE diagnostics SET patient_id = %s, visits_id = %s, diagnostics = %s, doctor_id = %s WHERE id = %s rz)Diagnostics not found or no changes made.rz!Diagnostics updated successfully.rOrzError in update_diagnostic: z*Error occurred while updating Diagnostics.r[r ) rr4rr0r/rr5r8rbr9r3r2r<) rr?r4rrrYrrE update_queryr>r(r(r+update_diagnosticbs2      $rf)flaskrrconfrServices.promptsrServices.ai_workerr contextlibrr0r:r;rVrXrardrfr(r(r(r+s    &@-.