add blueprint for better viewer experience

app.py

@@ -17,6 +17,39 @@ model = AsymmetricCroCo3DStereo.from_pretrained(
 ).to(DEVICE)
 
 
+def create_blueprint(image_name_list: list[str], log_path: str) -> rrb.Blueprint:
+    # dont show 2d views if there are more than 4 images as to not clutter the view
+    if len(image_name_list) > 4:
+        blueprint = rrb.Blueprint(
+            rrb.Horizontal(
+                rrb.Spatial3DView(origin=f"{log_path}"),
+            ),
+            collapse_panels=True,
+        )
+    else:
+        blueprint = rrb.Blueprint(
+            rrb.Horizontal(
+                contents=[
+                    rrb.Spatial3DView(origin=f"{log_path}"),
+                    rrb.Vertical(
+                        contents=[
+                            rrb.Spatial2DView(
+                                origin=f"{log_path}/camera_{i}/pinhole/",
+                                contents=[
+                                    "+ $origin/**",
+                                ],
+                            )
+                            for i in range(len(image_name_list))
+                        ]
+                    ),
+                ],
+                column_shares=[3, 1],
+            ),
+            collapse_panels=True,
+        )
+    return blueprint
+
+
 @spaces.GPU
 def predict(image_name_list: list[str] | str):
     # check if is list or string and if not raise error
@@ -38,9 +71,12 @@ def predict(image_name_list: list[str] | str):
         device=DEVICE,
         batch_size=1,
     )
+
+    blueprint: rrb.Blueprint = create_blueprint(image_name_list, log_path)
+    rr.send_blueprint(blueprint)
+
     rr.set_time_sequence("sequence", 0)
     log_optimized_result(optimized_results, log_path)
-    # blueprint = rrb.Spatial3DView(origin="cube")
     rr.save(filename.as_posix())
     return filename.as_posix()