Add 3D garment visualizer (Plotly)

garment_3d.py

@@ -0,0 +1,212 @@
+"""
+3D Garment Visualizer
+Generates a 3D garment mesh on a parametric body using Plotly.
+"""
+import numpy as np
+import plotly.graph_objects as go
+from typing import Dict, Optional
+
+def create_body_mesh(height=170, chest=88, waist=74, hip=96):
+    theta = np.linspace(0, 2*np.pi, 30)
+    z_levels = np.linspace(0, height, 40)
+    vertices = []
+    faces = []
+    for i, z in enumerate(z_levels):
+        t = z / height
+        if t < 0.12:
+            r = 8 + t * 40
+        elif t < 0.20:
+            r = chest/2 * 0.3 + (t-0.12)/0.08 * (chest/2*0.7-chest/2*0.3)
+        elif t < 0.45:
+            frac = (t-0.20)/0.25
+            r = chest/2 * (1 - 0.08*frac)
+        elif t < 0.55:
+            r = waist/2
+        elif t < 0.72:
+            frac = (t-0.55)/0.17
+            r = waist/2 + (hip/2 - waist/2) * frac
+        else:
+            frac = (t-0.72)/0.28
+            r = hip/2 * (1 - 0.5*frac)
+        for j, th in enumerate(theta):
+            x = r * np.cos(th) * 0.85
+            y = r * np.sin(th) * 0.5
+            vertices.append([x, y, z])
+    vertices = np.array(vertices)
+    n_theta = len(theta)
+    for i in range(len(z_levels)-1):
+        for j in range(n_theta-1):
+            v0 = i*n_theta + j
+            v1 = i*n_theta + j+1
+            v2 = (i+1)*n_theta + j
+            v3 = (i+1)*n_theta + j+1
+            faces.append([v0, v2, v1])
+            faces.append([v1, v2, v3])
+    return vertices, faces
+
+def _garment_mesh_shirt(params, body_verts, n_theta, z_levels):
+    parts = []
+    bust = params.get('bust', 92)/2
+    waist = params.get('waist', 74)/2
+    shoulder = params.get('shoulder_width', 42)/2
+    body_len = params.get('bodice_length', 72)
+    sleeve_len = params.get('sleeve_length', 60)
+    sleeve_lo = params.get('bicep', 30)/2 + 2
+    sleeve_hi = params.get('wrist', 18)/2 + 1
+    # Front panel
+    front_z = np.linspace(0, body_len, 20)
+    front_x = []
+    front_y = []
+    for z in front_z:
+        t = z / body_len
+        if t < 0.3:
+            x = shoulder * (1 - 0.1*t)
+        elif t < 0.55:
+            x = shoulder - (shoulder - bust*0.85) * ((t-0.3)/0.25)
+        else:
+            x = bust*0.85 - (bust*0.85 - waist*0.9) * ((t-0.55)/0.45)
+        front_x.append(x)
+        front_y.append(0)
+    pts = list(zip(front_x, front_y, front_z))
+    pts.append((0, 0, front_z[-1]))
+    pts.insert(0, (0, 0, 0))
+    n = len(pts)
+    xs = [p[0] for p in pts]
+    ys = [p[1] for p in pts]
+    zs = [p[2] for p in pts]
+    parts.append(dict(x=xs, y=ys, z=zs, type='scatter3d', mode='lines',
+        line=dict(color='#4A90D9', width=4), name='Front'))
+    # Back panel (mirror x)
+    back_x = [-x for x in xs]
+    parts.append(dict(x=back_x, y=ys, z=zs, type='scatter3d', mode='lines',
+        line=dict(color='#6BB3E0', width=4), name='Back'))
+    # Sleeves
+    shoulder_pt = [shoulder*0.9, 0, body_len*0.92]
+    sl_z = np.linspace(0, sleeve_len, 10)
+    for side in [1, -1]:
+        sx = [side * (shoulder*0.9 + (sleeve_lo - shoulder*0.3)*0.3 + sl*0.08*side) for sl in sl_z]
+        sy = [sl*0.6 for sl in sl_z]
+        sz = [body_len*0.92 - sl*0.15 for sl in sl_z]
+        color = '#7BC8F6' if side > 0 else '#5BA3D9'
+        parts.append(dict(x=sx, y=sy, z=sz, type='scatter3d', mode='lines',
+            line=dict(color=color, width=4), name='Sleeve R' if side > 0 else 'Sleeve L'))
+    return parts
+
+def _garment_mesh_pants(params, body_verts, n_theta, z_levels):
+    parts = []
+    waist_r = params.get('waist', 74)/2
+    hip_r = params.get('hip', 96)/2
+    pant_len = params.get('pant_length', 100)
+    crotch = params.get('crotch_depth', 27)
+    ankle = params.get('ankle', 24)/2 + 1
+    body_h = 170
+    for side in [1, -1]:
+        leg_x = [side * waist_r*0.5, side * hip_r*0.7, side * hip_r*0.9, side * ankle*1.5, side * ankle]
+        leg_y = [0, 0, 0, 0, 0]
+        leg_z = [body_h, body_h*0.6, body_h*0.55, body_h*0.55 - (pant_len-crotch)*0.5, body_h*0.55 - pant_len*0.7]
+        parts.append(dict(x=leg_x, y=leg_y, z=leg_z, type='scatter3d', mode='lines',
+            line=dict(color='#2C5F8A', width=5), name='Leg R' if side > 0 else 'Leg L'))
+    return parts
+
+def _garment_mesh_skirt(params, body_verts, n_theta, z_levels):
+    parts = []
+    waist_r = params.get('waist', 74)/2
+    hip_r = params.get('hip', 96)/2
+    skirt_len = params.get('skirt_length', 55)
+    flare = params.get('flare', 5)
+    body_h = 170
+    waist_z = body_h * 0.58
+    hem_z = waist_z - skirt_len
+    hem_r = hip_r + flare
+    theta = np.linspace(0, 2*np.pi, 40)
+    for t_val, r, z_val, c, nm in [(0, waist_r, waist_z, '#D4A5C9', 'Waist'), (1, hip_r, waist_z - 20, '#C48BB8', 'Hip'), (2, hem_r, hem_z, '#E8C5E0', 'Hem')]:
+        xs = [r * np.cos(th)*0.85 for th in theta]
+        ys = [r * np.sin(th)*0.5 for th in theta]
+        zs = [z_val]*len(theta)
+        parts.append(dict(x=xs, y=ys, z=zs, type='scatter3d', mode='lines',
+            line=dict(color=c, width=3), name=nm))
+    return parts
+
+def _garment_mesh_dress(params, body_verts, n_theta, z_levels):
+    parts = _garment_mesh_shirt(params, body_verts, n_theta, z_levels)
+    parts += _garment_mesh_skirt(params, body_verts, n_theta, z_levels)
+    return parts
+
+def _garment_mesh_hoodie(params, body_verts, n_theta, z_levels):
+    parts = _garment_mesh_shirt(params, body_verts, n_theta, z_levels)
+    head_circ = params.get('head_circumference', 57)/2
+    body_len = params.get('bodice_length', 65)
+    body_h = 170
+    hood_z = [body_h*0.9, body_h*0.95, body_h*1.02, body_h*1.05, body_h*1.02, body_h*0.95]
+    hood_x = [head_circ*0.8, head_circ, head_circ*0.7, 0, -head_circ*0.7, -head_circ]
+    hood_y = [0, head_circ*0.3, head_circ*0.5, head_circ*0.6, head_circ*0.5, head_circ*0.3]
+    parts.append(dict(x=hood_x, y=hood_y, z=hood_z, type='scatter3d', mode='lines',
+        line=dict(color='#3A7BD5', width=5), name='Hood'))
+    return parts
+
+def _garment_mesh_jacket(params, body_verts, n_theta, z_levels):
+    return _garment_mesh_shirt(params, body_verts, n_theta, z_levels)
+
+def _garment_mesh_default(params, body_verts, n_theta, z_levels):
+    return _garment_mesh_shirt(params, body_verts, n_theta, z_levels)
+
+GARMENT_MESH_FUNCS = {
+    'shirt': _garment_mesh_shirt,
+    'blouse': _garment_mesh_shirt,
+    'top': _garment_mesh_shirt,
+    't-shirt': _garment_mesh_shirt,
+    'tee': _garment_mesh_shirt,
+    'dress': _garment_mesh_dress,
+    'skirt': _garment_mesh_skirt,
+    'pants': _garment_mesh_pants,
+    'trousers': _garment_mesh_pants,
+    'jeans': _garment_mesh_pants,
+    'jacket': _garment_mesh_jacket,
+    'coat': _garment_mesh_jacket,
+    'blazer': _garment_mesh_jacket,
+    'hoodie': _garment_mesh_hoodie,
+    'sweatshirt': _garment_mesh_hoodie,
+    'vest': _garment_mesh_shirt,
+}
+
+def create_3d_figure(analysis: Dict) -> go.Figure:
+    """Create interactive 3D garment visualization."""
+    garment_type = analysis.get('garment_type', 'shirt').lower()
+    measurements = analysis.get('measurements', {})
+    features = analysis.get('features', {})
+    params = {**measurements, **features}
+    chest = params.get('bust', 92)
+    waist_m = params.get('waist', 74)
+    hip_m = params.get('hip', 96)
+    body_v, body_f = create_body_mesh(chest=chest, waist=waist_m, hip=hip_m)
+    fig = go.Figure()
+    # Body wireframe
+    n_t = 30
+    for i in range(0, len(body_v), n_t*3):
+        row = body_v[i:i+n_t]
+        fig.add_trace(go.Scatter3d(
+            x=row[:,0], y=row[:,1], z=row[:,2],
+            mode='lines', line=dict(color='#DDCCAA', width=1),
+            showlegend=False, hoverinfo='skip'))
+    # Garment
+    mesh_func = GARMENT_MESH_FUNCS.get(garment_type, _garment_mesh_default)
+    garment_traces = mesh_func(params, body_v, n_t, np.linspace(0, 170, 40))
+    for trace in garment_traces:
+        fig.add_trace(go.Scatter3d(
+            x=trace['x'], y=trace['y'], z=trace['z'],
+            mode=trace.get('mode', 'lines'),
+            line=trace.get('line', dict(width=3)),
+            name=trace.get('name', '')))
+    fig.update_layout(
+        scene=dict(
+            xaxis=dict(showgrid=False, showticklabels=False, title=''),
+            yaxis=dict(showgrid=False, showticklabels=False, title=''),
+            zaxis=dict(showgrid=False, showticklabels=False, title=''),
+            aspectmode='data',
+            camera=dict(eye=dict(x=1.5, y=1.5, z=0.8))),
+        margin=dict(l=0, r=0, t=30, b=0),
+        height=500,
+        title=dict(text=f'3D Preview: {garment_type.title()}', font=dict(size=14)),
+        paper_bgcolor='#fafafa',
+    )
+    return fig