Image generation recovery (#36)

* removing the model selection bug

* adding the media manager class

* returning the image generation functionality

Author: tomepajk

src/MCP/Client.php

@@ -110,6 +110,7 @@ static function () {
 					file_put_contents( $filename, $image_blob->get_binary_data() );
 
 					$image_url = $filename;
+					$image_id = \WP_CLI\AiCommand\MediaManager::upload_to_media_library($image_url);
 				}
 
 				break;
@@ -120,6 +121,8 @@ static function () {
 				// TODO: Save as file or so.
 				break;
 			}
+
+			return $image_id || 'no image found';
 		}
 
 		// See https://github.com/felixarntz/ai-services/blob/main/docs/Accessing-AI-Services-in-PHP.md for further processing.
@@ -188,17 +191,17 @@ static function () {
 
 			\WP_CLI::debug( 'Making request...' . print_r( $contents, true ), 'ai' );
 
-			if ( $service->get_service_slug() === 'openai' ) {
-				$model = 'gpt-4o';
-			} else {
-				$model = 'gemini-2.0-flash';
-			}
+			// if ( $service->get_service_slug() === 'openai' ) {
+			// 	$model = 'gpt-4o';
+			// } else {
+			// 	$model = 'gemini-2.0-flash';
+			// }
 
 			$candidates = $service
 				->get_model(
 					[
 						'feature'          => 'text-generation',
-						'model'            => $model,
+						// 'model'            => $model,
 						                     'tools'        => $tools,
 							'capabilities' => [
 								AI_Capability::MULTIMODAL_INPUT,

src/MediaManager.php

@@ -0,0 +1,38 @@
+<?php
+
+namespace WP_CLI\AiCommand;
+
+class MediaManager {
+
+	public static function upload_to_media_library($media_path) {
+		// Get WordPress upload directory information
+		$upload_dir = wp_upload_dir();
+
+		// Get the file name from the path
+		$file_name = basename($media_path);
+
+		// Copy file to the upload directory
+		$new_file_path = $upload_dir['path'] . '/' . $file_name;
+		copy($media_path, $new_file_path);
+
+		// Prepare attachment data
+		$wp_filetype = wp_check_filetype($file_name, null);
+		$attachment = array(
+			'post_mime_type' => $wp_filetype['type'],
+			'post_title'     => sanitize_file_name($file_name),
+			'post_content'   => '',
+			'post_status'    => 'inherit'
+		);
+
+		// Insert the attachment
+		$attach_id = wp_insert_attachment($attachment, $new_file_path);
+
+		// Generate attachment metadata
+		require_once(ABSPATH . 'wp-admin/includes/image.php');
+		$attach_data = wp_generate_attachment_metadata($attach_id, $new_file_path);
+		wp_update_attachment_metadata($attach_id, $attach_data);
+
+		return $attach_id;
+
+	}
+}