Update notebook_utils.py

notebook_utils.py

@@ -16,7 +16,7 @@ from typing import List, NamedTuple, Optional, Tuple
 
 import numpy as np
 from openvino.runtime import Core, Type, get_version
-from IPython.display import HTML, Image, display
+import gradio as gr
 
 import openvino as ov
 from openvino.runtime.passes import Manager, MatcherPass, WrapType, Matcher
@@ -607,7 +607,20 @@ class NotebookAlert(Exception):
         self.show_message()
 
     def show_message(self):
-        display(HTML(f"""<div class="alert alert-{self.alert_class}">{self.message}"""))
+        alert_colors = {
+            "info": "blue",
+            "warning": "yellow",
+            "success": "green",
+            "danger": "red"
+        }
+        
+        alert_color = alert_colors.get(self.alert_class, "gray")  # Default to gray if no match
+        
+        # Gradio will display the alert as an HTML block with a colored background
+        alert_html = f'<div style="padding: 10px; background-color: {alert_color}; color: white; border-radius: 5px;">{self.message}</div>'
+        
+        # You can use gr.HTML() to display this message in your Gradio app
+        gr.HTML(alert_html).render()
 
 
 class DeviceNotFoundAlert(NotebookAlert):