state graph display

README.md

@@ -159,6 +159,7 @@ To stop the Streamlit app, press `Ctrl+C` in the terminal where the app is runni
 **Prompt 2**: Yes or No.  
 **Reference**:  
 ![alt text](appointment_receptionist.png)  
+![alt text](state_graph_ar.png)
 
 ### 4. Customer Support  
 **Reference**:  

app.py

@@ -31,7 +31,7 @@ if __name__ == "__main__":
     user_input = ui.load_streamlit_ui()
 
     
-  
+    graph_display =''
     # is_add_message_to_history = st.session_state["chat_with_history"] 
     
     if user_input['selected_usecase'] == "Appointment Receptionist":
@@ -39,7 +39,7 @@ if __name__ == "__main__":
             # Configure LLM
             obj_llm_config = GroqLLM(user_controls_input=user_input)
             model = obj_llm_config.get_llm_model()
-            CONVERSATION,APPOINTMENTS= (submit_message(model))
+            CONVERSATION,APPOINTMENTS,graph_display= (submit_message(model))
             
             col1, col2 = st.columns(2)
             with col1:
@@ -122,7 +122,8 @@ if __name__ == "__main__":
             # Initialize and set up the graph based on use case
             usecase = user_input['selected_usecase']
             graph_builder = GraphBuilder(model)
-            graph = graph_builder.setup_graph(usecase)
+            graph_display = graph = graph_builder.setup_graph(usecase)
+            
 
             # Prepare state and invoke the graph
             initial_state = {"messages": [user_message]}
@@ -155,6 +156,12 @@ if __name__ == "__main__":
                         with st.chat_message("assistant"):
                             st.write(message.content)
                             
+                            
+    # display graph
+    if graph_display:
+        st.write('state graph workflow')
+        st.image(graph_display.get_graph(xray=True).draw_mermaid_png())
+                            
                 
                         
                     

src/langgraphagent/caller_agent.py

@@ -74,6 +74,8 @@ class Caller_Agent:
         self.caller_app = caller_workflow.compile()
         
         
+        
+        
    
     
 
@@ -84,10 +86,10 @@ class Caller_Agent:
             "messages": CONVERSATION,
         }
         print(state)
-        APPOINTMENT = self.call_tool()
+        graph = self.call_tool()
         new_state = self.caller_app.invoke(state)
         CONVERSATION.extend(new_state["messages"][len(CONVERSATION):])
-        return CONVERSATION, APPOINTMENTS
+        return CONVERSATION, APPOINTMENTS,self.caller_app