Spaces:
Paused
Paused
| """Run real inference on an explicitly named deployed Space and verify its downloaded GLB/ZIP.""" | |
| import argparse | |
| import json | |
| import os | |
| from pathlib import Path | |
| import zipfile | |
| import time | |
| from gradio_client import Client, handle_file | |
| from huggingface_hub import get_token | |
| import trimesh | |
| def main(): | |
| parser = argparse.ArgumentParser(description=__doc__) | |
| parser.add_argument('--space', required=True, help='Hugging Face owner/space') | |
| parser.add_argument('--image', type=Path) | |
| parser.add_argument('--prompt', default='') | |
| parser.add_argument('--animation-prompt', default='') | |
| parser.add_argument('--animation-seconds', type=float, default=6) | |
| parser.add_argument('--output-dir', type=Path, default=Path('outputs/hf-verification')) | |
| args = parser.parse_args() | |
| if not args.image and not args.prompt.strip(): | |
| parser.error('Supply --image, --prompt, or both') | |
| args.output_dir.mkdir(parents=True, exist_ok=True) | |
| client = Client(args.space, token=os.environ.get('HF_TOKEN') or get_token(), | |
| download_files=args.output_dir.resolve()) | |
| job = client.submit(handle_file(str(args.image)) if args.image else None, | |
| args.prompt, 8, 0, args.animation_prompt, args.animation_seconds, api_name='/generate') | |
| previous = None | |
| while not job.done(): | |
| update = job.status() | |
| current = (update.code.name, tuple(p.desc for p in update.progress_data or [])) | |
| if current != previous: | |
| print(json.dumps({'state': current[0], 'progress': current[1]}), flush=True) | |
| previous = current | |
| time.sleep(2) | |
| model, reference, views, bundle, status, manifest, video = job.result() | |
| (args.output_dir / f"{manifest['id']}.json").write_text(json.dumps(manifest, indent=2)) | |
| if args.animation_prompt: | |
| import struct | |
| raw = Path(model).read_bytes() | |
| size, kind = struct.unpack_from('<II', raw, 12) | |
| if kind != 0x4e4f534a: | |
| raise RuntimeError('Animated GLB has no JSON chunk') | |
| gltf = json.loads(raw[20:20 + size]) | |
| if not gltf.get('skins') or not gltf.get('animations'): | |
| raise RuntimeError('Requested animation returned a static mesh') | |
| if not video or not manifest.get('animation', {}).get('accepted'): | |
| raise RuntimeError('Animation lacks video or failed its technical/visual review') | |
| if manifest['status'] != 'complete': | |
| raise RuntimeError('Space did not complete all stages') | |
| scene = trimesh.load(model, force='scene', process=False) | |
| if len(scene.geometry) != manifest['metrics']['components']: | |
| raise RuntimeError('Downloaded component count differs from manifest') | |
| for mesh in scene.geometry.values(): | |
| if not len(mesh.faces) or mesh.visual.kind != 'texture' or mesh.visual.material.baseColorTexture is None: | |
| raise RuntimeError('Downloaded mesh has missing geometry or texture') | |
| with zipfile.ZipFile(bundle) as archive: | |
| embedded = json.loads(archive.read('manifest.json')) | |
| if embedded['id'] != manifest['id'] or 'final.glb' not in archive.namelist(): | |
| raise RuntimeError('Download archive does not match the completed run') | |
| print(json.dumps({'verified': True, 'run': manifest['id'], 'model': model, 'bundle': bundle, | |
| 'components': len(scene.geometry)}, indent=2)) | |
| if __name__ == '__main__': | |
| main() | |