download risks

.gitignore

@@ -1,4 +1,5 @@
 .DS_Store
 .env
 scratch.py
-__pycache__
+__pycache__
+static/download.json

app.py

@@ -81,6 +81,7 @@ class UI:
                 )
                 self.mitigations_text = gr.Markdown()
                 self.mitigations = gr.DataFrame(label=None, visible=False)
+                self.download = gr.DownloadButton("Download JSON", visible=False)
             
         gr.Markdown("---")
         gr.Markdown("<br>")
@@ -109,9 +110,10 @@ class UI:
                     self.model_name_or_path,
                     self.taxonomy
                 ],
-                outputs=[self.assessment_sec, self.risks, self.assessed_risks],
+                outputs=[self.assessment_sec, self.risks, self.assessed_risks, self.download],
                 api_name="risk_identifier"
             )
+
             self.assessed_risks.select(
                 fn=mitigations,
                 inputs=[self.assessed_risks, self.taxonomy], 

executor.py

@@ -10,7 +10,9 @@ import json
 from typing import List, Dict, Any
 import pandas as pd
 import gradio as gr
-
+import datetime
+from pathlib import Path
+import json
 
 from risk_atlas_nexus.blocks.inference import WMLInferenceEngine
 from risk_atlas_nexus.blocks.inference.params import WMLInferenceEngineParams
@@ -48,11 +50,21 @@ def risk_identifier(usecase: str,
     sample_labels = [r.name if r else r.id for r in risks]
 
     out_sec = gr.Markdown("""<h2> Potential Risks </h2> """)
+
+    # write out a JSON
+    data = {'time': str(datetime.datetime.now(datetime.timezone.utc)),
+                'intent': usecase,
+                'model': model_name_or_path,
+                'taxonomy': taxonomy,
+                'risks': [json.loads(r.json()) for r in risks]
+        }
+    file_path = Path("static/download.json")
+    file_path.write_text(json.dumps(data, indent=4), encoding='utf-8')
         
     #return out_df
     return out_sec, gr.State(risks), gr.Dataset(samples=[r.id for r in risks], 
                                      sample_labels=sample_labels, 
-                                     samples_per_page=50, visible=True, label="Estimated by an LLM.")
+                                     samples_per_page=50, visible=True, label="Estimated by an LLM."), gr.DownloadButton("Download JSON", visible=True, value="static/download.json")
     
 
 @lru_cache