Add Download button

app.py

@@ -15,8 +15,10 @@ from src.details import (
 )
 from src.results import (
     clear_results,
+    clear_results_file,
     display_loading_message_for_results,
     display_results,
+    download_results,
     fetch_result_paths,
     load_results_dataframes,
     plot_results,
@@ -69,6 +71,8 @@ with gr.Blocks(fill_height=True, fill_width=True, css=".col_heading {width: 50%}
             results = gr.HTML()
             results_dataframe_1 = gr.Dataframe(visible=False)
             results_dataframe_2 = gr.Dataframe(visible=False)
+            download_results_btn = gr.Button("Download")
+            results_file = gr.File(visible=False)
         with gr.Tab("Configs"):
             load_configs_btn = gr.Button("Load", interactive=False)
             clear_configs_btn = gr.Button("Clear")
@@ -161,6 +165,14 @@ with gr.Blocks(fill_height=True, fill_width=True, css=".col_heading {width: 50%}
         fn=plot_results,
         inputs=[results_task, results_dataframe_1, results_dataframe_2],  # results,
         outputs=[results_plot_1, results_plot_2],
+    ).then(
+        fn=clear_results_file,
+        outputs=results_file,
+    )
+    download_results_btn.click(
+        fn=download_results,
+        inputs=results,
+        outputs=results_file,
     )
     gr.on(
         triggers=[clear_results_btn.click, clear_configs_btn.click],
@@ -175,6 +187,9 @@ with gr.Blocks(fill_height=True, fill_width=True, css=".col_heading {width: 50%}
             results_task,
             configs_task,
         ],
+    ).then(
+        fn=clear_results_file,
+        outputs=results_file,
     )
 
     # DETAILS:

src/results.py

@@ -1,4 +1,6 @@
 import asyncio
+import shutil
+import tempfile
 
 import gradio as gr
 import pandas as pd
@@ -204,3 +206,26 @@ def plot_results(task, *dfs):
         return fig_1, fig_2
     else:
         return None, None
+
+
+tmpdirname = None
+
+
+def download_results(results):
+    global tmpdirname
+    if results:
+        if tmpdirname:
+            shutil.rmtree(tmpdirname)
+        tmpdirname = tempfile.mkdtemp()
+        path = f"{tmpdirname}/results.html"
+        with open(path, "w") as f:
+            f.write(results)
+        return gr.File(path, visible=True)
+
+
+def clear_results_file():
+    global tmpdirname
+    if tmpdirname:
+        shutil.rmtree(tmpdirname)
+        tmpdirname = None
+    return gr.File(visible=False)