<?php

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

/**
 * Add Statistics Refinement Step
 * 
 * Adds relevant statistics, data points, and metrics.
 */
class AIGH_Step_Refine_Add_Statistics extends AIGH_Step_Refinement_Base {
    
    protected $step_type = 'refine_add_statistics';
    protected $step_name = 'Add Statistics';
    protected $description = 'Add relevant statistics, data points, and metrics to support claims';
    
    /**
     * Get the refinement instruction
     *
     * @return string
     */
    protected function get_refinement_instruction() {
        return "Add relevant statistics, data points, and metrics throughout the article. Make claims more authoritative with data.";
    }
    
    /**
     * 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 .= " USE EVERY RELEVANT STATISTIC from the research findings. Add at least 5-10 specific data points.";
        }
        
        $prompt .= "\n\n";
        $prompt .= $this->get_standard_requirements(!empty($research_context));
        
        return $prompt;
    }
}