output files

app.py

@@ -146,7 +146,10 @@ def generate(input_file):
 
     generated_molecule = read_molecule_content(out_sdf)
     html = HTML_TEMPLATE.format(molecule=generated_molecule, fmt='sdf')
-    return IFRAME_TEMPLATE.format(html=html)
+    return [
+        IFRAME_TEMPLATE.format(html=html),
+        out_sdf,
+    ]
 
 
 demo = gr.Blocks()
@@ -163,11 +166,12 @@ with demo:
     
     gr.Markdown('')
     visualization = gr.HTML()
+    output_files = gr.File(file_count="single", label="Output files")
     
     button.click(
         fn=generate,
         inputs=[input_file],
-        outputs=[visualization],
+        outputs=[visualization, output_files],
     )
 
 demo.launch()