Adding a file type to attachments

from attachments import attach, load, present
from attachments.core import Attachment, loader, presenter

import pyvista as pv, io, base64

att_path = "/home/maxime/Projects/attachments/src/attachments/data/Llama.glb"
att_path.lower().endswith((".glb", ".gltf"))

from attachments import attach
att = attach(att_path)
att.path.lower().endswith((".glb", ".gltf"))

def glb_match(att):
    return att.path.lower().endswith((".glb", ".gltf"))

obj3d = pv.read(att_path)

from attachments import attach
att = attach(att_path)
if att.path.lower().endswith((".glb", ".gltf")):
    att._obj = pv.read(att.path)
else:
    raise ValueError("Not a 3D model")

@loader(match=glb_match) #using our own matcher
def three_d(att: Attachment): #presuming attachment input
    att._obj = pv.read(att.input_source) #using input_source as this handles remote urls and other helpful stuff
    return att #returning the attachment object

# Handle MultiBlock objects first
mesh_to_render = obj3d
if isinstance(obj3d, pv.MultiBlock):
    mesh_to_render = obj3d.combine(merge_points=True)

images_array = []
print("🎥 Rendering 8 views around the 3D model...")

for i in range(8):
    # Create a fresh plotter for each view to ensure clean rotation
    p = pv.Plotter(off_screen=True)
    p.add_mesh(mesh_to_render)
    
    # Set camera to isometric view and position it around the object
    p.camera_position = 'iso'
    p.camera.zoom(1.2)
    
    # Rotate the camera around the object (45 degrees per view)
    azimuth_angle = i * 45
    p.camera.azimuth = azimuth_angle
    
    # Take screenshot
    buffer = io.BytesIO()
    p.screenshot(buffer)
    buffer.seek(0)
    images_array.append("data:image/png;base64," + base64.b64encode(buffer.read()).decode())
    
    print(f"   📸 View {i+1}/8 captured (azimuth: {azimuth_angle}°)")
    
    p.close()

print("✅ All 8 views rendered successfully!")

import base64, io, matplotlib.pyplot as plt
fig, axes = plt.subplots(2, 4, figsize=(12, 6))
axes = axes.flatten()

for i, data_url in enumerate(images_array):
    header, b64data = data_url.split(",", 1)
    img_bytes = base64.b64decode(b64data)
    img = plt.imread(io.BytesIO(img_bytes), format='png')
    
    axes[i].imshow(img)
    axes[i].axis('off')
    axes[i].set_title(f"View {i+1}")

plt.tight_layout()
plt.show()

import copy
@presenter
def images(att: Attachment, notused: "pyvista.MultiBlock") -> Attachment:
    """Presenter for PyVista MultiBlock objects."""
    obj3d = copy.deepcopy(att._obj)   
    mesh_to_render = obj3d
    if isinstance(obj3d, pv.MultiBlock):
        mesh_to_render = obj3d.combine(merge_points=True)

    images_array = []
    print("🎥 Rendering 8 views around the 3D model...")

    for i in range(8):
        # Create a fresh plotter for each view to ensure clean rotation
        p = pv.Plotter(off_screen=True)
        p.add_mesh(mesh_to_render)
        
        # Set camera to isometric view and position it around the object
        p.camera_position = 'iso'
        p.camera.zoom(1.2)
        
        # Rotate the camera around the object (45 degrees per view)
        azimuth_angle = i * 45
        p.camera.azimuth = azimuth_angle
        
        # Take screenshot
        buffer = io.BytesIO()
        p.screenshot(buffer)
        buffer.seek(0)
        images_array.append("data:image/png;base64," + base64.b64encode(buffer.read()).decode())
        
        print(f"   📸 View {i+1}/8 captured (azimuth: {azimuth_angle}°)")
        
        p.close()
    att.images = images_array
    return att  

att_path = "/home/maxime/Projects/attachments/src/attachments/data/Llama.glb"
att = attach(att_path) | load.three_d | present.images
att

from IPython.display import HTML, display
images_html = ""
for i, data_url in enumerate(att.images):
    style = f"width:150px; display:inline-block; margin:2px; border:1px solid #ddd"
    images_html += f'<img src="{data_url}" style="{style}" />'
    if (i + 1) % 4 == 0:
        images_html += "<br>"
display(HTML(images_html))

att.claude("What do you see?")

@presenter
def text(att: Attachment, notused: "pyvista.MultiBlock") -> Attachment:
    att.text = f"{att.path} bounds: {att._obj.bounds}"
    return att

# Simple pipeline.
# This pipeline is a simple pipeline that loads the object, renders it into images,
# and then renders the object into text.
pipe = load.three_d | present.images + present.text
att = pipe("/home/maxime/Projects/attachments/src/attachments/data/Llama.glb")
att.text

from attachments.pipelines import processor

@processor(
    match=glb_match,
    description="A custom GLB processor"
)
def glb_to_llm(att: Attachment) -> Attachment:
    return att | load.three_d | present.images + present.text

from attachments import Attachments
att1 = Attachments("/home/maxime/Projects/attachments/src/attachments/data/Llama.glb")
att1

from IPython.display import HTML, display
images_html = ""
for i, data_url in enumerate(att1.images):
    style = f"width:150px; display:inline-block; margin:2px; border:1px solid #ddd"
    images_html += f'<img src="{data_url}" style="{style}" />'
    if (i + 1) % 4 == 0:
        images_html += "<br>"
display(HTML(images_html))

att2 = Attachments("https://en.wikipedia.org/wiki/Llama_(language_model)[images: false]")
att2

from attachments import split
from attachments.core import splitter, AttachmentCollection

@splitter
def views(att: Attachment, notused: "pyvista.MultiBlock") -> AttachmentCollection:
    """Split 3D model attachment into individual view attachments."""
    if not att.images:
        return AttachmentCollection([att])  # Return original if no images
    
    view_attachments = []
    for i, image in enumerate(att.images):
        # Create new attachment for each view
        view_att = Attachment(f"{att.path}#view-{i+1}")
        view_att.images = [image]  # Single image per attachment
        view_att.text = f"View {i+1} of 3D model: {att.path}"
        view_att.commands = att.commands
        view_att.metadata = {
            **att.metadata,
            'chunk_type': 'view',
            'view_index': i,
            'total_views': len(att.images),
            'original_path': att.path,
            'azimuth_angle': i * 45  # Each view is 45 degrees apart
        }
        view_attachments.append(view_att)
    
    return AttachmentCollection(view_attachments)

# Load and render the 3D model
att = attach("/home/maxime/Projects/attachments/src/attachments/data/Llama.glb") | load.three_d | present.images + present.text

# Split into individual views
view_collection = att | split.views

print(f"Original attachment had {len(att.images)} images")
print(f"Split into {len(view_collection)} individual view attachments")
print(f"Each view attachment has {len(view_collection[0].images)} image")

# You can now process each view separately
for i, view in enumerate(view_collection):
    print(f"View {i+1}: {view.text}")

from attachments import modify
from attachments.core import modifier
import copy

@modifier
def decimate(att: Attachment, notused: "pyvista.MultiBlock") -> Attachment:
    """
    Decimates a 3D model using a DSL command `[decimate: reduction_factor]`.
    Example: `[decimate: 0.8]` will reduce the number of faces by 80%.
    """

    if 'decimate' not in att.commands:
        return att

    try:
        factor = float(att.commands['decimate'])
        if not (0 < factor < 1):
            raise ValueError("Decimation factor must be between 0 and 1.")

        combined_mesh = att._obj.combine(merge_points=True)
        surface_mesh = combined_mesh.extract_surface()

        original_cells = surface_mesh.n_cells
        decimated_surface = surface_mesh.decimate(factor)
        
        # Wrap the decimated PolyData back into a MultiBlock to maintain type consistency
        att._obj = pv.MultiBlock([decimated_surface])
        
        att.metadata['decimation_factor'] = factor
        att.metadata['original_faces'] = original_cells
        att.metadata['decimated_faces'] = decimated_surface.n_cells

    except Exception as e:
        att.metadata['decimate_error'] = str(e)

    return att

from attachments.core import Pipeline
# Let's define a pipeline that includes our `decimate` modifier.
decimation_pipeline = Pipeline([
    load.three_d,
    modify.decimate,
    present.images
])

# Low-poly model (90% reduction)
print("Processing decimated model (90% reduction)...")
att_low_poly = decimation_pipeline("/home/maxime/Projects/attachments/src/attachments/data/Llama.glb[decimate:0.9]")

images_html = ""
for i, data_url in enumerate(att_low_poly.images):
    style = f"width:150px; display:inline-block; margin:2px; border:1px solid #ddd"
    images_html += f'<img src="{data_url}" style="{style}" />'
    if (i + 1) % 4 == 0:
        images_html += "<br>"
display(HTML(images_html))

# Low-poly model (97% reduction)
print("Processing decimated model (97% reduction)...")
att_low_poly = decimation_pipeline("/home/maxime/Projects/attachments/src/attachments/data/Llama.glb[decimate:0.97]")

images_html = ""
for i, data_url in enumerate(att_low_poly.images):
    style = f"width:150px; display:inline-block; margin:2px; border:1px solid #ddd"
    images_html += f'<img src="{data_url}" style="{style}" />'
    if (i + 1) % 4 == 0:
        images_html += "<br>"
display(HTML(images_html))