&i'! 0dZddlmZddlmZmZmZddlmZddl m Z m Z m Z m Z ddlmZddlmZddlmZdd lmZdd lmZd e d ed fdZd e d edfdZd e d e fdZd e d e fdZddedzfdZ ddedededzdedzd e f dZdede ded e fdZ y)zg Main SEO Campaign Graph Orchestrates the full SEO campaign workflow with human-in-the-loop interrupts )Literal) StateGraphSTARTEND)BaseCheckpointSaver)SEOCampaignState CampaignPhasecreate_initial_statetransition_phase)research_phase) content_phase)serp_optimization_phase)link_discovery_phase)generate_report_phasestatereturnawait_approvalskip_to_reportcJ|jdr|jdryy)z Conditional edge: Check if human review is required Returns "await_approval" if there are pending approvals, otherwise "skip_to_report" to proceed directly to reporting. requires_human_reviewpending_approvalsrr)get)rs H/var/www/html/eventheodds/langgraph-service/graphs/seo_campaign_graph.pycheck_human_review_requiredrs# yy()eii8K.L execute still_pending all_rejectedc0|jdg}|jdg}t|Dcgc]}|jddk(s|c}}|dkDryt|Dcgc]}|jddk(s|c}}|dkDryy cc}wcc}w) z Conditional edge: Check approval status after human review Returns: - "execute" if any opportunities were approved - "still_pending" if still awaiting approvals - "all_rejected" if all were rejected rlink_opportunitiesstatuspendingrrapprovedrr )rlen)rr link_oppso pending_countapproved_counts rcheck_approval_statusr+#s "5r: .3I$5VqxI9UVWMqYP!%%/Z2O!PQN W QsBB"B<Bc Ktdt|jdgd|jdg}|Dcgc]}|jddk(s|}}|Dcgc]}|jddk(s|}}|Dcgc]}|jddk(s|}}t||d <||d<|sd |d <td t|d t|dt||Scc}wcc}wcc}ww)a Human approval interrupt node This node is configured as an interrupt point. When the graph reaches this node, execution pauses and waits for human input via the resume API. The human can: - Approve opportunities (set status to "approved", optionally edit pitch) - Reject opportunities (set status to "rejected", provide reason) After human input, the graph resumes and this node processes the updated state. z&[HumanApproval] Awaiting approval for rz opportunitiesr"r#r%rejectedr$links_approvedFrz[HumanApproval] Approved: z , Rejected: z, Still pending: )printr&r)rr'r(r%r-rs rawait_human_approvalr0<s 23uyyATVX7Y3Z2[[i jk .3I$Fah:(EFHF$Fah:(EFHF )J1QUU8_ -IQJMJ!(mE !.E  ).%& &s8}o\#h-Pabefsbtau vw LGFJs<?DC4C4D%C9?C9D C>#C>'ADc~Ktd|jdg}|Dcgc]}|jddk(s|}}|D];}d|d<tdjj j |d<=tdt |d t|tj}|Scc}ww) z Execute approved link outreach Only runs after human approval. Sends outreach emails for approved opportunities. z?[ExecuteOutreach] Executing outreach for approved opportunitiesr"r#r%executeddatetime executed_atzxecuteOutreach] Executed z outreach campaigns) r/r __import__r3utcnow isoformatr&r r REPORT)rr'r(r%opps rexecute_approved_outreachr:_s  KM .3I$Fah:(EFHFR"H ' 3<<CCEOOQMR 'H 6I JK UM$8$8 9E LGs"B=B8B8A;B=N checkpointerctt}|jdt|jdt|jdt |jdt |jdt|jdt|jdt|jtd|jdd|jdd|jdd|jdtddd|jdtdtdd |jdd|jdt|j!|dg }|S) af Create the main SEO campaign workflow graph Phases: 1. Research - Gather GSC data, competitors, SERP features 2. Content - Prioritize and queue content generation 3. SERP Features - Queue optimizations for snippets, PAA, AI Overviews 4. Link Discovery - Find white-hat link opportunities 5. Human Review (INTERRUPT) - Pause for human approval of link opportunities 6. Execute - Send approved outreach 7. Report - Generate final campaign report The graph uses interrupt_before on the human_review node, allowing campaigns to pause and resume after human approval. researchcontent serp_featureslink_discovery human_reviewexecute_outreachreportrr)r;interrupt_before)rradd_noder r rrr0r:radd_edgeradd_conditional_edgesrr+rcompile)r;graphcompileds rcreate_seo_campaign_graphrKxs@" ' (E NN:~. NN9m, NN?$;< NN#%9: NN>#78 NN%'@A NN823 NN5*% NN:y) NN9o. NN?$45 #,&  ) $  NN%x0 NN8S!}}!()H Orsite_id campaign_typeconfigcKddl}t|j}t||||xsi}t |}dd|ii}|j ||d{} | S7w)a Create and run a new SEO campaign Args: site_id: The site ID to run the campaign for campaign_type: Type of campaign (zero_click_authority, link_building, etc.) config: Optional campaign configuration checkpointer: Optional checkpointer for state persistence Returns: Final campaign state (or paused state if awaiting human review) rN)rL campaign_idrMrN configurable thread_id)uuidstruuid4r rKainvoke) rLrMrNr;rSrP initial_staterI thread_config final_states r run_campaignrZss$djjl#K(#| M &l 3E$k;%?@M m]CCK DsAA"A A"rP updated_statechKt|}dd|ii}|j||d{}|S7w)a6 Resume a paused campaign after human approval Args: campaign_id: The campaign ID to resume updated_state: State with updated approvals checkpointer: Checkpointer with persisted state Returns: Final campaign state (or paused state if still awaiting approvals) rQrRN)rKrV)rPr[r;rIrXrYs rresume_campaignr]sB &l 3E#k;%?@M m]CCK Ds '202)N)zero_click_authorityNN)!__doc__typingrlanggraph.graphrrrlanggraph.checkpoint.baserstate.campaign_staterr r r nodes.research_noder nodes.content_noder nodes.serp_features_nodernodes.link_discovery_nodernodes.performance_noderrr+r0r:rKrTdictrZr]rrrks 229 /,<:8 '7 GDfh6i2 &6 ;K F+;@P2C,?$,FCR0/3 % %% 4K%&, %  %P#& r