dreamlessx commited on
Commit
ffb9ba7
·
1 Parent(s): dc32c69

Fix side-by-side before/after output in Single Procedure tab

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -239,6 +239,9 @@ def process_image(image_rgb, procedure, intensity):
239
  composited = mask_composite(warped, image_bgr, mask)
240
  composited_rgb = cv2.cvtColor(composited, cv2.COLOR_BGR2RGB)
241
 
 
 
 
242
  displacement = np.mean(
243
  np.linalg.norm(manipulated.pixel_coords - face.pixel_coords, axis=1)
244
  )
@@ -254,7 +257,7 @@ def process_image(image_rgb, procedure, intensity):
254
  f"Processing time: {elapsed:.2f}s\n"
255
  f"Mode: TPS (CPU)"
256
  )
257
- return wireframe_rgb, mask_vis, composited_rgb, image_rgb_512, info
258
 
259
  except Exception as exc:
260
  logger.error("Processing failed: %s\n%s", exc, traceback.format_exc())
 
239
  composited = mask_composite(warped, image_bgr, mask)
240
  composited_rgb = cv2.cvtColor(composited, cv2.COLOR_BGR2RGB)
241
 
242
+ # Side-by-side comparison: original on left, result on right
243
+ sbs = np.hstack([image_rgb_512, composited_rgb])
244
+
245
  displacement = np.mean(
246
  np.linalg.norm(manipulated.pixel_coords - face.pixel_coords, axis=1)
247
  )
 
257
  f"Processing time: {elapsed:.2f}s\n"
258
  f"Mode: TPS (CPU)"
259
  )
260
+ return wireframe_rgb, mask_vis, composited_rgb, sbs, info
261
 
262
  except Exception as exc:
263
  logger.error("Processing failed: %s\n%s", exc, traceback.format_exc())