qwqcc commited on
Commit
d4186fc
·
verified ·
1 Parent(s): 2bf991c

Upload 4 files

Browse files
Files changed (1) hide show
  1. bootstrap.py +14 -7
bootstrap.py CHANGED
@@ -153,13 +153,20 @@ def _openai_translate(text: str, lang_from: str = "auto", lang_to: str = "zh") -
153
  _utils.translate = _openai_translate
154
  print("[bootstrap] translate() monkey-patched for OpenAI provider", flush=True)
155
 
156
- # Manually load builtin memes AFTER patching translate
157
- pkg_dir = Path(importlib.import_module('meme_generator').__file__).parent
158
- memes_dir = pkg_dir / 'memes'
159
- if memes_dir.exists():
160
- for path in memes_dir.iterdir():
161
- if path.is_dir():
162
- load_meme(f"meme_generator.memes.{path.name}")
 
 
 
 
 
 
 
163
 
164
  # Optional override for /memes/render_list (fresh data each time)
165
  class MemeKeyWithProperties(BaseModel):
 
153
  _utils.translate = _openai_translate
154
  print("[bootstrap] translate() monkey-patched for OpenAI provider", flush=True)
155
 
156
+ # Load builtin memes from source tree if available (ensures assets are present);
157
+ # otherwise fall back to the installed site-packages copy.
158
+ src_memes_dir = Path("/app/meme-generator/meme_generator/memes")
159
+ if src_memes_dir.exists():
160
+ print(f"[bootstrap] Loading builtin memes from source: {src_memes_dir}", flush=True)
161
+ load_memes(str(src_memes_dir))
162
+ else:
163
+ pkg_dir = Path(importlib.import_module('meme_generator').__file__).parent
164
+ memes_dir = pkg_dir / 'memes'
165
+ print(f"[bootstrap] Loading builtin memes from package: {memes_dir}", flush=True)
166
+ if memes_dir.exists():
167
+ for path in memes_dir.iterdir():
168
+ if path.is_dir():
169
+ load_meme(f"meme_generator.memes.{path.name}")
170
 
171
  # Optional override for /memes/render_list (fresh data each time)
172
  class MemeKeyWithProperties(BaseModel):