37 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			37 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| import os
 | |
| import modules.path
 | |
| 
 | |
| from PIL import Image
 | |
| from modules.util import generate_temp_filename
 | |
| 
 | |
| 
 | |
| def log(img, dic):
 | |
|     date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.path.temp_outputs_path, extension='png')
 | |
|     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')
 | |
| 
 | |
|     if not os.path.exists(html_name):
 | |
|         with open(html_name, 'a+') as f:
 | |
|             f.write(f"<p>Fooocus Log {date_string} (private)</p>\n")
 | |
|             f.write(f"<p>All images do not contain any hidden data.</p>")
 | |
| 
 | |
|     with open(html_name, 'a+') as f:
 | |
|         f.write(f"<hr>\n")
 | |
|         f.write(f"<p>{only_name}</p>\n")
 | |
|         i = 0
 | |
|         for k, v in dic:
 | |
|             if i < 2:
 | |
|                 f.write(f"<p>{k}: <b>{v}</b> </p>\n")
 | |
|             else:
 | |
|                 if i % 2 == 0:
 | |
|                     f.write(f"<p>{k}: <b>{v}</b>, ")
 | |
|                 else:
 | |
|                     f.write(f"{k}: <b>{v}</b></p>\n")
 | |
|             i += 1
 | |
|         f.write(f"<p><img src=\"{local_temp_filename}\" width=512></img></p>\n")
 | |
| 
 | |
|     print(f'Image generated with private log at: {html_name}')
 | |
| 
 | |
|     return
 |