Current Path : /storage/v11800/affypharma/public_html/wp-content/plugins/ai-engine/classes/queries/ |
Linux v11800 5.3.0-1023-aws #25~18.04.1-Ubuntu SMP Fri Jun 5 15:19:18 UTC 2020 aarch64
|
Current File : /storage/v11800/affypharma/public_html/wp-content/plugins/ai-engine/classes/queries/droppedfile.php |
<?php
class Meow_MWAI_Query_DroppedFile {
private $data;
private $rawData;
private $type; // Defines what the data is about ('refId', 'url', or 'data')
private $purpose; // Can be 'assistant' or 'vision' => this needs to be checked
private $mimeType; // 'image/jpeg' or any other mime type
static function from_url( $url, $purpose, $mimeType = null ) {
if ( empty( $mimeType ) ) {
$mimeType = Meow_MWAI_Core::get_mime_type( $url );
}
return new Meow_MWAI_Query_DroppedFile( $url, 'url', $purpose, $mimeType );
}
static function from_data( $data, $purpose, $mimeType = null ) {
return new Meow_MWAI_Query_DroppedFile( $data, 'data', $purpose, $mimeType );
}
static function from_path( $path, $purpose, $mimeType = null ) {
$data = file_get_contents( $path );
if ( empty( $mimeType ) ) {
$mimeType = Meow_MWAI_Core::get_mime_type( $path );
}
return new Meow_MWAI_Query_DroppedFile( $data, 'data', $purpose, $mimeType );
}
public function __construct( $data, $type, $purpose, $mimeType = null ) {
if ( !empty( $type ) && $type !== 'refId' && $type !== 'url' && $type !== 'data' ) {
throw new Exception( "AI Engine: The file type can only be refId, url or data." );
}
if ( !empty( $purpose ) && $purpose !== 'assistant-in' && $purpose !== 'vision' ) {
throw new Exception( "AI Engine: The file purpose can only be assistant or vision." );
}
$this->data = $data;
$this->type = $type;
$this->purpose = $purpose;
$this->mimeType = $mimeType;
}
public function get_url() {
if ( $this->type === 'url' ) {
return $this->data;
}
throw new Exception( "AI Engine: The file is not an URL." );
}
private function get_raw_data() {
if ( !empty( $this->rawData ) ) {
return $this->rawData;
}
if ( $this->type === 'url' ) {
$this->rawData = file_get_contents( $this->data );
return $this->rawData;
}
else if ( $this->type === 'data' ) {
return $this->data;
}
throw new Exception( "AI Engine: The file is not data or an URL." );
}
public function get_data() {
if ( $this->type === 'url' ) {
return $this->get_raw_data();
}
else if ( $this->type === 'data' ) {
return $this->data;
}
throw new Exception( "AI Engine: The file is not data or an URL." );
}
public function get_base64() {
$data = $this->get_raw_data();
return base64_encode( $data );
}
// Will return something like "data:image/jpeg;base64,{data}"
public function get_inline_base64_url() {
$b64 = $this->get_base64();
return "data:{$this->mimeType};base64,{$b64}";
}
public function get_type() {
return $this->type;
}
public function get_purpose() {
return $this->purpose;
}
public function get_mimeType() {
return $this->mimeType;
}
}