Add GLB preview extraction for Lux3D zip outputs

agent_core/outputs.py

@@ -120,6 +120,7 @@ def write_lux3d_manifest(
     image_path: Path,
     requested_output_path: str,
     output_path: Path | None,
+    preview_path: Path | None,
     busid: str | int | None,
     status: int | None,
     result_url: str | None,
@@ -136,6 +137,7 @@ def write_lux3d_manifest(
         "image_path": workspace_relative(image_path),
         "requested_output_path": requested_output_path,
         "output_path": workspace_relative(output_path) if output_path else None,
+        "preview_path": workspace_relative(preview_path) if preview_path else None,
         "busid": busid,
         "status": status,
         "result_url": result_url,

agent_core/tools/lux3d_tool.py

@@ -8,6 +8,7 @@ import time
 import urllib.error
 import urllib.parse
 import urllib.request
+import zipfile
 from pathlib import Path
 
 from agent_core.config import (
@@ -154,6 +155,45 @@ def download_lux3d_result(result_url: str, output_path: Path) -> None:
     output_path.write_bytes(data)
 
 
+def safe_zip_target(extract_dir: Path, member_name: str) -> Path:
+    target = (extract_dir / member_name).resolve()
+    if not target.is_relative_to(extract_dir.resolve()):
+        raise RuntimeError(f"Unsafe path in Lux3D zip result: {member_name}")
+    return target
+
+
+def extract_lux3d_glb_preview(zip_path: Path, run_dir: Path) -> Path:
+    extract_dir = run_dir / "extracted"
+    try:
+        with zipfile.ZipFile(zip_path) as archive:
+            glb_members = sorted(
+                (
+                    item
+                    for item in archive.infolist()
+                    if not item.is_dir() and Path(item.filename).suffix.lower() == ".glb"
+                ),
+                key=lambda item: item.filename,
+            )
+            if not glb_members:
+                raise RuntimeError(f"Lux3D zip result did not include a .glb file: {workspace_relative(zip_path)}")
+
+            member = glb_members[0]
+            preview_path = safe_zip_target(extract_dir, member.filename)
+            preview_path.parent.mkdir(parents=True, exist_ok=True)
+            preview_path.write_bytes(archive.read(member))
+            return preview_path
+    except zipfile.BadZipFile as exc:
+        raise RuntimeError(f"Lux3D result is not a valid zip file: {workspace_relative(zip_path)}") from exc
+
+
+def resolve_lux3d_preview_path(output_path: Path, run_dir: Path) -> Path | None:
+    if zipfile.is_zipfile(output_path):
+        return extract_lux3d_glb_preview(output_path, run_dir)
+    if output_path.suffix.lower() == ".glb":
+        return output_path
+    return None
+
+
 def run_generate_3d_model(
     image_path: str,
     output_path: str | None = None,
@@ -165,6 +205,7 @@ def run_generate_3d_model(
     status = None
     result_url = None
     output = None
+    preview = None
     poll_count = 0
     run_dir = None
     run_id = None
@@ -213,6 +254,7 @@ def run_generate_3d_model(
                 output_name = requested_output_name or infer_lux3d_output_name(result_url)
                 output = run_dir / output_name
                 download_lux3d_result(result_url, output)
+                preview = resolve_lux3d_preview_path(output, run_dir)
                 manifest_path = write_lux3d_manifest(
                     run_dir=run_dir,
                     run_id=run_id,
@@ -220,6 +262,7 @@ def run_generate_3d_model(
                     image_path=image,
                     requested_output_path=requested_output_path,
                     output_path=output,
+                    preview_path=preview,
                     busid=busid,
                     status=status,
                     result_url=result_url,
@@ -232,6 +275,7 @@ def run_generate_3d_model(
                     "run_id": run_id,
                     "run_dir": workspace_relative(run_dir),
                     "output_path": workspace_relative(output),
+                    "preview_path": workspace_relative(preview) if preview else None,
                     "manifest_path": workspace_relative(manifest_path),
                     "latest_path": workspace_relative(output_root / "latest"),
                     "busid": busid,
@@ -257,6 +301,7 @@ def run_generate_3d_model(
                 image_path=image,
                 requested_output_path=requested_output_path,
                 output_path=output,
+                preview_path=preview,
                 busid=busid,
                 status=status,
                 result_url=result_url,