chatbot.php
2 years ago
discussions.php
2 years ago
files.php
2 years ago
security.php
2 years ago
tasks.php
2 years ago
utilities.php
2 years ago
wand.php
2 years ago
wand.php
269 lines
| 1 | <?php |
| 2 | |
| 3 | class Meow_MWAI_Modules_Wand |
| 4 | { |
| 5 | private $core; |
| 6 | |
| 7 | public static $features = [ |
| 8 | 'correctText' => [ |
| 9 | 'label' => 'Correct Text', |
| 10 | 'sublabel' => 'Grammar & Spelling', |
| 11 | 'arguments' => ['postId', 'text'], |
| 12 | 'where' => 'blockContext', |
| 13 | 'group' => 'first' |
| 14 | ], |
| 15 | 'enhanceText' => [ |
| 16 | 'label' => 'Enhance Text', |
| 17 | 'sublabel' => 'Readibility & Quality', |
| 18 | 'arguments' => ['postId', 'text'], |
| 19 | 'where' => 'blockContext', |
| 20 | 'group' => 'first' |
| 21 | ], |
| 22 | 'longerText' => [ |
| 23 | 'label' => 'Longer Text', |
| 24 | 'sublabel' => 'Readibility', |
| 25 | 'arguments' => ['postId', 'text'], |
| 26 | 'where' => 'blockContext', |
| 27 | 'group' => 'first' |
| 28 | ], |
| 29 | 'shorterText' => [ |
| 30 | 'label' => 'Shorter Text', |
| 31 | 'sublabel' => 'Readibility', |
| 32 | 'arguments' => ['postId', 'text'], |
| 33 | 'where' => 'blockContext', |
| 34 | 'group' => 'first' |
| 35 | ], |
| 36 | 'translateText' => [ |
| 37 | 'label' => 'Translate Text', |
| 38 | 'sublabel' => 'To Post Language', |
| 39 | 'arguments' => ['postId', 'text', 'language'], |
| 40 | 'where' => 'blockContext', |
| 41 | 'group' => 'first' |
| 42 | ], |
| 43 | 'suggestSynonyms' => [ |
| 44 | 'label' => 'Suggest Synonyms', |
| 45 | 'sublabel' => 'For Selected Words', |
| 46 | 'arguments' => ['postId', 'text', 'selectedText'], |
| 47 | 'where' => 'blockContext', |
| 48 | 'group' => 'second' |
| 49 | ], |
| 50 | 'generateImage' => [ |
| 51 | 'label' => 'Generate Image', |
| 52 | 'sublabel' => 'For This Text', |
| 53 | 'arguments' => ['postId', 'text'], |
| 54 | 'where' => 'blockContext', |
| 55 | 'group' => 'third' |
| 56 | ], |
| 57 | 'suggestExcerpts' => [ |
| 58 | 'label' => 'Suggest Excerpts', |
| 59 | 'sublabel' => 'Generate SEO-Optimized Excerpts', |
| 60 | 'arguments' => ['postId'], |
| 61 | 'where' => 'postActions' |
| 62 | ], |
| 63 | 'suggestTitles' => [ |
| 64 | 'label' => 'Suggest Titles', |
| 65 | 'sublabel' => 'Generate SEO-Optimized Titles', |
| 66 | 'arguments' => ['postId'], |
| 67 | 'where' => 'postActions' |
| 68 | ] |
| 69 | ]; |
| 70 | |
| 71 | public function __construct( $core ) |
| 72 | { |
| 73 | $this->core = $core; |
| 74 | $this->register_filters(); |
| 75 | } |
| 76 | |
| 77 | private function register_filters() |
| 78 | { |
| 79 | foreach ( self::$features as $action => $feature ) { |
| 80 | add_filter( 'mwai_magic_wand_' . $action, [ $this, 'action_' . $action ], 10, 2 ); |
| 81 | } |
| 82 | } |
| 83 | |
| 84 | /** |
| 85 | * Common method to process text actions (e.g., correct, enhance, lengthen, shorten text). |
| 86 | * |
| 87 | * @param array $arguments The arguments provided for the action. |
| 88 | * @param string $messagePrefix The prefix for the message to be set in the query. |
| 89 | * @return array The result of the text processing. |
| 90 | */ |
| 91 | private function processTextAction( $arguments, $messagePrefix ) |
| 92 | { |
| 93 | $postId = $arguments['postId']; |
| 94 | $text = $arguments['text']; |
| 95 | $query = new Meow_MWAI_Query_Text( "", 1024 ); |
| 96 | $query->set_scope( 'admin-tools' ); |
| 97 | $language = $keepLanguage = ""; |
| 98 | if ( !empty( $postId ) ) { |
| 99 | $language = $this->core->get_post_language( $postId ); |
| 100 | $keepLanguage = " Ensure the reply is in the same language as the original text ({$language})."; |
| 101 | } |
| 102 | $query->set_message( $messagePrefix . $keepLanguage . "\n\n" . $text ); |
| 103 | $reply = $this->core->run_query( $query ); |
| 104 | |
| 105 | return [ |
| 106 | 'mode' => 'replace', |
| 107 | 'type' => $reply->type, |
| 108 | 'result' => $reply->result, |
| 109 | 'results' => $reply->results |
| 110 | ]; |
| 111 | } |
| 112 | |
| 113 | /** |
| 114 | * Handles the correction of text by checking and correcting grammatical errors. |
| 115 | */ |
| 116 | public function action_correctText( $value, $arguments ) |
| 117 | { |
| 118 | $prompt = apply_filters( 'mwai_prompt_correctText', "Correct the typos and grammar mistakes in this text without altering its content. Ensure the reply is in the same language as the original text.\n\n", $arguments ); |
| 119 | return $this->processTextAction( $arguments, $prompt ); |
| 120 | } |
| 121 | |
| 122 | /** |
| 123 | * Enhances the text's readability and quality. |
| 124 | */ |
| 125 | public function action_enhanceText( $value, $arguments ) |
| 126 | { |
| 127 | $prompt = apply_filters( 'mwai_prompt_enhanceText', "Enhance this text by improving readability and quality, using a more suitable vocabulary, and refining its structure.\n\n", $arguments ); |
| 128 | return $this->processTextAction( $arguments, $prompt ); |
| 129 | } |
| 130 | |
| 131 | /** |
| 132 | * Lengthens the text to improve readability. |
| 133 | */ |
| 134 | public function action_longerText( $value, $arguments ) { |
| 135 | $prompt = apply_filters( 'mwai_prompt_longerText', "Expand the subsequent text to a minimum of three times its original length, integrating relevant and accurate information to enrich its content. If the text is a story, amplify its charm by elaborating on essential aspects, enhancing readability, and creating a sense of engagement for the reader. Maintain consistency in tone and vocabulary throughout the expansion process.\n\n", $arguments ); |
| 136 | return $this->processTextAction( $arguments, $prompt ); |
| 137 | } |
| 138 | |
| 139 | /** |
| 140 | * Shortens the text to improve readability. |
| 141 | */ |
| 142 | public function action_shorterText( $value, $arguments ) { |
| 143 | $prompt = apply_filters( 'mwai_prompt_shorterText', "Condense the following text by reducing its length to half, while retaining the core elements of the original narrative. Focus on maintaining the essence of the story and its key details.\n\n", $arguments ); |
| 144 | return $this->processTextAction( $arguments, $prompt ); |
| 145 | } |
| 146 | |
| 147 | /** |
| 148 | * Suggests synonyms for selected words in the text. |
| 149 | */ |
| 150 | public function action_suggestSynonyms( $value, $arguments ) { |
| 151 | $postId = $arguments['postId']; |
| 152 | $selectedText = $arguments['selectedText']; |
| 153 | $query = new Meow_MWAI_Query_Text( "", 1024 ); |
| 154 | $query->set_scope( 'admin-tools' ); |
| 155 | $language = $keepLanguage = ""; |
| 156 | if ( !empty( $postId ) ) { |
| 157 | $language = $this->core->get_post_language( $postId ); |
| 158 | $keepLanguage = " Ensure the reply is in the same language as the original text ({$language})."; |
| 159 | } |
| 160 | $prompt = apply_filters( 'mwai_prompt_suggestSynonyms', "Provide 5 synonyms or 5 ways of rephrasing the given word or sentence while retaining the original meaning and preserving the initial and final punctuation and spacing if any. Offer only the resulting word or expression, without additional context. If a suitable synonym or alternative cannot be identified, ensure that a creative response is still provided. Separate every suggestion with a new line, and that's it." . $keepLanguage . "\n\n", $arguments ); |
| 161 | $query->set_message( $prompt . $selectedText ); |
| 162 | $query->set_temperature( 1 ); |
| 163 | $reply = $this->core->run_query( $query ); |
| 164 | $lines = explode( "\n", $reply->result ); |
| 165 | $results = []; |
| 166 | foreach ( $lines as $line ) { |
| 167 | if ( !empty( $line ) ) { |
| 168 | $results[] = $line; |
| 169 | } |
| 170 | } |
| 171 | return [ |
| 172 | 'mode' => 'suggest', |
| 173 | 'type' => $reply->type, |
| 174 | 'result' => $results[0] ?? '', |
| 175 | 'results' => $results |
| 176 | ]; |
| 177 | } |
| 178 | |
| 179 | /** |
| 180 | * Generates an image relevant to the text. |
| 181 | */ |
| 182 | public function action_generateImage( $value, $arguments ) { |
| 183 | $postId = $arguments['postId']; |
| 184 | $text = $arguments['text']; |
| 185 | $prompt = apply_filters( 'mwai_prompt_generateImage', "Generate an image that is relevant to the following text:\n\n", $arguments ); |
| 186 | $query = new Meow_MWAI_Query_Image( $prompt . $text ); |
| 187 | $query->set_local_download( null ); |
| 188 | $reply = $this->core->run_query( $query ); |
| 189 | preg_match( '/\!\[Image\]\((.*?)\)/', $reply->result, $matches ); |
| 190 | $url = $matches[1] ?? $reply->result; |
| 191 | $attachmentId = $this->core->add_image_from_url( $url, null, null, null, null, null, $postId ); |
| 192 | if ( empty( $attachmentId ) ) { |
| 193 | throw new Exception( 'Could not add the image to the Media Library.' ); |
| 194 | } |
| 195 | $media = [ |
| 196 | 'id' => $attachmentId, |
| 197 | 'url' => wp_get_attachment_url( $attachmentId ), |
| 198 | 'title' => get_the_title( $attachmentId ), |
| 199 | 'caption' => wp_get_attachment_caption( $attachmentId ), |
| 200 | 'alt' => get_post_meta( $attachmentId, '_wp_attachment_image_alt', true ) |
| 201 | ]; |
| 202 | return [ |
| 203 | 'mode' => 'insertMedia', |
| 204 | 'type' => 'image', |
| 205 | 'media' => $media |
| 206 | ]; |
| 207 | } |
| 208 | |
| 209 | /** |
| 210 | * Translates the text to the specified language. |
| 211 | */ |
| 212 | public function action_translateText( $value, $arguments ) { |
| 213 | $postId = $arguments['postId']; |
| 214 | $text = $arguments['text']; |
| 215 | $language = $this->core->get_post_language( $postId ); |
| 216 | $query = new Meow_MWAI_Query_Text( "", 1024 ); |
| 217 | $query->set_scope( 'admin-tools' ); |
| 218 | $prompt = apply_filters( 'mwai_prompt_translateText', "Translate the text into {$language}, preserving the tone, mood, and nuance, while staying as true as possible to the original meaning. Provide only the translated text, without any additional content.\n\n", $arguments ); |
| 219 | $query->set_message( $prompt . $text ); |
| 220 | $reply = $this->core->run_query( $query ); |
| 221 | return [ |
| 222 | 'mode' => 'replace', |
| 223 | 'type' => $reply->type, |
| 224 | 'result' => $reply->result, |
| 225 | 'results' => $reply->results |
| 226 | ]; |
| 227 | } |
| 228 | |
| 229 | /** |
| 230 | * Suggests SEO-optimized excerpts for the text. |
| 231 | */ |
| 232 | public function action_suggestExcerpts( $value, $arguments ) { |
| 233 | $postId = $arguments['postId']; |
| 234 | $text = $this->core->get_post_content( $postId ); |
| 235 | $query = new Meow_MWAI_Query_Text( "", 1024 ); |
| 236 | $query->set_scope( 'admin-tools' ); |
| 237 | $prompt = apply_filters( 'mwai_prompt_suggestExcerpts', "Craft a clear, SEO-optimized introduction for the following text, using 120 to 170 characters. Ensure the introduction is concise and relevant, without including any URLs.\n\n", $arguments ); |
| 238 | $query->set_message( $prompt . $text ); |
| 239 | $query->set_max_results( 5 ); |
| 240 | $reply = $this->core->run_query( $query ); |
| 241 | return [ |
| 242 | 'mode' => 'suggest', |
| 243 | 'type' => $reply->type, |
| 244 | 'result' => $reply->result, |
| 245 | 'results' => $reply->results |
| 246 | ]; |
| 247 | } |
| 248 | |
| 249 | /** |
| 250 | * Suggests SEO-optimized titles for the text. |
| 251 | */ |
| 252 | public function action_suggestTitles( $value, $arguments ) { |
| 253 | $postId = $arguments['postId']; |
| 254 | $text = $this->core->get_post_content( $postId ); |
| 255 | $query = new Meow_MWAI_Query_Text( "", 1024 ); |
| 256 | $query->set_scope( 'admin-tools' ); |
| 257 | $prompt = apply_filters( 'mwai_prompt_suggestTitles', "Generate a concise, SEO-optimized title for the following text, without using quotes or any other formatting. Focus on clarity and relevance to the content.\n\n", $arguments ); |
| 258 | $query->set_message( $prompt . $text ); |
| 259 | $query->set_max_results( 5 ); |
| 260 | $reply = $this->core->run_query( $query ); |
| 261 | return [ |
| 262 | 'mode' => 'suggest', |
| 263 | 'type' => $reply->type, |
| 264 | 'result' => $reply->result, |
| 265 | 'results' => $reply->results |
| 266 | ]; |
| 267 | } |
| 268 | } |
| 269 |