Upload app.py with huggingface_hub

app.py

@@ -1,37 +1,50 @@
 import os
 import shutil
 import subprocess
+import urllib.request
 from pathlib import Path
 
 # ---- 自动拉取 Xet 指针文件（HF 用 Xet 管理大图片，构建时不会自动 materialize）----
 def _materialize_xet():
+    """检测并下载 Xet 指针文件，替换为真实图片。"""
     repo_id = "donotthink/meme"
     base_url = f"https://huggingface.co/spaces/{repo_id}/resolve/main"
     exts = (".png", ".jpg", ".jpeg", ".gif", ".webp", ".bmp")
     count = 0
+    failures = 0
     for root, _dirs, files in os.walk("meme_generator"):
         for fname in files:
             if not fname.lower().endswith(exts):
                 continue
             fpath = os.path.join(root, fname)
             try:
-                if os.path.getsize(fpath) > 1024:
+                # Xet 指针文件通常 < 1KB，内容以 "xet" 开头
+                if os.path.getsize(fpath) > 2048:
                     continue
                 with open(fpath, "r", errors="ignore") as f:
-                    head = f.read(64)
-                if not head.startswith("xet"):
+                    head = f.read(128)
+                if not head.lstrip().lower().startswith("xet"):
                     continue
                 rel = os.path.relpath(fpath, ".").replace(os.sep, "/")
                 url = f"{base_url}/{rel}"
-                subprocess.run(
-                    ["curl", "-fsSL", "-o", fpath, url],
-                    capture_output=True, timeout=30,
-                )
-                count += 1
-            except Exception:
-                pass
-    if count:
-        print(f"[xet] materialized {count} files")
+                # 用 urllib 下载真实文件（HF resolve 会自动返回实际内容）
+                for attempt in range(3):
+                    try:
+                        req = urllib.request.Request(url, headers={"User-Agent": "Mozilla/5.0"})
+                        with urllib.request.urlopen(req, timeout=60) as resp:
+                            data = resp.read()
+                        if data and not data[:8].startswith(b"version"):
+                            with open(fpath, "wb") as f:
+                                f.write(data)
+                            count += 1
+                            break
+                    except Exception:
+                        if attempt == 2:
+                            raise
+            except Exception as e:
+                failures += 1
+    print(f"[xet] materialized {count} files, {failures} failures")
+    return count
 
 _materialize_xet()
 
@@ -115,7 +128,22 @@ def _generate(meme_key, *args):
     meme = _meme_map[meme_key]
     try:
         result = meme(images=images, texts=texts)
-        return result.getvalue(), "✅ 生成成功！右键图片可保存"
+        data = result.getvalue()
+
+        # Gradio 6.x Image 组件不接受 bytes，需保存为临时文件返回路径
+        import io
+        import tempfile
+        from PIL import Image as PILImage
+
+        img = PILImage.open(io.BytesIO(data))
+        fmt = (img.format or "PNG").upper()
+        if fmt == "JPEG":
+            fmt = "JPG"
+        suffix = f".{fmt.lower()}"
+        fd, tmp_path = tempfile.mkstemp(suffix=suffix)
+        os.close(fd)
+        img.save(tmp_path, format=fmt)
+        return tmp_path, "✅ 生成成功！右键图片可保存"
     except MemeGeneratorException as e:
         return None, f"❌ {e}"
     except Exception as e: