Upload 2 files

api_client.py

@@ -167,6 +167,22 @@ class APIClient:
         except requests.exceptions.RequestException as e:
             return {"error": str(e)}
 
+    def load_casting(self, faces_dir: str, voices_dir: str, db_dir: str, drop_collections: bool = False) -> dict:
+        """Carga índices de caras y voces al motor de búsqueda Chroma del engine."""
+        url = f"{self.base_url}/load_casting"
+        data = {
+            "faces_dir": faces_dir,
+            "voices_dir": voices_dir,
+            "db_dir": db_dir,
+            "drop_collections": str(1 if drop_collections else 0),
+        }
+        try:
+            r = self.session.post(url, data=data, timeout=self.timeout * 5)
+            r.raise_for_status()
+            return r.json()
+        except requests.exceptions.RequestException as e:
+            return {"error": str(e)}
+
     def rebuild_video_with_ad(self, video_path: str, srt_path: str) -> dict:
         """
         Llama al space 'tts' para reconstruir un vídeo con audiodescripció a partir de un SRT.

app.py

@@ -448,6 +448,28 @@ if page == "Processar vídeo nou":
                 res_fc = api.finalize_casting(payload)
                 if isinstance(res_fc, dict) and res_fc.get("ok"):
                     st.success(f"Càsting consolidat. Identities: {len(res_fc.get('face_identities', []))} cares, {len(res_fc.get('voice_identities', []))} veus.")
+                    # Mostrar llistes
+                    f_id = res_fc.get('face_identities', []) or []
+                    v_id = res_fc.get('voice_identities', []) or []
+                    c3, c4 = st.columns(2)
+                    with c3:
+                        st.markdown("**Identitats de cara**")
+                        for n in f_id:
+                            st.write(f"- {n}")
+                    with c4:
+                        st.markdown("**Identitats de veu**")
+                        for n in v_id:
+                            st.write(f"- {n}")
+                    # Botó per carregar els índexs al buscador
+                    faces_dir = res_fc.get('faces_dir')
+                    voices_dir = res_fc.get('voices_dir')
+                    db_dir = res_fc.get('db_dir')
+                    if st.button("Carregar índexs al cercador (Chroma)"):
+                        load_res = api.load_casting(faces_dir=faces_dir, voices_dir=voices_dir, db_dir=db_dir, drop_collections=True)
+                        if isinstance(load_res, dict) and load_res.get('ok'):
+                            st.success(f"Índexs carregats: {load_res.get('faces', 0)} cares, {load_res.get('voices', 0)} veus.")
+                        else:
+                            st.error(f"Error carregant índexs: {load_res}")
                 else:
                     st.error(f"No s'ha pogut consolidar el càsting: {res_fc}")
         with colc2: