update

iMactool · Jun 7, 2023 · 6406602 · 6406602
1 parent 0438b57
commit 6406602
Show file tree

Hide file tree

Showing 4 changed files with 88 additions and 16 deletions.
diff --git a/src/Prompt.php b/src/Prompt.php
@@ -66,6 +66,9 @@ public function toString($inputParams = []): string
                 if (array_key_exists($unsetKey, $inputParams)) {
                     unset($inputParams[$unsetKey]);
                 }
+                if (array_key_exists('prompt', $inputParams)) {
+                    unset($inputParams['prompt']);
+                }
                 $prompt .= ', ' . implode(', ', array_values(array_unique($inputParams)));
             }
 

diff --git a/src/Replicate.php b/src/Replicate.php
@@ -30,6 +30,8 @@ class Replicate
 
         private string $version = '';
 
+        private string $webhook = '';
+
         private function __construct(
             public ?Prompt $prompt = null,
             private int $width = 512,
@@ -158,6 +160,12 @@ public function inputParams(string $key, mixed $value)
             return $this;
         }
 
+        public function setWebHook(string $url)
+        {
+            $this->webhook = $url;
+            return $this;
+        }
+
         public function width(int $width)
         {
             assert($width > 0, 'Width must be greater than 0');
@@ -199,13 +207,19 @@ public function generate()
             }
             $input = array_merge($input, $this->inputParams);
 
+            $json = [
+                'version' => $this->getVersion(),
+                'input' => $input,
+            ];
+
+            if (! empty($this->webhook)) {
+                $json['webhook'] = $this->webhook;
+            }
+
             $response = $this->client()->post(
                 $this->getBaseUrl(),
                 [
-                    'json' => [
-                        'version' => $this->getVersion(),
-                        'input' => $input,
-                    ],
+                    'json' => $json,
                 ]
             );
 

diff --git a/src/StableDiffusion.php b/src/StableDiffusion.php
@@ -43,6 +43,7 @@ public function supportApi(string $api = ''): array|string
             $aiApis = [
                 'StableDiffusionApiV3', // Stable Diffusion V3 APIs comes with below features https://documenter.getpostman.com/view/18679074/2s83zdwReZ#c7e3c6a0-b57d-4d17-ad5a-c4eb8571021f
                 'DreamboothApiV4', // [Beta] DreamBooth API https://documenter.getpostman.com/view/18679074/2s83zdwReZ#27db9713-6068-41c2-8431-ada0d08d3cd5
+                'EnterpriseApiV1', // [企业] Enterprise API  https://stablediffusionapi.com/docs/enterprise-plan/overview
             ];
 
             if ($api) {
@@ -206,20 +207,39 @@ public function superResolution()
               return json_decode($response->getBody()->getContents(), true);
           }
 
+        public function controlnet()
+        {
+            if (empty($this->payload)) {
+                throw new Exception('Invalid payload. @see https://stablediffusionapi.com/docs/controlnet-main/');
+            }
+
+            $response = $this->client()->post(
+                'https://stablediffusionapi.com/api/v5/controlnet',
+                [
+                    'json' => $this->payload,
+                ]
+            );
+
+            return json_decode($response->getBody()->getContents(), true);
+//            $this->saveResult($result, $this->apiBase->text2imgUrl());
+        }
+
         private function saveResult($result, $url)
         {
-            $data = [
-                'replicate_id' => Arr::has($result, 'id') ? Arr::get($result, 'id') : 0,
-                'platform' => self::$platform,
-                'user_prompt' => isset($this->payload['prompt']) ? $this->payload['prompt'] : '',
-                'full_prompt' => $this->payloadArr2String(),
-                'url' => $url,
-                'status' => Arr::has($result, 'status') ? Arr::get($result, 'status') : '',
-                'output' => Arr::has($result, 'output') ? Arr::get($result, 'output') : '',
-                'error' => Arr::has($result, 'error') ? Arr::get($result, 'error') : null,
-                'predict_time' => Arr::has($result, 'generationTime') ? Arr::get($result, 'generationTime') : null,
-            ];
-            StableDiffusionResult::create($data);
+            if (Arr::get($result, 'status') !== 'error') {
+                $data = [
+                    'replicate_id' => Arr::has($result, 'id') ? Arr::get($result, 'id') : 0,
+                    'platform' => self::$platform,
+                    'user_prompt' => isset($this->payload['prompt']) ? $this->payload['prompt'] : '',
+                    'full_prompt' => $this->payloadArr2String(),
+                    'url' => $url,
+                    'status' => Arr::has($result, 'status') ? Arr::get($result, 'status') : '',
+                    'output' => Arr::has($result, 'output') ? Arr::get($result, 'output') : '',
+                    'error' => Arr::has($result, 'error') ? Arr::get($result, 'error') : null,
+                    'predict_time' => Arr::has($result, 'generationTime') ? Arr::get($result, 'generationTime') : null,
+                ];
+                StableDiffusionResult::create($data);
+            }
         }
 
         private function client(): ClientInterface

diff --git a/src/Uri/EnterpriseApiV1.php b/src/Uri/EnterpriseApiV1.php
@@ -0,0 +1,35 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of the imactool/hyperf-stable-diffusion.
+ *
+ * (c) imactool <[email protected]>
+ * This source file is subject to the MIT license that is bundled
+ * with this source code in the file LICENSE.
+ */
+namespace Imactool\HyperfStableDiffusion\Uri;
+
+class EnterpriseApiV1
+{
+    public const ORIGIN = 'https://stablediffusionapi.com/api';
+
+    public const API_VERSION = 'v1';
+
+    public const OPEN_AI_URL = self::ORIGIN . '/' . self::API_VERSION;
+
+    public function text2imgUrl(): string
+    {
+        return self::OPEN_AI_URL . '/enterprise/text2img';
+    }
+
+    public function img2imgUrl(): string
+    {
+        return self::OPEN_AI_URL . '/enterprise/img2img';
+    }
+
+    public function inpaintUrl(): string
+    {
+        return self::OPEN_AI_URL . '/enterprise/inpaint';
+    }
+}