material api integration

app.py

@@ -49,6 +49,10 @@ RPK_DIR = os.path.join(os.path.dirname(__file__), "rpk")
 if not os.path.exists(RPK_DIR):
     os.makedirs(RPK_DIR)
 
+# Base URL of the Material Library API — used to build absolute texture URIs for PyPRT's
+# built-in HTTP resolver.  Set MATERIAL_API_URL env var in production (HF Spaces, Docker).
+MATERIAL_API_BASE = os.environ.get("MATERIAL_API_URL", "http://localhost:8000")
+
 class GenerateRequest(BaseModel):
     rpk_name: str
     geometry: Dict[str, Any]  # GeoJSON Feature or Geometry
@@ -125,6 +129,26 @@ def _ensure_report_attrs(rpk_name: str, attrs: Dict[str, Any]) -> Dict[str, Any]
     return merged
 
 
+def _build_texture_resolve_map(clean_attrs: Dict[str, Any]) -> Dict[str, str]:
+    """Build a PyPRT resolve_map for Material API texture paths.
+
+    When the user picks a texture from the Material Library UI the frontend stores
+    the API path (e.g. ``/texture/Architectural/Walls/Brick/.../Color.jpg``) as a
+    string attribute value.  The CGA rule calls ``texture(WallTexture)`` so PRT looks
+    up that exact string in the ResolveMap.  This helper maps every such value to an
+    absolute HTTP URL so PRT's built-in HTTP resolver can fetch and embed the texture.
+
+    Only string values starting with ``/texture/`` are mapped; all other attributes
+    are left to the RPK's internal ResolveMap (assets bundled inside the RPK).
+    """
+    resolve_map: Dict[str, str] = {}
+    for val in clean_attrs.values():
+        if isinstance(val, str) and val.startswith("/texture/"):
+            resolve_map[val] = f"{MATERIAL_API_BASE}{val}"
+            logger.info(f"Texture resolve_map: {val!r} → {resolve_map[val]!r}")
+    return resolve_map
+
+
 def _extract_reports(
     initial_shape: "pyprt.InitialShape",
     clean_attrs: Dict[str, Any],
@@ -291,26 +315,23 @@ async def generate_model(request: GenerateRequest):
         indices = list(range(len(coords)))
         face_counts = [len(coords)]
 
-        initial_shape = pyprt.InitialShape(flattened_coords, indices, face_counts)
+        # Sanitise attributes before InitialShape so we can build the texture resolve_map
+        clean_attributes = _clean_attributes(request.attributes)
+        logger.info(f"Generating with RPK: {rpk_path}")
+        logger.info(f"Clean attributes: {clean_attributes}")
+
+        initial_shape = pyprt.InitialShape(
+            flattened_coords, indices, face_counts,
+            resolve_map=_build_texture_resolve_map(clean_attributes),
+        )
 
         # Setup Model Generator
-        # Prepare Options
-        # We use OBJ encoder then convert to GLB.
-        # "emitReport": True is needed to get the attributes back? No, that's get_attributes.
         export_options = {
-            "outputPath": tempfile.mkdtemp(), # Create a unique temp directory
-            "outputFilename": "model",    # GLTFEncoder defaults to building model.glb if we pass proper flags
+            "outputPath": tempfile.mkdtemp(),
+            "outputFilename": "model",
             "emitReport": True,
             "emitGeometry": True
         }
-        
-        # Generate
-        logger.info(f"Generating with RPK: {rpk_path}")
-        logger.info(f"Using attributes: {request.attributes}")
-        
-        # Sanitise attributes
-        clean_attributes = _clean_attributes(request.attributes)
-        logger.info(f"Clean attributes: {clean_attributes}")
 
         is_geometry_only = request.rpk_name in GEOMETRY_ONLY_RPKS
 
@@ -475,7 +496,13 @@ async def generate_i3s(request: GenerateRequest):
         indices = list(range(len(ecef_coords_all)))
         face_counts = [len(ecef_coords_all)]
         
-        initial_shape = pyprt.InitialShape(flattened_coords, indices, face_counts)
+        # Sanitise attributes before InitialShape so we can build the texture resolve_map
+        clean_attributes = _clean_attributes(request.attributes)
+
+        initial_shape = pyprt.InitialShape(
+            flattened_coords, indices, face_counts,
+            resolve_map=_build_texture_resolve_map(clean_attributes),
+        )
 
         # 4. Encoder Options
         layer_id = str(uuid.uuid4())
@@ -495,8 +522,6 @@ async def generate_i3s(request: GenerateRequest):
             'globalOffset': anchor_ecef 
         }
         
-        # Sanitise attributes
-        clean_attributes = _clean_attributes(request.attributes)
         logger.info(f"Generating I3S to {output_dir}")
 
         # Pass 1 – extract CGA reports via PyEncoder
@@ -721,10 +746,13 @@ async def get_model_report(request: GenerateRequest):
 
         indices = list(range(len(coords)))
         face_counts = [len(coords)]
-        initial_shape = pyprt.InitialShape(flattened_coords, indices, face_counts)
 
-        # --- Sanitise attributes ---
+        # --- Sanitise attributes before InitialShape so we can build the texture resolve_map ---
         clean_attributes = _clean_attributes(request.attributes)
+        initial_shape = pyprt.InitialShape(
+            flattened_coords, indices, face_counts,
+            resolve_map=_build_texture_resolve_map(clean_attributes),
+        )
 
         # --- Run PyEncoder report pass ---
         try: