Apply TR suggestions, add requirements file

Co-authored-by: Bartosz <[email protected]>

Author: martin-martin

openai-dalle/convert.py

@@ -2,16 +2,16 @@
 from base64 import b64decode
 from pathlib import Path
 
-JSON_FILE_NAME = "An ec-1667994848"
-IMAGE_DIR = Path.cwd() / "images" / JSON_FILE_NAME
+JSON_FILE = Path("An ec-1668602437.json")
+IMAGE_DIR = Path.cwd() / "images" / JSON_FILE.stem
 DATA_DIR = Path.cwd() / "responses"
 
 IMAGE_DIR.mkdir(parents=True, exist_ok=True)
 
-with open(DATA_DIR / f"{JSON_FILE_NAME}.json", "r") as f:
-    response = json.load(f)
+with open(DATA_DIR / JSON_FILE, "r") as file:
+    response = json.load(file)
 
 for index, image_dict in enumerate(response["data"]):
     image_data = b64decode(image_dict["b64_json"])
-    with open(IMAGE_DIR / f"{JSON_FILE_NAME}-{index}.png", "wb") as f:
-        f.write(image_data)
+    with open(IMAGE_DIR / f"{JSON_FILE.stem}-{index}.png", "wb") as png_file:
+        png_file.write(image_data)

openai-dalle/create.py

@@ -20,5 +20,5 @@
 
 file_name = DATA_DIR / f"{PROMPT[:5]}-{response['created']}.json"
 
-with open(file_name, mode="w") as file_path:
-    json.dump(response, file_path)
+with open(file_name, mode="w", encoding="utf-8") as file:
+    json.dump(response, file)

openai-dalle/edit.py

@@ -25,5 +25,6 @@
 with open(
     DESTINATION_PATH / f"edit-{PROMPT[:5]}-{response['created']}.json",
     mode="w",
-) as fp:
-    json.dump(response, fp)
+    encoding="utf-8",
+) as file:
+    json.dump(response, file)

openai-dalle/requirements.txt

@@ -0,0 +1,16 @@
+certifi==2022.9.24
+charset-normalizer==2.1.1
+et-xmlfile==1.1.0
+idna==3.4
+numpy==1.23.4
+openai==0.25.0
+openpyxl==3.0.10
+pandas==1.5.1
+pandas-stubs==1.2.0.62
+python-dateutil==2.8.2
+pytz==2022.6
+requests==2.28.1
+six==1.16.0
+tqdm==4.64.1
+typing_extensions==4.4.0
+urllib3==1.26.12

openai-dalle/vary.py

@@ -23,5 +23,5 @@
 
 new_file_name = f"vary-{SOURCE_FILE[:4]}-{response['created']}.json"
 
-with open(DATA_DIR / new_file_name, mode="w") as new_json_file:
-    json.dump(response, new_json_file)
+with open(DATA_DIR / new_file_name, mode="w", encoding="utf-8") as file:
+    json.dump(response, file)