fix missing lines in outputs

invoke-ai · Sep 11, 2022 · 0ab5a36 · 0ab5a36
1 parent 5e43372
commit 0ab5a36
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/dream.py b/scripts/dream.py
@@ -321,6 +321,7 @@ def image_writer(image, seed, upscaled=False):
         print('Outputs:')
         log_path = os.path.join(current_outdir, 'dream_log.txt')
         write_log_message(results, log_path)
+        print()
 
     print('goodbye!')
 
@@ -371,7 +372,8 @@ def write_log_message(results, log_path):
     log_lines = [f'{path}: {prompt}\n' for path, prompt in results]
     for l in log_lines:
         output_cntr += 1
-        print(output_cntr)
+        print(f'[{output_cntr}] {l}',end='')
+
 
     with open(log_path, 'a', encoding='utf-8') as file:
         file.writelines(log_lines)