mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added function to check output path arg and override, other wise, use temp or fallback to config

Browse Source
This commit is contained in:
eddyizm 2024-01-27 22:25:17 -08:00
parent 1bcbd6501b
commit fecb97b59c
No known key found for this signature in database
GPG Key ID: 2367DCDC2F004E40
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
modules/private_logger.py
View File

@ -11,8 +11,20 @@ from modules.util import generate_temp_filename
log_cache = {} log_cache = {}
def get_path_output() -> str:
''' Checking output path argument and overriding default path
if disable image log path is not enabled.
'''
path_output = args_manager.args.temp_path if args_manager.args.disable_image_log else modules.config.path_outputs
if not args_manager.args.disable_image_log and args_manager.args.output_path:
print(f'[Config] Overriding path output to: {args_manager.args.output_path}')
path_output = args_manager.args.output_path
return path_output
def get_current_html_path(): def get_current_html_path():
date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs, path_outputs = get_path_output()
date_string, local_temp_filename, only_name = generate_temp_filename(folder=path_outputs,
extension='png') extension='png')
html_name = os.path.join(os.path.dirname(local_temp_filename), 'log.html') html_name = os.path.join(os.path.dirname(local_temp_filename), 'log.html')
return html_name return html_name
@ -21,8 +33,8 @@ def get_current_html_path():
def log(img, dic): def log(img, dic):
if args_manager.args.disable_image_log: if args_manager.args.disable_image_log:
return return
path_outputs = get_path_output()
date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs, extension='png') date_string, local_temp_filename, only_name = generate_temp_filename(folder=path_outputs, extension='png')
os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True) os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True)
Image.fromarray(img).save(local_temp_filename) Image.fromarray(img).save(local_temp_filename)
html_name = os.path.join(os.path.dirname(local_temp_filename), 'log.html') html_name = os.path.join(os.path.dirname(local_temp_filename), 'log.html')