buttons

app.py

@@ -71,8 +71,10 @@ def read_molecule(path):
 def show_input(input_file):
     if input_file is None:
         return ''
-
-    path = input_file.name
+    if isinstance(input_file, str):
+        path = input_file
+    else:
+        path = input_file.name
     extension = path.split('.')[-1]
     if extension not in ['sdf', 'pdb', 'mol', 'mol2']:
         msg = output.INVALID_FORMAT_MSG.format(extension=extension)
@@ -169,8 +171,6 @@ with demo:
                         example1 = gr.Button('Example 1')
                     with gr.Column():
                         example2 = gr.Button('Example 2')
-                    with gr.Column():
-                        example3 = gr.Button('Example 3')
 
                 button = gr.Button('Generate Linker!')
                 gr.Markdown('')
@@ -192,20 +192,16 @@ with demo:
     )
 
     example1.click(
-        fn=lambda: 'examples/example_1.sdf',
+        fn=lambda: ['examples/example_1.sdf', show_input('examples/example_1.sdf')],
         inputs=[],
-        outputs=[input_file],
+        outputs=[input_file, visualization],
     )
     example2.click(
-        fn=lambda: 'examples/example_2.sdf',
+        fn=lambda: ['examples/example_2.sdf', show_input('examples/example_2.sdf')],
         inputs=[],
-        outputs=[input_file],
-    )
-    example3.click(
-        fn=lambda: 'examples/example_1.sdf',
-        inputs=[],
-        outputs=[input_file],
+        outputs=[input_file, visualization],
     )
+
     # examples = gr.Examples(
     #     examples=[['examples/example_1.sdf'], ['examples/example_2.sdf']],
     #     inputs=[input_file],