add bulk generating

Author: ezzcodeezzlife

README.md

@@ -81,6 +81,31 @@ Download to directory: C:\Users\pc\dalle2
 ```
 
 
+or generate a specific amount 
+
+```python
+from dalle2 import Dalle2
+
+dalle = Dalle2("sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxx")
+generations = dalle.generate_amount("portal to another dimension", 12) # Every generation has batch size 6 -> amount % 6 == 0 works best
+
+print(generations)
+```
+
+```
+✔️  Task created with ID: task-lm0V4nZasgAFasd7AsStE67 and PROMT: portal to another dimension OVERALL: 1/ 2
+⌛ Waiting for task to finish .. 
+➕  Appended new generations to all_generations
+✔️  Task created with ID: task-WcetZOHt8asdvHb433gi and PROMT: portal to another dimension OVERALL: 2/ 2
+⌛ Waiting for task to finish .. 
+➕  Appended new generations to all_generations
+🙌 Task completed!
+
+```
+```
+-> [list]
+```
+
 [![Test In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/drive/1EEgZNAI58V_OiEfRJQSsQV_xkhHzQeRB?usp=sharing)
 
 

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = dalle2
-version = 1.0.1
+version = 1.0.4
 author = ezzcodeezzlife
 author_email = [email protected]
 description = Use DALL·E 2 in Python

src/dalle2/dalle2.py

@@ -1,3 +1,4 @@
+from tokenize import String
 import requests
 import json
 import time 
@@ -58,6 +59,47 @@ def generate_and_download(self, promt):
 
             urllib.request.urlretrieve(imageurl, id +".jpg")
             print("✔️  Downloaded: " , id + ".jpg")
+
+    def generate_amount(self, promt, amount):
+        url = "https://labs.openai.com/api/labs/tasks"
+        headers = {
+            'Authorization': "Bearer " + self.bearer,
+            'Content-Type': "application/json"
+        }
+        body = {
+            "task_type": "text2im",
+            "prompt": {
+                "caption": promt,
+                "batch_size": self.batch_size,
+            }
+        }
+        
 
+        all_generations = []
+        for i in range(1,int(amount / self.batch_size +1)):
+            url = "https://labs.openai.com/api/labs/tasks"
+            response = requests.post(url, headers=headers, data=json.dumps(body))
+            if response.status_code != 200:
+                print(response.text)
+                return None
+            data = response.json()
+            print("✔️  Task created with ID:", data["id"], "and PROMT:", promt, "OVERALL:" , str(i)  + "/" , int(amount / self.batch_size ))
+            print("⌛ Waiting for task to finish .. ")
 
-            
+            while True:
+                url = "https://labs.openai.com/api/labs/tasks/" + data["id"]
+                response = requests.get(url, headers=headers)
+                data = response.json()
+                if data["status"] == "succeeded":
+                    
+                    generations = data["generations"]["data"]
+                    print("➕  Appended new generations to all_generations")
+                    all_generations.append(generations)
+                    break
+                else:
+                    # print("Task not completed yet")
+                    time.sleep(3)
+                    continue
+        print("🙌 Task completed!")
+        print(all_generations)
+        return all_generations