Update core/image_generator.py

core/image_generator.py

@@ -6,6 +6,15 @@ from huggingface_hub import hf_hub_download
 from pathlib import Path
 from typing import List
 
+
+# ---------------- HF CACHE FIX ----------------
+# Force Hugging Face to use a writable cache folder
+HF_CACHE_DIR = Path("/tmp/hf_cache")
+HF_CACHE_DIR.mkdir(parents=True, exist_ok=True)
+os.environ["HF_HOME"] = str(HF_CACHE_DIR)
+os.environ["TRANSFORMERS_CACHE"] = str(HF_CACHE_DIR)
+os.environ["XDG_CACHE_HOME"] = str(HF_CACHE_DIR)
+
 # ---------------- MODEL CONFIG ----------------
 MODEL_REPO = "SG161222/RealVisXL_V4.0"
 MODEL_FILENAME = "realvisxlV40_v40LightningBakedvae.safetensors"
@@ -51,6 +60,13 @@ def load_pipeline() -> StableDiffusionXLPipeline:
 # ---------------- GLOBAL PIPELINE CACHE ----------------
 pipe: StableDiffusionXLPipeline | None = None
 
+def pil_to_base64(img: Image.Image) -> str:
+    buffered = BytesIO()
+    img.save(buffered, format="PNG")
+    img_bytes = buffered.getvalue()
+    img_b64 = base64.b64encode(img_bytes).decode("utf-8")
+    return f"data:image/png;base64,{img_b64}"
+
 # ---------------- IMAGE GENERATION ----------------
 def generate_images(prompt: str, seed: int = None, num_images: int = 3) -> List:
     """