tĀiY# dZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZej,eZgd Zd Zd ZGddee ZeGddZdej>fdZ dedefdZ! d dedede"de#e$ee%ffdZ&dedefdZ'dedede(e%defdZ)dedefdZ*y)!zj Fact-checking module using Gemini Pro with strict verification. Maximum 2 retry attempts before failing. N) dataclassfield)Enum)Path)Optional)genai)types) get_settings)Topic)zgemini-3-pro-previewzgemini-3-flash-previewzgemini-2.5-proc eZdZdZdZdZdZdZy) FactVerdictverifiedminor_variation incorrect unverifiableoutdatedN)__name__ __module__ __qualname__VERIFIEDMINOR_VARIATION INCORRECT UNVERIFIABLEOUTDATED6/var/www/html/content-pipeline/modules/fact_checker.pyrr"sH'OI!LHrrc`eZdZUeed<eed<eeZee ed<dZ eed<dZ e ed<y ) FactCheckResultpassedverdict)default_factoryissuessummary confidenceN) rrrbool__annotations__strrlistr%dictr'r)floatrrrr!r!*s3 L Lt4FDJ4GSJrr!returncVt}tj|jS)z+Initialize Gemini client for fact-checking.)api_key)r rClientgemini_api_key)settingss rget_gemini_clientr63s~H << 7 7 88rarticle_contentcftdj|dd} t}tjtj }d}d}t D]v}ttD]]} tjd|d|dzd td |jj||tj|g  }ns|vn|td|t)|j*} tj-d| j.dt1| j2d| j4dd | S#t$r}|}d t|vs dt|vrX|tdz krLtj!d|dt"d|dzd td t%j&t"Yd}~Rtj!d|d|Yd}~d}~wwxYw#t$r=}tj7d|t9dddt|dcYd}~Sd}~wwxYw)z Perform strict fact-check on article using Gemini Pro with web search. Identifies factual claims and verifies each one. zfact_check.txtNi:)r7) google_searchz Trying model z (attempt /))tools)modelcontentsconfig429RESOURCE_EXHAUSTEDzRate limited on z , waiting z s (attempt zModel z failed: zAll models failed. Last error: zFact-check completed: z (z issues, confidence: z.0%zFact-check failed: FFAILzFact-check error: r(r"r#r'r)) _load_promptformatr6r Tool GoogleSearch TEXT_MODELSrange MAX_RETRIESloggerdebugmodelsgenerate_contentGenerateContentConfig Exceptionr,warning RETRY_DELAYtimesleep_parse_fact_check_responsetextinfor#lenr%r)errorr!) r7promptclientgoogle_search_toolresponse last_errorr>attempteresults r fact_checkrc9sX * + 2 2'/3F6 "$"ZZe6H6H6JK   E - LL=z'A+aP[}\]!^_%}}==#!'$::#5"6  > H .#5 8  =j\JK K+HMM: $V^^$45FMM"##89J9J38Oq R  3! !"JA*>#a&*H"[1_4"NN-=eWJ{m[fgnqrgrfsstvAuBBC,DE JJ{3$NNVE7)A3#?@ 6   *1#./(Q1    s]AG*2AE G*A0G* G' A.G":G*G"G*"G''G** H032H+%H0+H0topic max_attemptsc"|}gdd}t|D]}tjd|dzd|t|}|dj |dz|j |j |jd|j dk(r d|d <d ||fcS|j d k(r|jd k\r|j Dcgc]&}|jd djdvr|(} }t| dkr8d|d <tjdt|j dd ||fcS|j dk(r.d|d <tjd|jd||fcS|j d k(so|ss||dz kr8tjdt|||j } ||| }d |d <d||fcSd|d <d||fScc}w#t$r#} tjd| Yd} ~ d} ~ wwxYw)aX Fact-check with retry logic. Args: article_content: The article markdown to check topic: The topic for regeneration context max_attempts: Maximum fact-check attempts (default 2) regenerate_func: Function to regenerate article if needed Returns: (passed, final_content, fact_check_log) N)attempts final_verdictzFact-check attempt r:r;rg)r`r#r%r)PASSrhTNEEDS_REVISIONg?categoryr&)rrPASS_WITH_WARNINGSz!Fact-check passed with warnings: z minor issuesrCzFact-check FAIL: Fz8Attempting article revision based on fact-check feedbackzRevision failed: MAX_ATTEMPTS_REACHED)rJrLrXrcappendr#r%r)getupperrYrZr'_build_revision_promptrQ) r7rdreregenerate_funccurrent_contentlogr`rbicritical_issuesrevision_promptras rfact_check_with_retryry|s:$&OD 1C&23 )'A+a ~FGO, J"Q;!>> --$//    >>V ##)C #- - >>- -&2C2Ct2K"==55R(..04MMO?#q(';O$ ?FMM@R?SS`ab_c11 >>V ##)C LL,V^^,<= >/3. . >>- -/)) VW"8?FMM#&5e_&MO (8O$os22e23h2C /3 &&C0!LL#4QC!89s+G; G"" H+H  Hfilenamecttjjdz |z }|jst d||j S)z,Load prompt template from prompts directory.promptszPrompt file not found: )r__file__parentexistsFileNotFoundError read_text)rz prompt_paths rrErEsOx.''..:XEK    "9+ GHH  ""rcontentr%c dj|Dcgc]*}d|jddd|jdd,c}}d|d |d Scc}w) z3Build a revision prompt based on fact-check issues. z- claimz Unknown claimz: issuezNeeds verificationugÜberarbeite den folgenden Artikel und korrigiere die identifizierten Probleme. # PROBLEME ZU BEHEBEN z # ORIGINAL-ARTIKEL u # ANWEISUNGEN 1. Korrigiere ALLE genannten Probleme mit aktuellen, verifizierbaren Fakten 2. Behalte die Struktur und den Stil bei 3. Stelle sicher, dass alle Zahlen und Behauptungen korrekt sind 4. Füge Quellenangaben hinzu wo möglich Liefere den korrigierten Artikel: )joinrp)rdrr%r issues_texts rrrrrsz))  7O45R 'K_8`7a b K      s/A response_textc  |j}d|vr'|jddjdd}n-d|vr'|jddjdd}n|}tj|}|j ddj }|dvrd}|j d g}t |j d d }|d k(}t||||j d d|S#tjttf$r=}tjd|tdddt|dcYd}~Sd}~wwxYw)z6Parse Gemini fact-check response into FactCheckResult.z```jsonr:z```rr#rC)rirjrCrjr%r)g?rir'r&)r"r#r%r'r)z%Failed to parse fact-check response: Fz Parse error: r(rDN)stripsplitjsonloadsrprqr/r!JSONDecodeErrorKeyError ValueErrorrLrRr,) rrW json_textdatar#r%r)r"ras rrVrVsU' ""$   9-a066u=a@I d] 5)!,2259!qcBC$#CF8,    sC4C77E2E E E)rlN)+__doc__rloggingrerT dataclassesrrenumrpathlibrtypingrgoogler google.genair config.settingsr database.modelsr getLoggerrrLrIrKrSr,rr!r3r6rcinttupler*r.ryrEr-rrrVrrrrs0   ((!   8 $   #t  95<<9 @ @ @ L J'J' J'J'  4d? J'Z#3#3#%#tDzc6) c) o) r