app

app.py

@@ -164,17 +164,12 @@ with demo:
             with gr.Column():
                 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')
+                input_file = gr.File(file_count='single', label='Input Fragments') 
                 examples = gr.Dataset(
-                    components=[input_file],
+                    components=[gr.File(visible=False)],
                     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')
+                    type='index',
+                ) 
 
                 button = gr.Button('Generate Linker!')
                 gr.Markdown('')
@@ -194,9 +189,15 @@ with demo:
         inputs=[input_file],
         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 inp: [inp, show_input(inp)],
-        inputs=[input_file],
+        fn=lambda idx: [f'examples/example_{idx+1}.sdf', show_input(f'examples/example_{idx+1}.sdf')],
+        inputs=[examples],
         outputs=[input_file, visualization]
     )