datasets

app.py

@@ -165,12 +165,16 @@ with demo:
                 gr.Markdown('## Input Fragments')
                 gr.Markdown('Upload the file with 3D-coordinates of the input fragments in .pdb, .mol2 or .sdf format:')
                 input_file = gr.File(file_count='single', label='Input Fragments')
+                examples = gr.Dataset(
+                    components=[input_file],
+                    samples=[['examples/example_1.sdf'], ['examples/example_2.sdf']],
+                )
 
-                with gr.Row(css="#button {background-color: white; color: black;}"):
-                    with gr.Column():
-                        example1 = gr.Button('Example 1', elem_id='button')
-                    with gr.Column():
-                        example2 = gr.Button('Example 2', elem_id='button')
+                # with gr.Row(css="#button {background-color: white; color: black;}"):
+                #     with gr.Column():
+                #         example1 = gr.Button('Example 1', elem_id='button')
+                #     with gr.Column():
+                #         example2 = gr.Button('Example 2', elem_id='button')
 
                 button = gr.Button('Generate Linker!')
                 gr.Markdown('')
@@ -190,18 +194,23 @@ with demo:
         inputs=[input_file],
         outputs=[visualization, output_files],
     )
-
-    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.click(
+        fn=lambda inp: [inp, show_input(inp)],
+        inputs=[input_file],
+        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],