Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

moved image extension to webui via async worker. Passing as parameter to log and get_current_html_path functions per feedback

Browse Source
This commit is contained in:
eddyizm
2024-01-12 22:49:07 -08:00
parent 9dfa1a99e3
commit ab3acb7817
4 changed files with 13 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/private_logger.py
+6 -4
View File
@@ -11,18 +11,20 @@ from modules.util import generate_temp_filename
log_cache = {}
def get_current_html_path():
def get_current_html_path(image_extension=None):
_image_extension = image_extension if image_extension else modules.config.default_image_extension
date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs,
extension=modules.config.default_image_extension)
extension=_image_extension)
html_name = os.path.join(os.path.dirname(local_temp_filename), 'log.html')
return html_name
def log(img, dic):
def log(img, dic, image_extension=None):
if args_manager.args.disable_image_log:
return
date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs, extension=modules.config.default_image_extension)
_image_extension = image_extension if image_extension else modules.config.default_image_extension
date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs, extension=_image_extension)
os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True)
Image.fromarray(img).save(local_temp_filename)
html_name = os.path.join(os.path.dirname(local_temp_filename), 'log.html')