adds lotties

core/seed.py

@@ -1,6 +1,6 @@
 """
 seed.py — Dados de demonstração para o mapa de Brasília, DF.
-Chamado no startup do app.py somente se o banco estiver vazio.
+Chamado no startup do app.py. Garante que os 6 animais fixos estão sempre no banco.
 """
 import json
 import logging
@@ -10,9 +10,6 @@ import numpy as np
 
 log = logging.getLogger(__name__)
 
-# Animais de demonstração espalhados por Brasília, DF.
-# photo_path é relativo ao DATA_DIR (ex: 'photos/seed/pretao.jpg'),
-# servido em /photos/seed/pretao.jpg pelo static mount.
 _SEED_ANIMALS = [
     {
         "species": "dog",
@@ -46,7 +43,7 @@ _SEED_ANIMALS = [
             "secondary_colors": ["caramelo"],
             "distinctive_marks": ["manchas caramelo nas costas", "orelhas caramelo"],
             "condition": "saudável",
-            "description_text": "medium white dog with caramel patches, resting on wooden surface near closed shop",
+            "description_text": "medium white dog with caramel patches, resting near closed shop",
             "_ai_success": True,
         },
         "sightings": [
@@ -137,47 +134,49 @@ _SEED_ANIMALS = [
 ]
 
 
-def _random_embedding() -> bytes:
-    """Gera um embedding aleatório normalizado (384-dim, float32)."""
+def _random_embedding():
     v = np.random.randn(384).astype(np.float32)
     v /= np.linalg.norm(v)
     return v.tobytes()
 
 
-def seed_if_empty(db) -> None:
-    """Garante que os animais fixos de demonstração estão sempre no banco.
-    Insere apenas os que ainda não existem (comparando pelo nome).
-    Dados reais dos usuários nunca são removidos.
+def seed_if_empty(db):
+    """Garante que os animais fixos estão sempre no banco.
+    Insere apenas os que ainda nao existem (por nome).
+    Dados reais dos usuarios nunca sao removidos.
     """
     with db._conn() as conn:
         existing_names = {
-            row[0] for row in conn.execute("SELECT name FROM animals WHERE name IS NOT NULL")
+            row[0] for row in conn.execute(
+                "SELECT name FROM animals WHERE name IS NOT NULL"
+            )
         }
 
     to_insert = [a for a in _SEED_ANIMALS if a["name"] not in existing_names]
     if not to_insert:
-        log.info("Animais fixos já presentes — seed ignorado.")
+        log.info("Animais fixos ja presentes — seed ignorado.")
         return
 
-    log.info("Inserindo %d animais fixos de demonstração...", len(to_insert))
+    log.info("Inserindo %d animais fixos de demonstracao...", len(to_insert))
     now = datetime.utcnow()
 
     with db._conn() as conn:
         for animal_data in to_insert:
             sightings = animal_data["sightings"]
             desc = animal_data["description"]
-            photo_path = f"photos/seed/{animal_data['photo_filename']}"
+            photo_path = "photos/seed/" + animal_data["photo_filename"]
 
             most_recent_days = min(s["days_ago"] for s in sightings)
             last_seen = now - timedelta(days=most_recent_days)
             first_seen = now - timedelta(days=max(s["days_ago"] for s in sightings))
 
             cur = conn.execute(
-                "INSERT INTO animals (species, name, description, embedding, first_seen, last_seen, sighting_count)"
+                "INSERT INTO animals"
+                " (species, name, description, embedding, first_seen, last_seen, sighting_count)"
                 " VALUES (?, ?, ?, ?, ?, ?, ?)",
                 (
                     animal_data["species"],
-                    animal_data.get("name"),
+                    animal_data["name"],
                     json.dumps(desc, ensure_ascii=False),
                     _random_embedding(),
                     first_seen.strftime("%Y-%m-%d %H:%M:%S"),
@@ -189,10 +188,10 @@ def seed_if_empty(db) -> None:
 
             for i, s in enumerate(sightings):
                 created_at = now - timedelta(days=s["days_ago"])
-                # foto só no avistamento mais recente
                 s_photo = photo_path if i == len(sightings) - 1 else None
                 conn.execute(
-                    "INSERT INTO sightings (animal_id, photo_path, latitude, longitude, notes, created_at)"
+                    "INSERT INTO sightings"
+                    " (animal_id, photo_path, latitude, longitude, notes, created_at)"
                     " VALUES (?, ?, ?, ?, ?, ?)",
                     (
                         animal_id,
@@ -204,6 +203,6 @@ def seed_if_empty(db) -> None:
                     ),
                 )
 
-    log.info("Seed concluído: %d animais inseridos (%d avistamentos).",
+    log.info("Seed concluido: %d animais inseridos (%d avistamentos).",
              len(to_insert),
              sum(len(a["sightings"]) for a in to_insert))