Show no mitigations status

app.py

@@ -79,6 +79,7 @@ class UI:
                 """<h2> Mitigations </h2> 
                 Select a potential risk to determine possible mitigations. """
                 )
+                self.mitigations_text = gr.Markdown()
                 self.mitigations = gr.DataFrame(label=None, visible=False)
             
         gr.Markdown("---")
@@ -115,7 +116,7 @@ class UI:
                 fn=mitigations,
                 inputs=[self.assessed_risks, self.taxonomy], 
                 # NOTETOSELF: Intent based risk is stored in self.risk (if needed)
-                outputs=[self.assessed_risk_definition, self.relatedrisks, self.mitigations]
+                outputs=[self.assessed_risk_definition, self.relatedrisks, self.mitigations, self.mitigations_text]
             )
         return demo
 

executor.py

@@ -56,7 +56,7 @@ def risk_identifier(usecase: str,
     
 
 @lru_cache
-def mitigations(riskid: str, taxonomy: str) -> tuple[gr.Markdown, gr.Dataset, gr.DataFrame]:
+def mitigations(riskid: str, taxonomy: str) -> tuple[gr.Markdown, gr.Dataset, gr.DataFrame, gr.Markdown]:
     """
     For a specific risk (riskid), returns
     (a) a risk description
@@ -104,13 +104,17 @@ def mitigations(riskid: str, taxonomy: str) -> tuple[gr.Markdown, gr.Dataset, gr
         asamples = None
         asample_labels = None
         mitdf = pd.DataFrame()
+        
     else:
         alabel = f"Mitigation actions related to risk {riskid}."
         asamples = action_ids
         asample_labels = [ran.get_action_by_id(i).description for i in asamples] # type: ignore
         asample_name = [ran.get_action_by_id(i).name for i in asamples] #type: ignore
         mitdf = pd.DataFrame({"Mitigation": asample_name, "Description": asample_labels})
+        
+    status = gr.Markdown(" ") if len(mitdf) > 0 else gr.Markdown("No mitigations found.")
 
     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)) 
+            gr.DataFrame(mitdf, wrap=True, show_copy_button=True, show_search="search", label=alabel, visible=True),
+            status)