fix: fallback bucket

core/database.py

@@ -13,8 +13,36 @@ import numpy as np
 from PIL import Image
 
 # ─── Paths ───────────────────────────────────────────────────────────────────
-_hf_data = Path("/data")
-DATA_DIR   = _hf_data if (_hf_data.exists() and os.access(_hf_data, os.W_OK)) else Path("./data")
+def _is_writable(path: Path) -> bool:
+    """Testa escrita REAL (não só os.access).
+
+    O HF Storage Bucket pode reportar W_OK=True nos bits de permissao
+    mas falhar em runtime com Errno 5 (EIO) — ex.: acesso revogado apos
+    o hackathon. O que vale é a escrita de verdade.
+    """
+    try:
+        path.mkdir(parents=True, exist_ok=True)
+        probe = path / ".write_test"
+        probe.write_text("ok")          # o que realmente importa: a ESCRITA
+    except OSError:
+        return False
+    try:
+        probe.unlink()                  # limpeza best-effort
+    except OSError:
+        pass
+    return True
+
+
+def _pick_data_dir() -> Path:
+    """Escolhe o primeiro destino realmente gravável."""
+    for candidate in (Path("/data"), Path("./data"), Path("/tmp/viralata-data")):
+        if _is_writable(candidate):
+            print(f"[database] DATA_DIR = {candidate}")
+            return candidate
+    return Path("/tmp/viralata-data")
+
+
+DATA_DIR   = _pick_data_dir()
 DB_PATH    = DATA_DIR / "viralata.db"
 PHOTOS_DIR = DATA_DIR / "photos"
 SCHEMA     = Path(__file__).parent.parent / "db" / "schema.sql"
@@ -74,7 +102,6 @@ class Database:
                 conn.execute("ALTER TABLE animals ADD COLUMN name TEXT")
 
     # ─── Photos ──────────────────────────────────────────────────────────────
-
     def save_photo(self, image: Image.Image, animal_id: Optional[int] = None) -> str:
         """Salva foto organizada por animal_id. Ex: photos/animal_42/abc123.jpg"""
         from datetime import datetime
@@ -97,7 +124,6 @@ class Database:
         return str(full) if full.exists() else None
 
     # ─── Animals ─────────────────────────────────────────────────────────────
-
     def create_animal(self, description: dict, embedding: list,
                       name: Optional[str] = None) -> int:
         emb_blob = np.array(embedding, dtype=np.float32).tobytes()
@@ -149,7 +175,6 @@ class Database:
         return result
 
     # ─── Sightings ───────────────────────────────────────────────────────────
-
     def add_sighting(
         self,
         animal_id: int,
@@ -206,11 +231,9 @@ class Database:
         return {"animal": animal, "sightings": sightings, "help_events": help_events}
 
     # ─── Map data ────────────────────────────────────────────────────────────
-
     def get_map_data(self, species: str = "all", timeframe: str = "all") -> list:
         filters = []
         params = []
-
         if species in ("dog", "cat"):
             filters.append("a.species = ?")
             params.append(species)
@@ -218,9 +241,7 @@ class Database:
             filters.append("date(a.last_seen) = date('now')")
         elif timeframe == "week":
             filters.append("a.last_seen >= datetime('now', '-7 days')")
-
         where = ("WHERE " + " AND ".join(filters)) if filters else ""
-
         sql = (
             "SELECT"
             "  a.id,"
@@ -243,10 +264,8 @@ class Database:
             " " + where +
             " ORDER BY a.last_seen DESC"
         )
-
         with self._conn() as conn:
             rows = conn.execute(sql, params).fetchall()
-
         result = []
         for row in rows:
             d = dict(row)
@@ -267,7 +286,6 @@ class Database:
         return result
 
     # ─── Animals list ────────────────────────────────────────────────────────
-
     def get_recent_animals(self, limit: int = 30) -> list:
         sql = (
             "SELECT"
@@ -296,4 +314,4 @@ class Database:
 
     def total_animals(self) -> int:
         with self._conn() as conn:
-            return conn.execute("SELECT COUNT(*) FROM animals").fetchone()[0]
+            return conn.execute("SELECT COUNT(*) FROM animals").fetchone()[0]