Skip to content

Commit 280fba6

Browse files
ezzcodeezzlifeezzcodeezzlife
committed
initial commit
0 parents  commit 280fba6

7 files changed

Lines changed: 366 additions & 0 deletions

File tree

‎.gitignore‎

Lines changed: 222 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,222 @@
1+
# Byte-compiled / optimized / DLL files
2+
__pycache__/
3+
*.py[cod]
4+
*$py.class
5+
6+
# C extensions
7+
*.so
8+
9+
# Distribution / packaging
10+
.Python
11+
build/
12+
develop-eggs/
13+
dist/
14+
downloads/
15+
eggs/
16+
.eggs/
17+
lib/
18+
lib64/
19+
parts/
20+
sdist/
21+
var/
22+
wheels/
23+
share/python-wheels/
24+
*.egg-info/
25+
.installed.cfg
26+
*.egg
27+
MANIFEST
28+
29+
# PyInstaller
30+
# Usually these files are written by a python script from a template
31+
# before PyInstaller builds the exe, so as to inject date/other infos into it.
32+
*.manifest
33+
*.spec
34+
35+
# Installer logs
36+
pip-log.txt
37+
pip-delete-this-directory.txt
38+
39+
# Unit test / coverage reports
40+
htmlcov/
41+
.tox/
42+
.nox/
43+
.coverage
44+
.coverage.*
45+
.cache
46+
nosetests.xml
47+
coverage.xml
48+
*.cover
49+
*.py,cover
50+
.hypothesis/
51+
.pytest_cache/
52+
cover/
53+
54+
# Translations
55+
*.mo
56+
*.pot
57+
58+
# Django stuff:
59+
*.log
60+
local_settings.py
61+
db.sqlite3
62+
db.sqlite3-journal
63+
64+
# Flask stuff:
65+
instance/
66+
.webassets-cache
67+
68+
# Scrapy stuff:
69+
.scrapy
70+
71+
# Sphinx documentation
72+
docs/_build/
73+
74+
# PyBuilder
75+
.pybuilder/
76+
target/
77+
78+
# Jupyter Notebook
79+
.ipynb_checkpoints
80+
81+
# IPython
82+
profile_default/
83+
ipython_config.py
84+
85+
# pyenv
86+
# For a library or package, you might want to ignore these files since the code is
87+
# intended to run in multiple environments; otherwise, check them in:
88+
# .python-version
89+
90+
# pipenv
91+
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92+
# However, in case of collaboration, if having platform-specific dependencies or dependencies
93+
# having no cross-platform support, pipenv may install dependencies that don't work, or not
94+
# install all needed dependencies.
95+
#Pipfile.lock
96+
97+
# poetry
98+
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
99+
# This is especially recommended for binary packages to ensure reproducibility, and is more
100+
# commonly ignored for libraries.
101+
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
102+
#poetry.lock
103+
104+
# pdm
105+
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
106+
#pdm.lock
107+
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
108+
# in version control.
109+
# https://pdm.fming.dev/#use-with-ide
110+
.pdm.toml
111+
112+
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
113+
__pypackages__/
114+
115+
# Celery stuff
116+
celerybeat-schedule
117+
celerybeat.pid
118+
119+
# SageMath parsed files
120+
*.sage.py
121+
122+
# Environments
123+
.env
124+
.venv
125+
env/
126+
venv/
127+
ENV/
128+
env.bak/
129+
venv.bak/
130+
131+
# Spyder project settings
132+
.spyderproject
133+
.spyproject
134+
135+
# Rope project settings
136+
.ropeproject
137+
138+
# mkdocs documentation
139+
/site
140+
141+
# mypy
142+
.mypy_cache/
143+
.dmypy.json
144+
dmypy.json
145+
146+
# Pyre type checker
147+
.pyre/
148+
149+
# pytype static type analyzer
150+
.pytype/
151+
152+
# Cython debug symbols
153+
cython_debug/
154+
155+
# PyCharm
156+
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
157+
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
158+
# and can be added to the global gitignore or merged into this file. For a more nuclear
159+
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
160+
#.idea/
161+
162+
# Byte-compiled / optimized / DLL files
163+
__pycache__/
164+
*.py[cod]
165+
166+
# C extensions
167+
*.so
168+
169+
# Distribution / packaging
170+
bin/
171+
build/
172+
develop-eggs/
173+
dist/
174+
eggs/
175+
lib/
176+
lib64/
177+
parts/
178+
sdist/
179+
var/
180+
*.egg-info/
181+
.installed.cfg
182+
*.egg
183+
184+
# Installer logs
185+
pip-log.txt
186+
pip-delete-this-directory.txt
187+
188+
# Unit test / coverage reports
189+
.tox/
190+
.coverage
191+
.cache
192+
nosetests.xml
193+
coverage.xml
194+
195+
# Translations
196+
*.mo
197+
198+
# Mr Developer
199+
.mr.developer.cfg
200+
.project
201+
.pydevproject
202+
203+
# Rope
204+
.ropeproject
205+
206+
# Django stuff:
207+
*.log
208+
*.pot
209+
210+
# Sphinx documentation
211+
docs/_build/
212+
213+
.installed.cfg
214+
bin
215+
develop-eggs
216+
dist
217+
downloads
218+
eggs
219+
parts
220+
src/*.egg-info
221+
lib
222+
lib64

‎LICENSE‎

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
Copyright (c) 2018 The Python Packaging Authority
2+
3+
Permission is hereby granted, free of charge, to any person obtaining a copy
4+
of this software and associated documentation files (the "Software"), to deal
5+
in the Software without restriction, including without limitation the rights
6+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
7+
copies of the Software, and to permit persons to whom the Software is
8+
furnished to do so, subject to the following conditions:
9+
10+
The above copyright notice and this permission notice shall be included in all
11+
copies or substantial portions of the Software.
12+
13+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
18+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
19+
SOFTWARE.

‎README.md‎

Lines changed: 53 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,53 @@
1+
# Get Access
2+
3+
[labs.openai.com/waitlist](https://labs.openai.com/waitlist)
4+
5+
- Go to https://labs.openai.com/
6+
- Open Network Tab in Developer Tools
7+
- Type a promt and press "Generate"
8+
- Look for fetch to https://labs.openai.com/api/labs/tasks
9+
- In the request header look for authorization then get the Bearer Token
10+
11+
12+
# Usage
13+
```bash
14+
pip install dalle2
15+
```
16+
```python
17+
from dalle2 import Dalle2
18+
19+
dalle = Dalle2("sess-xxxxxxxxxxxxxxxxxxxxxxxxxxxx")
20+
generations = dalle.generate("portal to another dimension, digital art")
21+
print(generations)
22+
```
23+
24+
```
25+
[
26+
{
27+
'id': 'generation-sCnERSYDPP0Zu14fsdXEcKmL',
28+
'object': 'generation',
29+
'created': 1553332711,
30+
'generation_type': 'ImageGeneration',
31+
'generation': {
32+
'image_path': 'https://openailabsprodscus.blob.core.windows.net/private/user-hadpVzldsfs28CwvEZYMUT/generations/generation...'
33+
},
34+
'task_id': 'task-nERkiKsdjVCSZ50yD69qewID',
35+
'prompt_id': 'prompt-2CtaLQsgUbJHHDoJQy9Lul3T',
36+
'is_public': false
37+
},
38+
{
39+
'id': 'generation-hZWt2Nasrx8R0tJjbaROfKVy',
40+
'object': 'generation',
41+
'created': 1553332711,
42+
'generation_type': 'ImageGeneration',
43+
'generation': {
44+
'image_path': 'https://openailabsprodscus.blob.core.windows.net/private/user-hadpVzldsfs28CwvEZYMUT/generations/generation...'
45+
},
46+
'task_id': 'task-nERkiKhjasdSZ50yD69qewID',
47+
'prompt_id': 'prompt-2CtaLasdUbJHHfoJQy9Lul3T',
48+
'is_public': false
49+
},
50+
// 8 more ...
51+
]
52+
```
53+

‎pyproject.toml‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
[build-system]
2+
requires = ["setuptools>=42"]
3+
build-backend = "setuptools.build_meta"

‎setup.cfg‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
[metadata]
2+
name = dalle2
3+
version = 0.0.4
4+
author = ezzcodeezzlife
5+
author_email = author@example.com
6+
description = Use DALL·E 2 in Python
7+
long_description = file: README.md
8+
long_description_content_type = text/markdown
9+
url = https://github.com/xx/xx
10+
project_urls =
11+
Bug Tracker = https://github.com/xx/xx/issues
12+
classifiers =
13+
Programming Language :: Python :: 3
14+
License :: OSI Approved :: MIT License
15+
Operating System :: OS Independent
16+
17+
[options]
18+
package_dir =
19+
= src
20+
packages = find:
21+
python_requires = >=3.6
22+
23+
[options.packages.find]
24+
where = src

‎src/dalle2/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
from .dalle2 import Dalle2

‎src/dalle2/dalle2.py‎

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
import requests
2+
import json
3+
import time
4+
5+
class Dalle2():
6+
def __init__(self, bearer):
7+
self.bearer = bearer
8+
9+
def generate(self, promt):
10+
# make post request to https://labs.openai.com/api/labs/tasks with header: authorization: "Bearer asdasdasdas" and data: data: { task_type: "text2im", prompt: {caption: promt,batch_size: 6,},}
11+
url = "https://labs.openai.com/api/labs/tasks"
12+
headers = {
13+
'Authorization': "Bearer " + self.bearer,
14+
'Content-Type': "application/json"
15+
}
16+
body = {
17+
"task_type": "text2im",
18+
"prompt": {
19+
"caption": promt,
20+
"batch_size": 10,
21+
}
22+
}
23+
24+
response = requests.post(url, headers=headers, data=json.dumps(body))
25+
if response.status_code != 200:
26+
print(response.text)
27+
return None
28+
data = response.json()
29+
print("✔️ Task created with ID:", data["id"], "and PROMT:", promt)
30+
print("⌛ Waiting for task to finish .. ")
31+
32+
while True:
33+
url = "https://labs.openai.com/api/labs/tasks/" + data["id"]
34+
response = requests.get(url, headers=headers)
35+
data = response.json()
36+
if data["status"] == "succeeded":
37+
print("🙌 Task completed!")
38+
generations = data["generations"]["data"]
39+
return(generations)
40+
else:
41+
# print("Task not completed yet")
42+
time.sleep(3)
43+
continue
44+

0 commit comments

Comments
 (0)