feat: image creator boosts

Author: maximerenou

README.md

@@ -40,9 +40,10 @@ Edit and run `examples/chat.php` to test it.
 use MaximeRenou\BingAI\BingAI;
 use MaximeRenou\BingAI\Chat\Prompt;
 
-$ai = new BingAI;
 // $cookie - your "_U" cookie from bing.com
-$conversation = $ai->createChatConversation($cookie);
+$ai = new BingAI($cookie);
+
+$conversation = $ai->createChatConversation();
 
 // $text - Text-only version of Bing's answer
 // $messages - Message objects array
@@ -70,7 +71,7 @@ list($final_text, $final_messages) = $conversation->ask($prompt, function ($text
   <summary>Locale and location preferences</summary>
 
 ```php
-$conversation = $ai->createChatConversation($cookie)
+$conversation = $ai->createChatConversation()
     ->withLocation($latitude, $longitude, $radius) // Optional
     ->withPreferences('fr-FR', 'fr-FR', 'FR'); // Optional
 ```
@@ -87,8 +88,8 @@ If you want to resume a previous conversation, you can retrieve its identifiers:
 $identifiers = $conversation->getIdentifiers();
 
 // ...
-// Resume conversation with $identifiers parameter, and number of previous questions
-$conversation = $ai->createChatConversation($cookie, $identifiers, 1);
+// Resume conversation with $identifiers parameter, and number of previous questions asked
+$conversation = $ai->resumeChatConversation($identifiers, 1);
 ```
 
 </details> 
@@ -132,14 +133,13 @@ if ($remaining === 0) {
 
 Edit and run `examples/images.php` to test it.
 
-> Image generation becomes slower after using it a few times. Bing limits the number of images generated fast per user.
-
 ```php
 use MaximeRenou\BingAI\BingAI;
 
-$ai = new BingAI;
 // $cookie - your "_U" cookie from bing.com
-$creator = $ai->createImages($cookie, "A 3D teddy bear");
+$ai = new BingAI($cookie);
+
+$creator = $ai->createImages("A 3D teddy bear");
 
 $creator->wait();
 
@@ -149,20 +149,34 @@ if (! $creator->hasFailed()) {
 }
 ```
 
+> Image generation can become slower after consuming all of your "boosts". Check the section below to stay aware of your remaining boosts.
+
+<details>
+  <summary>Check remaining boosts</summary>
+
+```php
+$creator = $ai->getImageCreator();
+
+$remaining_boosts = $creator->getRemainingBoosts();
+```
+
+</details>
+
 <details>
   <summary>Asynchronous generation</summary>
 You may quit after calling `createImages()` and check generation later using its ID:
 
 ```php
 $prompt = "A 3D teddy bear";
-$creator = $ai->createImages($cookie, $prompt);
+$creator = $ai->createImages($prompt);
 $generation_id = $creator->getGenerationId();
 
 // ...
 
-$creator = new ImageCreator($cookie);
+$creator = $ai->getImageCreator();
 $creator->resume($generation_id, $prompt);
 ```
+
 </details>
 
 <details>
@@ -174,6 +188,7 @@ do {
     sleep(1);
 } while ($creator->isGenerating());
 ```
+
 </details>
 
 ---------------------------------------

examples/chat.php

@@ -6,10 +6,9 @@
 
 date_default_timezone_set("Europe/Paris");
 
-$ai = new \MaximeRenou\BingAI\BingAI();
+$ai = new \MaximeRenou\BingAI\BingAI($cookie);
 
-$conversation = $ai->createChatConversation($cookie)
-    ->withPreferences('fr-FR', 'fr-FR', 'FR');
+$conversation = $ai->createChatConversation();
 
 \MaximeRenou\BingAI\Tools::$debug = false; // Set true for verbose
 

examples/images.php

@@ -4,10 +4,14 @@
 
 require __DIR__ . '/../vendor/autoload.php';
 
-$ai = new \MaximeRenou\BingAI\BingAI();
+$ai = new \MaximeRenou\BingAI\BingAI($cookie);
 
 \MaximeRenou\BingAI\Tools::$debug = false; // Set true for verbose
 
+$boosts = $ai->getImageCreator()->getRemainingBoosts();
+
+echo "You have $boosts remaining boosts." . PHP_EOL;
+
 echo 'Type "q" to quit' . PHP_EOL;
 
 echo PHP_EOL . "> ";
@@ -16,7 +20,7 @@
 if ($text == 'q')
     exit(0);
 
-$creator = $ai->createImages($cookie, $text);
+$creator = $ai->createImages($text);
 
 echo 'Generating...' . PHP_EOL;
 

src/BingAI.php

@@ -7,20 +7,30 @@
 
 class BingAI
 {
-    public function __construct()
+    protected $cookie;
+
+    public function __construct($cookie)
+    {
+        $this->cookie = $cookie;
+    }
+
+    public function createChatConversation($data = null)
     {
-        //
+        return new Conversation($this->cookie, $data, 0);
     }
 
-    public function createChatConversation($cookie, $data = null, $invocations = 0)
+    public function resumeChatConversation($data = null, $invocations = 1)
     {
-        return new Conversation($cookie, $data, $invocations);
+        return new Conversation($this->cookie, $data, $invocations);
     }
 
-    public function createImages($cookie, $prompt)
+    public function getImageCreator()
     {
-        $creator = new ImageCreator($cookie);
+        return new ImageCreator($this->cookie);
+    }
 
-        return $creator->create($prompt);
+    public function createImages($prompt)
+    {
+        return $this->getImageCreator()->create($prompt);
     }
 }

src/Images/ImageCreator.php

@@ -134,6 +134,28 @@ public function isGenerating()
         return $this->generating;
     }
 
+    public function getRemainingBoosts()
+    {
+        $request = curl_init();
+        curl_setopt($request, CURLOPT_RETURNTRANSFER, true);
+        curl_setopt($request, CURLOPT_FOLLOWLOCATION, true);
+        curl_setopt($request, CURLOPT_URL, "https://www.bing.com/images/create");
+        curl_setopt($request, CURLOPT_HTTPHEADER, [
+            'cookie: _U=' . $this->cookie,
+            'method: GET',
+            'accept: text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
+            "accept-language: en;q=0.9",
+        ]);
+
+        $data = curl_exec($request);
+        curl_close($request);
+
+        if (! is_string($data) || ! preg_match('/<div id="token_bal" aria-label="[^"]+">([0-9]+)<\/div>/', $data, $matches))
+            return 0;
+
+        return intval($matches[1]);
+    }
+
     public function wait()
     {
         while (! $this->hasFailed()) {