Update modules/langchain_assistant/cvision_forge_v9.py

modules/langchain_assistant/cvision_forge_v9.py

@@ -93,27 +93,46 @@ def ejecutar_forja_3d(imagen_generada_flux: str):
                         print(f"[TRELLIS] Ciclo de pooling {ciclo_progreso}/45...")
                         ciclo_progreso += 1
                         
-                    elif msg_tipo == "process_completed":
-                        print("🎯 [TRELLIS Engine] ¡Procesamiento completado con éxito detectado en el flujo!")
+                    elif msg_tipo == "process_completed" or "output" in json_evento:
+                        print("🎯 [TRELLIS Engine] ¡Procesamiento completado detectado en el flujo!")
                         
                         output = json_evento.get("output", {})
-                        array_salida = output.get("data", [])
                         
-                        if array_salida and isinstance(array_salida, list):
-                            primer_nodo = array_salida[0]
-                            nombre_archivo_glb = None
+                        # --- EXTRACTOR UNIVERSAL ULTRA-FLEXIBLE ---
+                        # Convertimos todo el bloque 'output' a texto plano para rastrear el archivo sin importar las llaves
+                        output_str = json.dumps(output)
+                        nombre_archivo_glb = None
+                        
+                        # Si el output contiene un .glb, lo cazamos quirúrgicamente
+                        if ".glb" in output_str:
+                            # Opción 1: Validar si viene en la lista clásica de 'data'
+                            array_salida = output.get("data", [])
+                            if array_salida and isinstance(array_salida, list):
+                                primer_nodo = array_salida[0]
+                                if isinstance(primer_nodo, dict):
+                                    # Intentamos extraer de cualquier llave común de Gradio (name, url, value, path)
+                                    nombre_archivo_glb = (primer_nodo.get("name") or 
+                                                          primer_nodo.get("url") or 
+                                                          primer_nodo.get("value") or 
+                                                          primer_nodo.get("path"))
+                                elif isinstance(primer_nodo, str):
+                                    nombre_archivo_glb = primer_nodo
                             
-                            if isinstance(primer_nodo, dict):
-                                nombre_archivo_glb = primer_nodo.get("name") or primer_nodo.get("url")
-                            elif isinstance(primer_nodo, str):
-                                nombre_archivo_glb = primer_nodo
-                                
-                            if nombre_archivo_glb:
-                                if nombre_archivo_glb.startswith("http"):
-                                    return nombre_archivo_glb
-                                return f"https://trellis-community-trellis.hf.space/gradio_api/file={nombre_archivo_glb}"
+                            # Opción 2: Si el parseo estructurado falló pero el String está ahí, lo extraemos a la fuerza
+                            if not nombre_archivo_glb:
+                                import re
+                                # Busca cualquier patrón de texto que termine en .glb dentro del JSON
+                                coincidencias = re.findall(r'([^"\s]+\.glb)', output_str)
+                                if coincidencias:
+                                    nombre_archivo_glb = coincidencias[0]
+                        
+                        if nombre_archivo_glb:
+                            print(f"[TRELLIS Engine] Malla .glb localizada con éxito: {nombre_archivo_glb}")
+                            if nombre_archivo_glb.startswith("http"):
+                                return nombre_archivo_glb
+                            return f"https://trellis-community-trellis.hf.space/gradio_api/file={nombre_archivo_glb}"
                                 
-                        raise Exception("El flujo de salida no contenía las claves de archivo esperadas.")
+                        raise Exception("El flujo de salida concluyó pero no se encontró ningún archivo .glb en el payload.")
                         
                     elif msg_tipo == "process_error" or "event: error" in linea_decodificada:
                         raise Exception("El clúster remoto reportó una falla interna en las matrices de difusión.")