Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add markdown log option

Browse Source
This commit is contained in:
冯不游 2022-09-16 17:21:54 +08:00 committed by Lincoln Stein
parent e33ed45cfc
commit a28636dd4a
1 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
scripts/dream.py
View File

@ -292,8 +292,8 @@ def main_loop(gen, opt, infile):
continue continue
print('Outputs:') print('Outputs:')
log_path = os.path.join(current_outdir, 'dream_log.txt') log_path = os.path.join(current_outdir, 'dream_log')
write_log_message(results, log_path) write_log_message(results, log_path ,('txt', 'md'))
print() print()
print('goodbye!') print('goodbye!')
@ -339,17 +339,29 @@ def dream_server_loop(gen, host, port, outdir):
dream_server.server_close() dream_server.server_close()
def write_log_message(results, log_path): def write_log_message(results, log_path, file_types):
"""logs the name of the output image, prompt, and prompt args to the terminal and log file""" """logs the name of the output image, prompt, and prompt args to the terminal and log file"""
global output_cntr global output_cntr
log_lines = [f'{path}: {prompt}\n' for path, prompt in results] log_lines = [f'{path}: {prompt}\n' for path, prompt in results]
for l in log_lines: for l in log_lines:
output_cntr += 1 output_cntr += 1
print(f'[{output_cntr}] {l}',end='') print(f'[{output_cntr}] {l}',end='')
for file_type in file_types:
if file_type == 'txt':
with open(log_path + '.txt', 'a', encoding='utf-8') as file:
file.writelines(log_lines)
elif file_type == 'md' or file_type == 'markdown':
md_lines = []
for path,prompt in results:
file_name = os.path.basename(path)
md_lines.append(f'## {file_name}\n![]({file_name})\n{prompt}\n')
with open(log_path + '.md', 'a', encoding='utf-8') as file:
file.writelines(md_lines)
else:
print(f"'{file_type}' format is not supported, so write in plain text")
with open(log_path + '.' + file_type, 'a', encoding='utf-8') as file:
file.writelines(log_lines)
with open(log_path, 'a', encoding='utf-8') as file:
file.writelines(log_lines)
if __name__ == '__main__': if __name__ == '__main__':
main() main()