<?php

if (!defined('ABSPATH')) {
    exit;
}

/**
 * Improve Conclusion Refinement Step
 * 
 * Rewrites and improves the article conclusion.
 */
class AIGH_Step_Refine_Improve_Conclusion extends AIGH_Step_Refinement_Base {
    
    protected $step_type = 'refine_improve_conclusion';
    protected $step_name = 'Improve Conclusion';
    protected $description = 'Rewrite and improve the conclusion to be more impactful with a strong call-to-action';
    protected $estimated_duration = 10;
    
    /**
     * Get the refinement instruction
     *
     * @return string
     */
    protected function get_refinement_instruction() {
        return "Rewrite and improve the conclusion to be more impactful. Include a strong call-to-action.";
    }
    
    /**
     * Build the user prompt
     *
     * @param string $original_content
     * @param string $research_context
     * @param AIGH_Pipeline_Context $context
     * @return string
     */
    protected function build_user_prompt($original_content, $research_context, $context) {
        $prompt = "Here is the original article:\n\n{$original_content}\n\n";
        
        if (!empty($research_context)) {
            $prompt .= "\n{$research_context}\n\n";
            $prompt .= "IMPORTANT: Use the specific data, statistics, and examples from the research above to improve the article.\n\n";
        }
        
        $prompt .= "Refinement Task: " . $this->get_refinement_instruction();
        
        if (!empty($research_context)) {
            $prompt .= " Reference key statistics from the research.";
        }
        
        $prompt .= "\n\n";
        $prompt .= $this->get_standard_requirements(!empty($research_context));
        
        return $prompt;
    }
}