radio

app.py

@@ -91,7 +91,7 @@ def show_input(input_file):
     return output.IFRAME_TEMPLATE.format(html=html)
 
 
-def generate(input_file):
+def generate(input_file, samples):
     if input_file is None:
         return ''
 
@@ -165,6 +165,7 @@ 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,
@@ -193,6 +194,13 @@ with demo:
                 output_files = gr.File(file_count='multiple', label='Output Files')
             with gr.Column():
                 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,
@@ -201,39 +209,13 @@ with demo:
     )
     button.click(
         fn=generate,
-        inputs=[input_file],
+        inputs=[input_file, samples],
         outputs=[visualization, output_files],
     )
-
-    # def foo(idx):
-    #     print('INPUT:', idx)
-    #     input_path = f'examples/example_{idx+1}.sdf'
-    #     return [input_path, show_input(input_path)]
-
     examples.click(
         fn=lambda idx: [f'examples/example_{idx+1}.sdf', show_input(f'examples/example_{idx+1}.sdf')],
         inputs=[examples],
         outputs=[input_file, visualization]
     )
 
-    # example1.click(
-    #     fn=lambda: ['examples/example_1.sdf', show_input('examples/example_1.sdf')],
-    #     inputs=[],
-    #     outputs=[input_file, visualization],
-    # )
-    # example2.click(
-    #     fn=lambda: ['examples/example_2.sdf', show_input('examples/example_2.sdf')],
-    #     inputs=[],
-    #     outputs=[input_file, visualization],
-    # )
-
-    # examples = gr.Examples(
-    #     examples=[['examples/example_1.sdf'], ['examples/example_2.sdf']],
-    #     inputs=[input_file],
-    #     outputs=[visualization],
-    #     fn=show_input,
-    #     run_on_click=True,
-    #     cache_examples=False,
-    # )
-
 demo.launch(server_name=args.ip)