Update app.py

app.py

@@ -3,11 +3,9 @@ from pathlib import Path
 import os
 import shutil
 from huggingface_hub import HfApi, snapshot_download
-from PIL import Image, ExifTags
-import base64
-import tempfile
 
-HF_TOKEN = os.environ.get("HF_TOKEN")
+HF_TOKEN = os.environ["HF_TOKEN"]
+
 if not HF_TOKEN:
     raise ValueError("HF_TOKEN not set")
 
@@ -16,14 +14,10 @@ username = api.whoami(token=HF_TOKEN)["name"]
 repo = "sd_out"
 user_repo = f"{username}/{repo}"
 
-
 def refresh_images():
-    # hapus cache lama (jika ada) lalu download snapshot dataset dan salin ke folder local "images"
     try:
-        old = os.environ.get("IMAGE_DIR", "")
-        if old:
-            shutil.rmtree(old)
-    except Exception:
+        shutil.rmtree(os.environ["IMAGE_DIR"])
+    except:
         pass
 
     image_dir = Path(
@@ -31,7 +25,7 @@ def refresh_images():
     )
     os.environ["IMAGE_DIR"] = str(image_dir)
 
-    image_files = sorted(list(image_dir.rglob("*.[pjw]*[npjg]*[ge]*")))
+    image_files = list(Path(image_dir).rglob("*.[pjw]*[npjg]*[ge]*"))
 
     local_dir = Path("images")
     local_dir.mkdir(exist_ok=True)
@@ -39,205 +33,22 @@ def refresh_images():
     copied_files = []
     for img in image_files:
         dst = local_dir / img.name
-        try:
-            shutil.copy2(img, dst)
-            copied_files.append(str(dst))
-        except Exception:
-            # kalau gagal salin, lewati
-            continue
+        shutil.copy(img, dst)
+        copied_files.append(str(dst))
 
     return copied_files
 
 
-def _human_size(n):
-    n = float(n)
-    for unit in ["B", "KB", "MB", "GB", "TB"]:
-        if n < 1024.0:
-            return f"{n:.2f} {unit}"
-        n /= 1024.0
-    return f"{n:.2f} PB"
-
-
-def _extract_prompt_from_image_or_exif(img_path_or_b64):
-    """
-    Kembalikan tuple (prompt_text_or_None, source_description_or_None).
-    Cek urutan: PNG tEXt/info keys (parameters, prompt, etc.) -> EXIF common tags.
-    """
-    tmp_file = None
-    try:
-        # Bila base64 data: simpan sementara
-        if isinstance(img_path_or_b64, str) and img_path_or_b64.startswith("data:"):
-            header, b64 = img_path_or_b64.split(",", 1)
-            data = base64.b64decode(b64)
-            tmpf = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
-            tmpf.write(data)
-            tmpf.flush()
-            tmpf.close()
-            img_path = tmpf.name
-            tmp_file = img_path
-        else:
-            img_path = str(img_path_or_b64)
-
-        with Image.open(img_path) as im:
-            info = im.info or {}
-            # Periksa beberapa kunci umum di PNG tEXt / info
-            for k in ("parameters", "prompt", "caption", "description", "Comment", "comment"):
-                if k in info and info[k]:
-                    return info[k], f"PNG info ('{k}')"
-            # kadang Pillow menempatkan text chunks di 'text' dict
-            if "text" in info and isinstance(info["text"], dict) and info["text"]:
-                lines = []
-                for k, v in info["text"].items():
-                    lines.append(f"{k}: {v}")
-                if lines:
-                    return "\n".join(lines), "PNG text chunks"
-
-            # EXIF (umumnya untuk JPEG)
-            try:
-                exif_raw = im.getexif()
-                if exif_raw:
-                    exif = {}
-                    for tag_id, val in exif_raw.items():
-                        tag = ExifTags.TAGS.get(tag_id, tag_id)
-                        # decode bytes jika perlu
-                        if isinstance(val, bytes):
-                            try:
-                                val = val.decode("utf-8", errors="ignore")
-                            except Exception:
-                                val = str(val)
-                        exif[tag] = val
-                    for tagname in ("ImageDescription", "UserComment", "XPComment"):
-                        if tagname in exif and exif[tagname]:
-                            return exif[tagname], f"EXIF '{tagname}'"
-            except Exception:
-                pass
-
-        return None, None
-    finally:
-        if tmp_file:
-            try:
-                os.remove(tmp_file)
-            except Exception:
-                pass
-
-
-def get_image_info(selected):
-    """
-    Fungsi yang dipanggil saat user memilih item di Gallery.
-    Bisa menerima nilai dalam beberapa bentuk (list, dict, path string, base64).
-    Mengembalikan string yang akan ditampilkan di Textbox.
-    """
-    if not selected:
-        return "No image selected."
-
-    # Gallery kadang mengirim list berisi single path, handle itu:
-    if isinstance(selected, list) and len(selected) > 0:
-        selected = selected[0]
-
-    # Gallery kadang mengirim dict dengan key 'name' atau 'image'
-    if isinstance(selected, dict):
-        for k in ("name", "image", "src", "path", "filename"):
-            if k in selected:
-                selected = selected[k]
-                break
-
-    # Jika selected adalah data URI (base64), kita akan simpan sementara untuk baca metadata
-    is_data_uri = isinstance(selected, str) and selected.startswith("data:")
+with gr.Blocks(
+    analytics_enabled=False, title="Image Gallery", theme="NoCrypt/miku"
+) as demo:
+    gr.HTML("""<center><h1>Image Gallery</h1></center>""")
+    submit = gr.Button("Refresh", variant="primary")
 
-    tmp_path = None
-    cleanup_tmp = False
-    try:
-        if is_data_uri:
-            header, b64 = selected.split(",", 1)
-            data = base64.b64decode(b64)
-            tmpf = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
-            tmpf.write(data)
-            tmpf.flush()
-            tmpf.close()
-            tmp_path = tmpf.name
-            path = tmp_path
-            cleanup_tmp = True
-        else:
-            path = str(selected)
-
-        lines = []
-        p = Path(path)
-        lines.append(f"File: {p.name}")
-
-        # ukuran file (jika ada path lokal)
-        try:
-            size_bytes = os.path.getsize(path)
-            lines.append(f"Size: {_human_size(size_bytes)} ({size_bytes} bytes)")
-        except Exception:
-            # jika not a local file (mis. base64), skip
-            pass
-
-        # buka image untuk dapatkan dimensi/format/mode/dpi
-        try:
-            with Image.open(path) as im:
-                lines.append(f"Format: {im.format}")
-                lines.append(f"Mode: {im.mode}")
-                lines.append(f"Dimensions: {im.width} x {im.height} px")
-                dpi = im.info.get("dpi")
-                if dpi:
-                    lines.append(f"DPI: {dpi}")
-        except Exception as e:
-            lines.append(f"(Could not open image: {e})")
-
-        # cari prompt dari metadata/image info/exif
-        prompt, src = _extract_prompt_from_image_or_exif(path if not is_data_uri else selected)
-        if prompt:
-            lines.append("")
-            lines.append("--- Prompt ---")
-            lines.append(f"Source: {src}")
-            lines.append(prompt)
-        else:
-            # cek sidecar .png.txt atau .txt (prioritas)
-            cand1 = p.with_suffix(p.suffix + ".txt")  # e.g. image.png.txt
-            cand2 = p.with_suffix(".txt")             # e.g. image.txt
-            found = False
-            for c in (cand1, cand2):
-                try:
-                    if c.exists():
-                        txt = c.read_text(encoding="utf-8")
-                        lines.append("")
-                        lines.append("--- Prompt (from sidecar) ---")
-                        lines.append(txt)
-                        found = True
-                        break
-                except Exception:
-                    pass
-            if not found:
-                lines.append("")
-                lines.append("Prompt: (not found)")
-
-        return "\n".join(lines)
-    finally:
-        if cleanup_tmp and tmp_path:
-            try:
-                os.remove(tmp_path)
-            except Exception:
-                pass
-
-
-with gr.Blocks(analytics_enabled=False, title="Image Gallery", theme="NoCrypt/miku") as demo:
-    gr.HTML("<center><h1>Image Gallery</h1></center>")
-
-    with gr.Row():
-        submit = gr.Button("Refresh", variant="primary")
-
-    gallery = gr.Gallery(value=[], columns=4, show_label=False, height=800, object_fit="contain")
-
-    info_box = gr.Textbox(
-        label="Image Info / Prompt",
-        lines=12,
-        interactive=False,
-        placeholder="Select an image to see its info...",
+    gallery = gr.Gallery(
+        value=[], columns=4, show_label=False, height=800, object_fit="contain"
     )
 
     submit.click(refresh_images, outputs=[gallery])
 
-    # Saat user memilih item di gallery -> panggil get_image_info
-    gallery.select(fn=get_image_info, inputs=[gallery], outputs=[info_box])
-
-demo.launch(debug=True)
+demo.launch(debug=True)