ai-1.0.x-dev/modules/ai_api_explorer/src/Plugin/AiApiExplorer/ImageAndAudioToVideoGenerator.php
modules/ai_api_explorer/src/Plugin/AiApiExplorer/ImageAndAudioToVideoGenerator.php
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 | <?php declare (strict_types=1); namespace Drupal\ai_api_explorer\Plugin\AiApiExplorer; use Drupal\Core\File\FileExists; use Drupal\Core\File\FileSystemInterface; use Drupal\Core\File\FileUrlGeneratorInterface; use Drupal\Core\Form\FormStateInterface; use Drupal\Core\StringTranslation\TranslatableMarkup; use Drupal\ai\AiProviderPluginManager; use Drupal\ai\OperationType\ImageAndAudioToVideo\ImageAndAudioToVideoInput; use Drupal\ai\Service\AiProviderFormHelper; use Drupal\ai_api_explorer\AiApiExplorerPluginBase; use Drupal\ai_api_explorer\Attribute\AiApiExplorer; use Drupal\ai_api_explorer\ExplorerHelper; use Symfony\Component\DependencyInjection\ContainerInterface; use Symfony\Component\HttpFoundation\RequestStack; /** * Plugin implementation of the ai_api_explorer. */ #[AiApiExplorer( id: 'image_and_audio_to_video_generator' , title: new TranslatableMarkup( 'Image-and-Audio-to-Video Explorer' ), description: new TranslatableMarkup( 'Contains a form where you can experiment and test the AI image and audio to video tool with prompts.' ), )] final class ImageAndAudioToVideoGenerator extends AiApiExplorerPluginBase { /** * Constructs the base plugin. * * @param array $configuration * A configuration array containing information about the plugin instance. * @param string $plugin_id * The plugin ID for the plugin instance. * @param mixed $plugin_definition * The plugin implementation definition. * @param \Symfony\Component\HttpFoundation\RequestStack $requestStack * The request stack. * @param \Drupal\ai\Service\AiProviderFormHelper $aiProviderHelper * The AI Provider Helper. * @param \Drupal\ai_api_explorer\ExplorerHelper $explorerHelper * The Explorer helper. * @param \Drupal\ai\AiProviderPluginManager $providerManager * The Provider Manager. * @param \Drupal\Core\File\FileUrlGeneratorInterface $fileUrlGenerator * The File Url Generator. * @param \Drupal\Core\File\FileSystemInterface $fileSystem * The File System. */ public function __construct( array $configuration , $plugin_id , $plugin_definition , RequestStack $requestStack , AiProviderFormHelper $aiProviderHelper , ExplorerHelper $explorerHelper , AiProviderPluginManager $providerManager , protected FileUrlGeneratorInterface $fileUrlGenerator , protected FileSystemInterface $fileSystem ) { parent::__construct( $configuration , $plugin_id , $plugin_definition , $requestStack , $aiProviderHelper , $explorerHelper , $providerManager ); } /** * {@inheritdoc} */ public static function create(ContainerInterface $container , array $configuration , $plugin_id , $plugin_definition ): static { return new static ( $configuration , $plugin_id , $plugin_definition , $container ->get( 'request_stack' ), $container ->get( 'ai.form_helper' ), $container ->get( 'ai_api_explorer.helper' ), $container ->get( 'ai.provider' ), $container ->get( 'file_url_generator' ), $container ->get( 'file_system' ), ); } /** * {@inheritDoc} */ public function isActive(): bool { return $this ->providerManager->hasProvidersForOperationType( 'image_and_audio_to_video' ); } /** * {@inheritdoc} */ public function buildForm( array $form , FormStateInterface $form_state ): array { // Get the query string for provider_id, model_id. $request = $this ->getRequest(); if ( $request ->query->get( 'provider_id' )) { $form_state ->setValue( 'ata_ai_provider' , $request ->query->get( 'provider_id' )); } if ( $request ->query->get( 'model_id' )) { $form_state ->setValue( 'ata_ai_model' , $request ->query->get( 'model_id' )); } $form = $this ->getFormTemplate( $form , 'ai-video-response' ); $form [ 'left' ][ 'file' ] = [ '#type' => 'file' , // Only mp3 files are allowed in this case, since that covers most models. '#accept' => '.mp3' , '#title' => $this ->t( 'Upload your audio here. When submitted, your provider will generate a response. Please note that each query counts against your API usage if your provider is a paid provider.' ), '#description' => $this ->t( 'Based on the complexity of your prompt, traffic, and other factors, a response can take time to complete. Please allow the operation to finish.' ), '#required' => TRUE, ]; $form [ 'left' ][ 'image' ] = [ '#type' => 'file' , '#accept' => '.jpg, .jpeg, .png' , '#title' => $this ->t( 'Upload your image here. When submitted, your provider will generate a response. Please note that each query counts against your API usage if your provider is a paid provider.' ), '#description' => $this ->t( 'Based on the complexity of your prompt, traffic, and other factors, a response can take time to complete. Please allow the operation to finish.' ), '#required' => TRUE, ]; // Load the LLM configurations. $this ->aiProviderHelper->generateAiProvidersForm( $form [ 'left' ], $form_state , 'image_and_audio_to_video' , 'ata' , AiProviderFormHelper::FORM_CONFIGURATION_FULL); $form [ 'left' ][ 'ata_ai_provider' ][ '#ajax' ][ 'callback' ] = $this :: class . '::loadModelsAjaxCallback' ; $form [ 'left' ][ 'submit' ] = [ '#type' => 'submit' , '#value' => $this ->t( 'Generate an Video File' ), '#ajax' => [ 'callback' => $this ->getAjaxResponseId(), 'wrapper' => 'ai-video-response' , ], ]; return $form ; } /** * {@inheritdoc} */ public function getResponse( array & $form , FormStateInterface $form_state ): array { $provider = $this ->aiProviderHelper->generateAiProviderFromFormSubmit( $form , $form_state , 'image_and_audio_to_video' , 'ata' ); $audio_file = $this ->generateFile(); $image_file = $this ->generateFile( 'image' ); if ( $audio_file && $image_file ) { $input = new ImageAndAudioToVideoInput( $image_file , $audio_file ); try { $video_normalized = $provider ->ImageAndAudioToVideo( $input , $form_state ->getValue( 'ata_ai_model' ), [ 'ai_api_explorer' ])->getNormalized(); } catch (\Exception $e ) { $form [ 'right' ][ 'response' ][ '#context' ][ 'ai_response' ][ 'response' ] = [ '#type' => 'inline_template' , '#template' => '{{ error|raw }}' , '#context' => [ 'error' => $this ->explorerHelper->renderException( $e ), ], ]; // Early return if we've hit an error. return $form [ 'right' ]; } // Save the binary data to a file. if ( $video_normalized ) { $file_url = $this ->fileSystem->saveData( $video_normalized ->getBinary(), 'public://image-and-audio-to-video-test.mp4' , FileExists::Replace); $form [ 'right' ][ 'response' ][ '#context' ][ 'ai_response' ][ 'response' ] = [ '#type' => 'inline_template' , '#template' => '{{ player|raw }}' , '#context' => [ 'player' => '<video controls><source src="' . $this ->fileUrlGenerator->generateAbsoluteString( $file_url ) . '" type="audio/mpeg"></video>' , ], ]; } } return $form [ 'right' ]; } } |