Include risk descriptions

app.py

@@ -47,7 +47,7 @@ class UI:
                 )
                 self.model_name_or_path = gr.Dropdown(
                     #choices=['codellama/codellama-34b-instruct-hf', 'google/flan-t5-xl', 'google/flan-t5-xxl', 'google/flan-ul2', 'ibm/granite-13b-instruct-v2', 'ibm/granite-20b-code-instruct', 'ibm/granite-20b-multilingual', 'ibm/granite-3-2-8b-instruct-preview-rc', 'ibm/granite-3-2b-instruct', 'ibm/granite-3-8b-instruct', 'ibm/granite-34b-code-instruct', 'ibm/granite-3b-code-instruct', 'ibm/granite-8b-code-instruct', 'ibm/granite-guardian-3-2b', 'ibm/granite-guardian-3-8b', 'meta-llama/llama-2-13b-chat', 'meta-llama/llama-3-1-70b-instruct', 'meta-llama/llama-3-1-8b-instruct', 'meta-llama/llama-3-2-11b-vision-instruct', 'meta-llama/llama-3-2-1b-instruct', 'meta-llama/llama-3-2-3b-instruct', 'meta-llama/llama-3-2-90b-vision-instruct', 'meta-llama/llama-3-3-70b-instruct', 'meta-llama/llama-3-405b-instruct', 'meta-llama/llama-guard-3-11b-vision', 'mistralai/mistral-large', 'mistralai/mixtral-8x7b-instruct-v01'],
-                    choices=["ibm/granite-20b-code-instruct"]
+                    choices=["ibm/granite-20b-code-instruct"],
                     value="ibm/granite-20b-code-instruct",
                     multiselect=False,
                     label="Choose language model to use",
@@ -65,6 +65,7 @@ class UI:
             
                 self.assessment_sec = gr.Markdown()
                 self.assessed_risks = gr.Dataset(label=None, visible=False)
+                self.assessed_risk_definition = gr.Markdown()
 
                 gr.Markdown(
                 """<h2> Related Risks </h2> 
@@ -114,7 +115,7 @@ class UI:
                 fn=mitigations,
                 inputs=[self.assessed_risks, self.taxonomy], 
                 # NOTETOSELF: Intent based risk is stored in self.risk (if needed)
-                outputs=[self.relatedrisks, self.mitigations]
+                outputs=[self.assessed_risk_definition, self.relatedrisks, self.mitigations]
             )
         return demo
 

executor.py

@@ -1,3 +1,4 @@
+from ast import Attribute
 from dotenv import load_dotenv
 
 load_dotenv(override=True)
@@ -55,15 +56,23 @@ def risk_identifier(usecase: str,
     
 
 @lru_cache
-def mitigations(riskid: str, taxonomy: str) -> tuple[gr.Dataset, gr.DataFrame]:
+def mitigations(riskid: str, taxonomy: str) -> tuple[gr.Markdown, gr.Dataset, gr.DataFrame]:
     """
     For a specific risk (riskid), returns
-    (a) related risks - as a dataset
-    (b) mitigations
+    (a) a risk description
+    (b) related risks - as a dataset
+    (c) mitigations
 
     """
     
+    try:
+        risk_desc = ran.get_risk_by_id(riskid).description # type: ignore
+        risk_sec = f"<h3>Description: </h3> {risk_desc}"
+    except AttributeError:
+        risk_sec = ""
+
     related_risk_ids = ran.get_related_risk_ids_by_risk_id(riskid)
+
     action_ids = []
 
     if taxonomy == "ibm-risk-atlas":
@@ -102,5 +111,6 @@ def mitigations(riskid: str, taxonomy: str) -> tuple[gr.Dataset, gr.DataFrame]:
         asample_name = [ran.get_action_by_id(i).name for i in asamples] #type: ignore
         mitdf = pd.DataFrame({"Mitigation": asample_name, "Description": asample_labels})
 
-    return (gr.Dataset(samples=samples, label=label, sample_labels=sample_labels, visible=True),
+    return (gr.Markdown(risk_sec), 
+            gr.Dataset(samples=samples, label=label, sample_labels=sample_labels, visible=True),
             gr.DataFrame(mitdf, wrap=True, show_copy_button=True, show_search="search", label=alabel, visible=True))