From 05993f5b2e987e7be5b4f839b3f6e9780be3fa59 Mon Sep 17 00:00:00 2001
From: Kyle <zerouex@gmail.com>
Date: Thu, 26 Jan 2023 21:38:56 -0500
Subject: [PATCH] EXIF from .txt

---
 scripts/images_history.py | 31 +++++++++++++++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

diff --git a/scripts/images_history.py b/scripts/images_history.py
index 9a2e970..63b00c9 100644
--- a/scripts/images_history.py
+++ b/scripts/images_history.py
@@ -8,7 +8,8 @@ import modules.ui
 import json
 import re
 from modules.shared import opts, cmd_opts
-from modules import shared, scripts
+from modules import shared, scripts, images
+from modules.ui_common import plaintext_to_html
 from modules import script_callbacks
 from pathlib import Path
 from typing import List, Tuple
@@ -494,7 +495,7 @@ def create_tab(tabname):
     img_file_name.change(get_ranking, inputs=img_file_name, outputs=ranking)
 
    
-    hidden.change(fn=modules.extras.run_pnginfo, inputs=[hidden], outputs=[info1, img_file_info, info2])
+    hidden.change(fn=run_pnginfo, inputs=[hidden, img_path, img_file_name], outputs=[info1, img_file_info, info2])
     
     #ranking
     ranking.change(create_ranked_file, inputs=[img_file_name, ranking])
@@ -505,7 +506,33 @@ def create_tab(tabname):
         modules.generation_parameters_copypaste.bind_buttons(send_to_buttons, img_file_name, img_file_info)
     except:
         pass
+    
+def run_pnginfo(image, image_path, image_file_name):
+    if image is None:
+        return '', '', ''
+    print(f"Image Path is {image_path}")
+    print(f"Image is {image_file_name}")
+    geninfo, items = images.read_info_from_image(image)
+    items = {**{'parameters': geninfo}, **items}
 
+    info = ''
+    for key, text in items.items():
+        info += f"""
+<div>
+<p><b>{plaintext_to_html(str(key))}</b></p>
+<p>{plaintext_to_html(str(text))}</p>
+</div>
+""".strip()+"\n"
+
+    if geninfo is None:
+        print(f"Nothing found in the image, checking for separate file")
+        filename = os.path.splitext(image_file_name)[0] + ".txt"
+        print(f"Filename is {filename}")
+        geninfo = ""
+        with open(filename) as f:
+            for line in f:
+                geninfo += line
+    return '', geninfo, info
 
 
 def on_ui_tabs():