From acb6152edad3eb5a69bd74ea7a543c33e5c18ea8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Pedersen?= <andrped94@gmail.com>
Date: Fri, 2 Jun 2023 09:20:33 +0200
Subject: [PATCH] 2D and 3D viewer of equal height + 2D viewer center aligned

---
 demo/src/gui.py | 22 ++++++++++++++++------
 1 file changed, 16 insertions(+), 6 deletions(-)

diff --git a/demo/src/gui.py b/demo/src/gui.py
index 8136e8e..dd9836d 100644
--- a/demo/src/gui.py
+++ b/demo/src/gui.py
@@ -22,7 +22,8 @@ def __init__(self, model_name:str = None, class_name:str = None, cwd:str = None)
         self.volume_renderer = gr.Model3D(
             clear_color=[0.0, 0.0, 0.0, 0.0],
             label="3D Model",
-            visible=True
+            visible=True,
+            elem_id="model-3d",
         ).style(height=512)
 
     def combine_ct_and_seg(self, img, pred):
@@ -47,9 +48,18 @@ def get_img_pred_pair(self, k):
         return out
 
     def run(self):
-        with gr.Blocks() as demo:
+        css="""
+        #model-3d {
+        height: 512px;
+        }
+        #model-2d {
+        height: 512px;
+        margin: auto;
+        }
+        """
+        with gr.Blocks(css=css) as demo:
 
-            with gr.Row().style(equal_height=True):
+            with gr.Row():
                 file_output = gr.File(
                     file_types=[".nii", ".nii.nz"],
                     file_count="single"
@@ -63,7 +73,7 @@ def run(self):
                     outputs=self.volume_renderer
                 )
             
-            with gr.Row().style(equal_height=True):
+            with gr.Row():
                 gr.Examples(
                     examples=[self.cwd + "test-volume.nii"],
                     inputs=file_output,
@@ -72,12 +82,12 @@ def run(self):
                     cache_examples=True,
                 )
             
-            with gr.Row().style(equal_height=True):
+            with gr.Row():
                 with gr.Box():
                     image_boxes = []
                     for i in range(self.nb_slider_items):
                         visibility = True if i == 1 else False
-                        t = gr.AnnotatedImage(visible=visibility)\
+                        t = gr.AnnotatedImage(visible=visibility, elem_id="model-2d")\
                             .style(color_map={self.class_name: "#ffae00"}, height=512, width=512)
                         image_boxes.append(t)