generate lang graph

agent.py

@@ -16,6 +16,9 @@ from langchain_groq import ChatGroq
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint, HuggingFaceEmbeddings,  HuggingFacePipeline
 from langchain_anthropic import ChatAnthropic
 
+##!other libraries 
+from IPython.display import Image, display
+
 
 load_dotenv()
 
@@ -185,4 +188,10 @@ def build_graph(provider: str = "huggingface"):
     # Compile graph
     compiled_graph = builder.compile() # This line should already be there or be the next line
 
+    try:
+        display(Image(graph.get_graph().draw_mermaid_png()))
+    except Exception:
+        # This requires some extra dependencies and is optional
+        pass
+
     return compiled_graph