radio

app.py

@@ -165,10 +165,13 @@ def generate(input_file):
         molecule=generated_molecule_content,
         molecule_fmt='sdf',
     )
-    samples.update(visible=True)
     return [
         output.IFRAME_TEMPLATE.format(html=html),
         [inp_sdf, inp_xyz] + out_files,
+        gr.Radio.update(
+            choices=['Sample 1', 'Sample 2', 'Sample 3', 'Sample 4', 'Sample 5'],
+            value='Sample 1',
+        )
     ]
 
 
@@ -193,14 +196,9 @@ with demo:
                 gr.Markdown('Download files with the generated molecules here:')
                 output_files = gr.File(file_count='multiple', label='Output Files')
             with gr.Column():
+                gr.Markdown('## Visualization')
+                samples = gr.Radio(interactive=True, type='index', label='Samples')
                 visualization = gr.HTML()
-                samples = gr.Radio(
-                    choices=['Sample 1', 'Sample 2', 'Sample 3', 'Sample 4', 'Sample 5'],
-                    value='Sample 1',
-                    type='index',
-                    show_label=False,
-                    visible=False,
-                )
 
     input_file.change(
         fn=show_input,
@@ -210,7 +208,7 @@ with demo:
     button.click(
         fn=generate,
         inputs=[input_file],
-        outputs=[visualization, output_files],
+        outputs=[visualization, output_files, samples],
     )
     examples.click(
         fn=lambda idx: [f'examples/example_{idx+1}.sdf', show_input(f'examples/example_{idx+1}.sdf')],