Stabilize added artist tabs

README.md

@@ -4,7 +4,7 @@ emoji: 🎨
 colorFrom: red
 colorTo: yellow
 sdk: gradio
-sdk_version: 6.13.0
+sdk_version: 4.44.1
 app_file: app.py
 pinned: false
 license: other

app.py

@@ -31,6 +31,7 @@ ROOT = Path(__file__).resolve().parent
 STORE_ROOT = Path(os.environ.get("CUSTOM_STORE_DIR") or ("/data/custom_store" if Path("/data").exists() else ROOT / "custom_store"))
 INDEX_PATH = STORE_ROOT / "index.json"
 ARTISTS_DIR = STORE_ROOT / "artists"
+STORE_ERROR: Optional[str] = None
 
 
 def _env_path(name: str, default: Path) -> str:
@@ -38,13 +39,20 @@ def _env_path(name: str, default: Path) -> str:
 
 
 def _ensure_store() -> None:
+    global STORE_ERROR
     ARTISTS_DIR.mkdir(parents=True, exist_ok=True)
     if not INDEX_PATH.exists():
         INDEX_PATH.write_text(json.dumps({"schema_version": 1, "artists": []}, indent=2), encoding="utf-8")
+    STORE_ERROR = None
 
 
 def _load_index() -> dict:
-    _ensure_store()
+    try:
+        _ensure_store()
+    except Exception as exc:
+        global STORE_ERROR
+        STORE_ERROR = f"Added-artist storage is not writable: {exc}"
+        return {"schema_version": 1, "artists": []}
     try:
         payload = json.loads(INDEX_PATH.read_text(encoding="utf-8"))
     except Exception:
@@ -55,7 +63,10 @@ def _load_index() -> dict:
 
 
 def _save_index(payload: dict) -> None:
-    _ensure_store()
+    try:
+        _ensure_store()
+    except Exception as exc:
+        raise gr.Error(f"Added-artist storage is not writable: {exc}") from exc
     tmp = INDEX_PATH.with_suffix(".tmp")
     tmp.write_text(json.dumps(payload, ensure_ascii=False, indent=2), encoding="utf-8")
     tmp.replace(INDEX_PATH)
@@ -306,8 +317,11 @@ def add_artist(display_name: str, files: List[object], use_tta: bool, device_nam
         "prototype_descriptors": prototype_descriptors,
         "crop_statuses": statuses,
     }
-    _ensure_store()
-    torch.save(payload, STORE_ROOT / relative_path)
+    try:
+        _ensure_store()
+        torch.save(payload, STORE_ROOT / relative_path)
+    except Exception as exc:
+        raise gr.Error(f"Failed to save added artist. Check persistent storage: {exc}") from exc
 
     index = _load_index()
     index["artists"].append(
@@ -325,18 +339,19 @@ def add_artist(display_name: str, files: List[object], use_tta: bool, device_nam
     return f"Added {display_name}. Refresh the Manage tab to see it."
 
 
-def added_artist_choices() -> gr.Dropdown:
+def added_artist_choices():
     index = _load_index()
     choices = [
         (f"{item['display_name']} ({item['artist_id'][:8]})", item["artist_id"])
         for item in index.get("artists", [])
         if item.get("enabled", True)
     ]
-    return gr.Dropdown(choices=choices, value=choices[0][1] if choices else None)
+    return gr.update(choices=choices, value=choices[0][1] if choices else None)
 
 
 def refresh_added_artists():
-    return added_artist_choices(), "List refreshed."
+    status = STORE_ERROR or "List refreshed."
+    return added_artist_choices(), status
 
 
 def delete_artist(artist_id: str):
@@ -402,8 +417,8 @@ def build_app() -> gr.Blocks:
             refresh_button.click(refresh_added_artists, outputs=[artist_select, manage_status])
             delete_button.click(delete_artist, inputs=[artist_select], outputs=[artist_select, manage_status])
             demo.load(refresh_added_artists, outputs=[artist_select, manage_status])
-    return demo
+    return demo.queue()
 
 
 if __name__ == "__main__":
-    build_app().launch()
+    build_app().launch(show_error=True)

requirements.txt

@@ -2,7 +2,7 @@ torch
 torchvision
 pillow
 numpy
-gradio
+gradio==4.44.1
 spaces
 opencv-python
 matplotlib