Interface

app.py

@@ -218,7 +218,7 @@ with demo:
                 gr.Markdown('')
                 gr.Markdown('## Output Files')
                 gr.Markdown('Download files with the generated molecules here:')
-                output_files = gr.File(file_count='multiple', label='Output Files')
+                output_files = gr.File(file_count='multiple', label='Output Files', interactive=False)
             with gr.Column():
                 gr.Markdown('## Visualization')
                 # gr.Markdown('Below you will see input and output molecules')
@@ -238,9 +238,14 @@ with demo:
         outputs=[visualization],
     )
     examples.click(
-        fn=lambda idx: [f'examples/example_{idx+1}.sdf', 10, show_input(f'examples/example_{idx+1}.sdf')],
+        fn=lambda idx: [
+            f'examples/example_{idx+1}.sdf',
+            10,
+            show_input(f'examples/example_{idx+1}.sdf'),
+            gr.Radio(value='Sample 1', visible=False)
+        ],
         inputs=[examples],
-        outputs=[input_file, n_steps, visualization]
+        outputs=[input_file, n_steps, visualization, samples]
     )
     button.click(
         fn=generate,
@@ -252,5 +257,10 @@ with demo:
         inputs=[samples, output_files],
         outputs=[visualization],
     )
+    output_files.clear(
+        fn=lambda: ['', gr.Radio(value='Sample 1', visible=False)],
+        inputs=[],
+        outputs=[visualization, samples],
+    )
 
 demo.launch(server_name=args.ip)