Fix Gallery metadata extraction: robust path handling, fallback, include filename

ui/events/run_handlers.py

@@ -115,22 +115,53 @@ def create_run_event(prefix: str, task_type: str, ui_components: dict):
             outputs=[res_gal]
         )
         # When a user selects an image from the global Gallery tab, load its generation info
-        def _show_meta(selected_path, *_):
-            # `selected_path` may be a string (filepath) or a list/tuple of filepaths.
+        def _show_meta(selected, *_):
+            # `selected` may be a string (filepath), a list, or a tuple containing the path.
+            # Gradio can sometimes pass a tuple like (path, None) or a list of such tuples.
+            # We extract the first string path we find.
+            def extract_path(val):
+                if isinstance(val, str):
+                    return val
+                if isinstance(val, (list, tuple)):
+                    for item in val:
+                        result = extract_path(item)
+                        if result:
+                            return result
+                return None
+
+            path = extract_path(selected)
+            if not path:
+                print(f"[DEBUG] No valid image path extracted from {selected}")
+                return "(ไม่มีข้อมูลเมตาดาต้า)"
+
+            # Try to read metadata from the extracted path. If this fails (e.g., path points to a temporary Gradio copy), fall back to the most‑recent file in the output directory.
+            meta = ""
             try:
-                # Recursively flatten until we get a string path
-                path = selected_path
-                while isinstance(path, (list, tuple)) and len(path) > 0:
-                    path = path[0]
-                print(f"[DEBUG] Extracted image path: {path}")
                 from PIL import Image
                 with Image.open(path) as im:
                     meta = im.info.get("parameters", "")
+                print(f"[DEBUG] Metadata extracted from {path}: {meta}")
             except Exception as e:
-                print(f"[DEBUG] Failed to read metadata: {e}")
-                meta = ""
-            # If no metadata present, show a friendly placeholder.
-            return meta if meta else "(ไม่มีข้อมูลเมตาดาต้า)"
+                print(f"[DEBUG] Failed to read metadata from {path}: {e}")
+                # Fallback: find latest file in the configured OUTPUT_DIR
+                try:
+                    from pathlib import Path
+                    out_dir = Path(os.path.abspath(OUTPUT_DIR))
+                    files = sorted(out_dir.glob("*.png"), key=lambda p: p.stat().st_mtime, reverse=True)
+                    if files:
+                        fallback_path = files[0]
+                        with Image.open(fallback_path) as im2:
+                            meta = im2.info.get("parameters", "")
+                        print(f"[DEBUG] Fallback metadata from {fallback_path}: {meta}")
+                except Exception as e2:
+                    print(f"[DEBUG] Fallback metadata extraction failed: {e2}")
+
+            # If metadata exists, we can also include the filename for clarity
+            if meta:
+                filename = os.path.basename(path)
+                return f"{filename}\n{meta}"
+            else:
+                return "(ไม่มีข้อมูลเมตาดาต้า)"
 
         # Connect the select event of the global gallery to update its metadata textbox
         gallery_tab = ui_components.get('gallery')