added vivid and b64 in dall-e3

jrcastropy · Jan 25, 2024 · 693b627 · 693b627
1 parent aa8854e
commit 693b627
Show file tree

Hide file tree

Showing 5 changed files with 595 additions and 97 deletions.
diff --git a/.gitignore b/.gitignore
@@ -5,4 +5,5 @@ build/
 test.py
 archives/
 saved_images/
-*__pycache__*
+*__pycache__*
+.env
diff --git a/images/test2.png b/images/test2.png
diff --git a/serverless_openai/apis.py b/serverless_openai/apis.py
@@ -147,7 +147,9 @@ def dall_e(
             prompt: str,
             model: Union[ImageCreationModels, str] = ImageCreationModels.dalle_3,
             n: int = 1,
-            size: str = "1024x1024",
+            response_format: str = 'url', # or b64_json
+            style: str = 'natural', # or vivid
+            size: str = "1024x1024", # 1792x1024 or 1024x1792
             timeout: int = 500,
             tries: int = 5
         ) -> OpenAIResults:
@@ -157,12 +159,15 @@ def dall_e(
             "prompt": prompt,
             "n": n,
             "size": size,
+            "response_format": response_format,
+            "style": style
         }
 
         results = {}
         for _ in range(tries):
             results = requests.post(self.imagecreation_url, headers=self.headers, json=data, timeout=timeout).json()
-            print(results)
+            if response_format == 'b64_json':
+                return OpenAIResults(result=results['data'][0]['b64_json'], result_json=results)
             return OpenAIResults(result=results['data'][0]['url'], result_json=results)
         return OpenAIResults(result=False, result_json=results)
 

diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-VERSION = '1.2.6'
+VERSION = '1.2.7'
 DESCRIPTION = "A package for using Openai in serverless environment"
 LONG_DESCRIPTION = 'A package for using Openai with scraping and etc. in serverless application such as AWS Lambda and GCP Cloud Function'
 

diff --git a/test.ipynb b/test.ipynb