Push the HDI input after the model selection

app.py

@@ -163,12 +163,28 @@ class ComparisonPlotter(DataPlotter):
         except ArithmeticError:
             pass
 
-def cplot(df):
-    def _plot(model_1, model_2, ci):
-        cp = ComparisonPlotter(df, model_1, model_2, ci)
+class ComparisonMenu:
+    def __init__(self, df, ci=0.95):
+        self.df = df
+        self.ci = ci
+
+    def __call__(self, model_1, model_2, ci):
+        cp = ComparisonPlotter(self.df, model_1, model_2, ci)
         return cp.plot()
 
-    return _plot
+    def build_and_get(self):
+        models = self.df['model'].unique()
+        choices = sorted(models, key=lambda x: x.lower())
+
+        for i in range(1, 3):
+            label = f'Model {i}'
+            yield gr.Dropdown(label=label, choices=choices)
+
+        yield gr.Number(value=self.ci,
+                        label='HDI',
+                        minimum=0,
+                        maximum=1,
+                        step=1e-2)
 
 #
 #
@@ -231,19 +247,10 @@ def layout(tab):
 
                 ''')
             with gr.Column():
-                models = df['model'].unique()
-                choices = sorted(models, key=lambda x: x.lower())
-                partial = ft.partial(gr.Dropdown, choices=choices)
-                drops = (partial(label=f'Model {x}') for x in range(1, 3))
-
-                ci = gr.Number(value=0.95, minimum=0, maximum=1, step=1e-2)
-
-                inputs = [
-                    *drops,
-                    ci,
-                ]
+                menu = ComparisonMenu(df)
+                inputs = list(menu.build_and_get())
                 button = gr.Button(value='Compare!')
-                button.click(cplot(df), inputs=inputs, outputs=[display])
+                button.click(menu, inputs=inputs, outputs=[display])
 
     with gr.Accordion('Disclaimer', open=False):
         gr.Markdown(docs['disclaimer'])