// AJAX 处理函数
function query_coursehero_ajax_handler() {
    if (isset($_POST['url'])) {
        $url = sanitize_text_field($_POST['url']);

        // 构建 Python 脚本路径
        $python_script_path = get_template_directory() . '/scripts/coursehero_query.py';

        // 使用 shell_exec 调用 Python 脚本并传递 URL 参数
        $output = shell_exec("python3 " . escapeshellarg($python_script_path) . " " . escapeshellarg($url));

        if ($output) {
            // 清理返回路径，确保路径指向 PDF 文件
            $pdf_path = trim($output);
            $pdf_url = content_url('themes/url-input-theme/scripts/query_result.pdf');

            // 返回 PDF 文件的 URL
            wp_send_json_success(['pdf_url' => $pdf_url]);
        } else {
            wp_send_json_error(['message' => '无法从 Course Hero 获取数据']);
        }
    } else {
        wp_send_json_error(['message' => '未提供网址']);
    }
}
add_action('wp_ajax_query_coursehero', 'query_coursehero_ajax_handler');
add_action('wp_ajax_nopriv_query_coursehero', 'query_coursehero_ajax_handler');


<form id="urlForm" method="post" action="">
    <label for="urlInput">网址：</label>
    <input type="text" id="urlInput" name="url" placeholder="https://example.com" required>
    <button type="submit">查询</button>
</form>
<div id="result" style="margin-top: 20px;"></div>

<script>
document.getElementById('urlForm').addEventListener('submit', function(e) {
    e.preventDefault();
    const url = document.getElementById('urlInput').value;

    fetch(ajaxurl, {
        method: 'POST',
        headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
        body: new URLSearchParams({
            action: 'query_coursehero',
            url: url
        })
    })
    .then(response => response.json())
    .then(data => {
        if (data.success) {
            document.getElementById('result').innerHTML = `
                <p>查询完成。<a href="${data.pdf_url}" download>点击下载 PDF</a></p>`;
        } else {
            document.getElementById('result').innerHTML = `<p>错误：${data.message}</p>`;
        }
    })
    .catch(error => console.error('Error:', error));
});
</script>