FERMAT_TEST_REST_SERVER/GLASSES_AI_SERVER
4
0
Fork 0
Code Issues Pull Requests Projects Releases 1 Wiki Activity

edit : 이미지 생성시 생성 갯수 지정 가능

Browse Source
This commit is contained in:
김재웅
2025-01-17 13:26:19 +09:00
parent bb54fdd85e
commit c296204541
6 changed files with 36 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
custom_apps/dalle3/BingImageCreator.py
View File

@@ -196,12 +196,13 @@ class ImageGenAsync:
os.mkdir(output_dir)
try:
model = "dalle3"
jpeg_index = 0
jpeg_index = 1
parsing_file_name = prompt_to_filenames(prompt)
_datetime = D.date_file_name()
for link in links[:download_count]:
if download_count == 1:
_path = os.path.join(output_dir, f"{model}_{parsing_file_name}_{D.date_file_name()}.jpg")
_path = os.path.join(output_dir, f"{model}_{parsing_file_name}_{_datetime}.jpg")
if os.path.exists(_path):
raise Exception("파일 이미 존재함")
@@ -216,7 +217,7 @@ class ImageGenAsync:
jpeg_index += 1
else:
_path = os.path.join(output_dir, f"{model}_{parsing_file_name}_{jpeg_index}_{D.date_file_name()}.jpg")
_path = os.path.join(output_dir, f"{model}_{parsing_file_name}_{jpeg_index}_{_datetime}.jpg")
if os.path.exists(_path):
raise Exception("파일 이미 존재함")

5
custom_apps/imagen/custom_imagen.py
View File

@@ -22,6 +22,7 @@ def imagen_generate_image(prompt,download_count=1):
_file_name = prompt_to_filenames(prompt)
_folder = os.path.join(OUTPUT_FOLDER,"imagen")
_datetime = D.date_file_name()
if not os.path.isdir(_folder):
raise FileExistsError(f"FileExistsError: {_folder}")
@@ -40,7 +41,7 @@ def imagen_generate_image(prompt,download_count=1):
)
if len(images.images) <=1 :
images[0].save(location=os.path.join(_folder,f"imagen_{_file_name}_{D.date_file_name()}.png"), include_generation_parameters=False)
images[0].save(location=os.path.join(_folder,f"imagen_{_file_name}_{_datetime}.png"), include_generation_parameters=False)
else:
for i in range(len(images.images)):
images[i].save(location=os.path.join(_folder,f"imagen_{_file_name}_{i}_{D.date_file_name()}.png"), include_generation_parameters=False)
images[i].save(location=os.path.join(_folder,f"imagen_{_file_name}_{i+1}_{_datetime}.png"), include_generation_parameters=False)