radio

app.py

@@ -91,6 +91,13 @@ def show_input(input_file):
     return output.IFRAME_TEMPLATE.format(html=html)
 
 
+def draw_sample(idx, output_files):
+    print(idx)
+    print(output_files)
+    print(output_files[0].name)
+    return
+
+
 def generate(input_file):
     if input_file is None:
         return ''
@@ -152,7 +159,6 @@ def generate(input_file):
         out_xyz = f'results/output_{i+1}_{name}_.xyz'
         out_sdf = f'results/output_{i+1}_{name}_.sdf'
         subprocess.run(f'obabel {out_xyz} -O {out_sdf}', shell=True)
-        out_files.append(out_xyz)
         out_files.append(out_sdf)
     print('Converted to SDF')
 
@@ -167,7 +173,7 @@ def generate(input_file):
     )
     return [
         output.IFRAME_TEMPLATE.format(html=html),
-        [inp_sdf, inp_xyz] + out_files,
+        [inp_sdf] + out_files,
         gr.Radio.update(
             choices=['Sample 1', 'Sample 2', 'Sample 3', 'Sample 4', 'Sample 5'],
             value='Sample 1',
@@ -197,8 +203,8 @@ with demo:
                 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(interactive=True, type='index', label='Samples')
 
     input_file.change(
         fn=show_input,
@@ -215,5 +221,10 @@ with demo:
         inputs=[examples],
         outputs=[input_file, visualization]
     )
+    samples.change(
+        fn=draw_sample,
+        inputs=[samples, output_files],
+        outputs=[],
+    )
 
 demo.launch(server_name=args.ip)