Update app_gradio_messenger.py

app_gradio_messenger.py

@@ -2,7 +2,9 @@ from __future__ import annotations
 
 import html
 import os
+from pathlib import Path
 from threading import Lock
+from urllib.parse import quote
 
 import gradio as gr
 
@@ -16,10 +18,33 @@ INITIAL_HISTORY = [{"role": "assistant", "content": INITIAL_GREETING}]
 
 SERVICES: ChatServices | None = None
 SERVICES_LOCK = Lock()
-MEGUMIN_IMAGE_URL = os.getenv(
-    "MEGUMIN_IMAGE_URL",
-    "https://huggingface.co/datasets/Junhoee/megumin-chat/resolve/main/%EB%A9%94%EA%B5%AC%EB%B0%8D%EC%82%AC%EC%A7%84.png",
-)
+APP_DIR = Path(__file__).resolve().parent
+IMAGE_EXTENSIONS = (".png", ".jpg", ".jpeg", ".webp")
+
+
+def resolve_megumin_image_path() -> Path:
+    configured_path = os.getenv("MEGUMIN_IMAGE_PATH", "").strip()
+    if configured_path:
+        return Path(configured_path).expanduser().resolve()
+
+    source_dir = APP_DIR / "source_file"
+    preferred_stem = "megumin_profile"
+
+    for extension in IMAGE_EXTENSIONS:
+        candidate = source_dir / f"{preferred_stem}{extension}"
+        if candidate.exists():
+            return candidate.resolve()
+
+    for extension in IMAGE_EXTENSIONS:
+        matches = sorted(source_dir.glob(f"*{extension}"))
+        if matches:
+            return matches[0].resolve()
+
+    return source_dir / f"{preferred_stem}.png"
+
+
+MEGUMIN_IMAGE_PATH = resolve_megumin_image_path()
+MEGUMIN_IMAGE_URL = f"/gradio_api/file={quote(MEGUMIN_IMAGE_PATH.as_posix(), safe='/:')}"
 
 MEGUMIN_SVG = """
 <svg viewBox="0 0 420 560" xmlns="http://www.w3.org/2000/svg" role="img" aria-label="Megumin inspired illustration">