hnidZddlZddlZddlZddlZddlZddlZddlmZejjddGddZ dZ e dk(re yy) a% Test Article Generator for CleanKitchens - Weaviate Version Processes 2-3 inspection records through complete pipeline: 1. Get inspection data from Weaviate 2. Send to Claude for article generation 3. Select appropriate image 4. Save complete article back to Weaviate 5. Display on live site N)datetimezO/var/www/twin-digital-media/public_html/_sites/cleankitchens/production/scriptsc\eZdZdZddZdZdZdZdZdZ dZ d Z d Z d Z d Zd Zy)TestArticleGeneratorcd|_d|_d|_tjdd|_ddd d d d d dddd|_dddgddddgddddgdddi|_y)Nzl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%https://api.anthropic.com/v1/messages localhosti)hostport)zUSDA Danger Zonezphttps://www.fsis.usda.gov/food-safety/safe-food-handling-and-preparation/food-safety-basics/danger-zone-40f-140f)zFDA Food Code 3-501.16z5https://www.fda.gov/food/fda-food-code/food-code-2022)zFDA Cold Storage RequirementszDhttps://www.fda.gov/food/buy-store-serve-safe-food/safe-food-storage) danger_zone hot_holding cold_holding)zCDC Handwashing Guidelinesz9https://www.cdc.gov/handwashing/when-how-handwashing.html)z"FDA Preventing Cross-ContaminationzQhttps://www.fda.gov/food/buy-store-serve-safe-food/preventing-cross-contamination) handwashing contamination)zCDC Rodent Disease Informationz/https://www.cdc.gov/rodents/diseases/index.html)zCDC Disease Transmissionz0https://www.cdc.gov/rodents/diseases/direct.html)rodentsdiseases) temperaturehygienepests neighborhoodszfar South Side neighborhoodzOCTA Red Line at 95th Street and Electric District line at 103rd Street/Roseland)z Fernwood ParkzRoseland Community HospitalzMichigan Avenue corridor descriptiontransit landmarkszLoop business districtz'multiple CTA lines converge in the Loop)zMillennium Parkz City Hallz Chicago Riverz!North Side lakefront neighborhoodzBrown and Red Lines)zLincoln Park ZoozDePaul UniversityzNorth Avenue Beach)ROSELANDDOWNTOWNz LINCOLN PARK)claude_api_key claude_modelclaude_api_urlweaviateconnect_to_localweaviate_clienteducational_referenceschicago_context)selfs i/var/www/twin-digital-media/public_html/_sites/cleankitchens/production/scripts/test_article_generator.py__init__zTestArticleGenerator.__init__sM8E (88kPTU hr!J {"] ql' #$ #@p!m $<H!R $G4!`! c* |jjjd}|jj |gd}g}|j D]}|j jdd}|j|d}|j|}|j|d} |j|d} |j jd |j jd |j jd |j jd |j jd ||| | |t|jd } |j| |S#t$r} td| gcYd} ~ Sd} ~ wwxYw)z$Get inspection records from Weaviate RawInspection) inspection_iddba_namecityresultsraw_datainspection_date source_apistatus)limitreturn_propertiesr-Addressz Facility TypeZipr)r*r+r,r.) r) facility_namer+r,r.address violations facility_typezip_coder-uuidzError getting inspection data: N)r collectionsgetquery fetch_objectsobjects propertiesextract_field_from_rawextract_violations_from_rawstrr;append Exceptionprint) r#r1 collectionresponse inspectionsobjr-r7r8r9r: inspectiones r$get_inspection_dataz(TestArticleGenerator.get_inspection_dataGs}* --99==oNJ"''55#6HK'' />>--j"=55h J!==hG $ ; ;Ho V 66xG&)^^%7%7%H%(^^%7%7 %CNN..v6"~~11)<'*~~'9'9:K'L&",%2 ( (M  "":./ /2   3A37 8I sE,E// F8F F Fc ddl}|d}|j||}|r|jdjS y#YyxYw)z+Extract specific field from raw_data stringrNz:([^,]+)r3)researchgroupstrip)r#r- field_namerQpatternmatchs r$rBz+TestArticleGenerator.extract_field_from_rawusW  # H-GIIgx0E{{1~++--  s ;AAch d|vr%|jddd}|jS y#YyxYw)z Extract violations from raw_dataz Violations:rPr3)splitrT)r#r-violations_parts r$rCz0TestArticleGenerator.extract_violations_from_rawsJ ("*.."B1"E&,,..)   s(-1c|j|d}dddddddd d ddd d d dd}|j||d}||d|dd|jvr|ddS|ddS)z9Parse inspection result and get city-specific explanationResultsz Pass/FailzkChicago uses a Pass/Fail system where restaurants either pass inspection or fail due to critical violationszJindicates critical food safety violations that pose immediate health riskszJindicates the restaurant met basic food safety standards during inspection)typer fail_meaning pass_meaningz Letter Gradeztypically corresponds to a C grade with significant violationszz?TestArticleGenerator.get_relevant_references..s^Dt''^)rcoldhotholdingrr c3&K|]}|v ywrvrwrxs r$r|z?TestArticleGenerator.get_relevant_references..sYDt''Yr})handwashgloverrr c3&K|]}|v ywrvrwrxs r$r|z?TestArticleGenerator.get_relevant_references..s`Dt''`r})rodentmouseratroachpestrrc3&K|]}|v ywrvrwrxs r$r|z?TestArticleGenerator.get_relevant_references..sODt''Or})rcrossrN)rhanyrEr!)r#violations_text relevant_refsr{s @r$get_relevant_referencesz,TestArticleGenerator.get_relevant_referencess*002  ^4]^ ^  !z }|d?|d@d3z }|dA|ddBz }|dC|d3z }|dD|dEd3z }|jdFr |dG|dFd3z }|dHz }|dI|dJd3z }|dK|dLd3z }|dM|dNd3z }|dO|dd=z }|dPz }|dQz }|dRz }|dSz }|dTz }|dUz }|dVz }|dWz }|dUz }|S#|d}Y xYw)Xz0Build Claude prompt with your educational formatr7r:r.z%m/%d/%Yz %B %d, %Yr8r3r-r+zpYou are a food safety educator and journalist creating educational content about restaurant health violations. z_Create an article that TEACHES while it INFORMS about food safety using real inspection data. zvCRITICAL: You must follow the EXACT format specified below. Each section must be wrapped with the specified markers. zYour article should: zJ1. Have a compelling, specific headline that includes the restaurant name zJ2. Report the violations factually and objectively with neutral news tone z93. Focus on what readers can learn from these violations z"4. Be between 600-900 words total z<5. Include local context (neighborhood, transit, landmarks) zJ6. Use neutral news reporter tone - NO hyperbole, NO fictional characters z47. Include specific details about violations found zIMPORTANT FORMATTING: zB- Start content with

(NOT

) as this goes into a template zM- Use proper heading hierarchy:

for main sections,

for subsections z- Use

tags for paragraphs z8- This is article CONTENT ONLY, not a complete web page zD- Do NOT include , , or any page structure tags z5REQUIRED: Add 'The Food Safety Lesson' section with: z?- 2-3 sentences explaining the food safety principles involved zZ- MUST include inline hyperlinks to government sources using HTML: text zI- Every educational statement should link to its source (FDA, CDC, USDA) z)- One practical tip for home food safety z7- Format section with

The Food Safety Lesson

zgREQUIRED: Add 'Frequently Asked Questions' section with EXACTLY these 4 questions in this exact order: z7QUESTION 1: 'When was this restaurant last inspected?' z.QUESTION 2: 'What was the inspection result?' z*QUESTION 3: 'What does this rating mean?' z*QUESTION 4: 'What violations were found?' zE- Use the inspection data provided to answer each question factually z=- For Question 3, use the rating system information provided z6- Include hyperlinks to government sources in answers z1- Format as:

Frequently Asked Questions

zS

Q: When was this restaurant last inspected?
A: [Answer]

zJ

Q: What was the inspection result?
A: [Answer]

zF

Q: What does this rating mean?
A: [Answer]

zG

Q: What violations were found?
A: [Answer]

z0Use these government references where relevant: z- rz: rP zLOCAL CONTEXT for z: z- Neighborhood: rz - Transit: rz - Landmarks: z, rz zRESTAURANT DETAILS: zRestaurant Name: r6 Address: z, Chicago, IL zInspection Date: zInspection Result: r,r9zFacility Type: z RATING SYSTEM INFORMATION: z System Type: rezSystem Description: rfzResult Meaning: rgz VIOLATIONS FOUND: zFOUTPUT FORMAT - You must structure your response EXACTLY as follows: z===META_DATA=== z=title: [55-60 char title - restaurant name + violation type] zJdescription: [150-160 char description mentioning location and violation] z===META_DATA=== z===CONTENT=== zF[Your complete article HTML with Food Safety Lesson and FAQ sections] zC[Use HTML tags:

,

,

, , text] ) rs isinstancerDrstrptimestrftimerr=rkjoin) r#rL neighborhooddate_objr.r rating_infopromptrefs r$build_claude_promptz(TestArticleGenerator.build_claude_promptsv44Z 5JJWaLbc  <*%67=#,,Z8I-JJW%&78&// >MM||LLCC????ZZRRKKFFhh__[[]]  I IF$ 4Bs1vhbQ33 4 dNF &z&'9&:#>>$\-%@$ADDK Y 78;;M$))L,E"F!GtLL ))%j&A%B"EEIj34ODD%o%6b99' 9(='>bAA >>/ *  ?(C'DBG GF 23M+m"]+> z...rILr6)ChicagozHealth Inspectionz Food Safetyr published)rrrrrr+rrrrrrr)rr0u✓ Saved article to Weaviate: zError saving to Weaviate: )!r r<r=r>r?r@rGrAweaviate.classesclassescreateconfig Configure Vectorizertext2vec_transformersPropertyDataTypeTEXT TEXT_ARRAYrQsubrhrTlenrsplitrrYrnow isoformatrinsertrF)r#rLrrrrHexistingwvcrQr text_onlyr article_datardrMs r$save_article_to_weaviatez-TestArticleGenerator.save_article_to_weaviatesQ ' !11==AA*M &++99!0 1$+%/%@  :##I*UdJeIfgh#++A.999$: 66-immGR.H.N.N.PQWWX[\D4y3DSz((a03 z2w7Ihhy0"56>G#w3""$-MM-$D"6*&0&A"*,,.":":"<' 4' 4G!+O!< +%L& __++L9F 3F8< =Mw .!11==DD&)jj&:&:&E&E&[&[&] ++CJJDWDWD\D\+] ++3::CVCVC[C[+\ ++cjjFYFYF^F^+_ ++cjjFYFYF^F^+_ ++1CszzObObOgOg+h ++3::CVCVC[C[+\ ++CJJDWDWD\D\+] ++1EQTQ[Q[QdQdQiQi+j ++1ASZZM`M`MeMe+f ++ H[H[H`H`+a ++ H[H[H`H`+a ++3::CVCVCaCa+b ++CJJL_L_LdLd+e ++3::K^K^KcKc+d ++SZZEXEXE]E]+^ E t  .qc2 3 s,A?F8D4V8OU=:V V! VV!c tdtd|jd}|s tdytdt|dt|d D]\}}td |d t|d td |dtd|dtd|d td|j |}td|j |}|s tdtd|j |\}}|s tdtd|jddtdt|dtd|j|}td|dtd |j||||} | retd!td"|jdtd#|jdd$jjd%d&n td't|t|kr td(tjd)td+td,y#t$r} td*| Yd} ~ ,d} ~ wwxYw)-zMain test processingz=== Test Article Generator ===z ? HZ 234 5 Ij345 61 1211*=-."&"6"6v">&9:67%)%:%:?%K" 756/ gy0Q/RST*3w<. DE67!..z: ,Z -D,EFG-.66z9gWab;=IimmG&<%=>?,Y]]7B-G-M-M-O-W-WX[]`-a,bcd23s;''FGJJqMg7 r %& =>  21#67 s&1AI*7,I*$D,I** J 3JJ N)r)__name__ __module__ __qualname__r%rNrBrCrkrsrrrrrrrrwr&r$rrsL- ^,\  " H"!.kZ$L40SjH?r&rct} |j|jjy#|jjwxYwrv)rrr close) generators r$mainr >s@$&I*'')!!'') !!'')s 7A__main__) __doc__ossysrrrrrrrrr rrwr&r$rs\  dee?e?N* zFr&