appointment booking - completed

README.md

@@ -159,7 +159,9 @@ To stop the Streamlit app, press `Ctrl+C` in the terminal where the app is runni
 **Prompt 1**: Book an appointment for priti.  
 **Prompt 2**: Yes or No.  
 **Reference**:  
-![appointment receptionist](screenshots/appointment_receptionist.png)  
+![appointment receptionist](screenshots/appointment_receptionist.png) 
+![Graph appointment booking](screenshots/graph_appointment_booking.png)  
+
 
 ### 4. Customer Support  
 **Reference**:  

src/langgraphagenticai/graph/graph_builder.py

@@ -1,6 +1,9 @@
-from langgraph.graph import StateGraph
-from langgraph.prebuilt import tools_condition
-
+from langgraph.graph import StateGraph, END, MessagesState
+from langgraph.prebuilt import tools_condition,ToolNode
+from langchain_core.prompts import ChatPromptTemplate
+import datetime
+#module import
+from src.langgraphagenticai.tools.customtool import book_appointment, cancel_appointment, get_next_available_appointment
 from src.langgraphagenticai.tools.search_tool import create_tool_node, get_tools
 from src.langgraphagenticai.node.chatbot_with_tool_node import ChatbotWithToolNode
 from src.langgraphagenticai.node.basic_chatbot_node import BasicChatbotNode
@@ -57,6 +60,54 @@ class GraphBuilder:
 
         # Set entry point and compile graph
         self.graph_builder.set_entry_point("chatbot")
+    
+    # Nodes
+    def call_caller_model(self,state: MessagesState):
+        state["current_time"] = datetime.datetime.now().strftime("%Y-%m-%d %H:%M")
+        response = self.caller_model.invoke(state)
+        return {"messages": [response]}
+    
+     # Edges
+    def should_continue_caller(self,state: MessagesState):
+        messages = state["messages"]
+        last_message = messages[-1]
+        if not last_message.tool_calls:
+            return "end"
+        else:
+            return "continue"
+            
+    def appointment_receptionist_bot_build_graph(self):
+        caller_tools = [book_appointment, get_next_available_appointment, cancel_appointment]
+        tool_node = ToolNode(caller_tools)
+
+        caller_pa_prompt = """You are a personal assistant, and need to help the user to book or cancel appointments, you should check the available appointments before booking anything. Be extremely polite, so much so that it is almost rude.
+        Current time: {current_time}
+        """
+
+        caller_chat_template = ChatPromptTemplate.from_messages([
+            ("system", caller_pa_prompt),
+            ("placeholder", "{messages}"),
+        ])
+
+        self.caller_model = caller_chat_template | self.llm.bind_tools(caller_tools)
+
+        # Add Nodes
+        self.graph_builder.add_node("agent", self.call_caller_model)
+        self.graph_builder.add_node("action", tool_node)
+
+        # Add Edges
+        self.graph_builder.add_conditional_edges(
+            "agent",
+            self.should_continue_caller,
+            {
+                "continue": "action",
+                "end": END,
+            },
+        )
+        self.graph_builder.add_edge("action", "agent")
+
+        # Set Entry Point and build the graph
+        self.graph_builder.set_entry_point("agent")
 
 
     def setup_graph(self, usecase: str):
@@ -67,6 +118,8 @@ class GraphBuilder:
             self.basic_chatbot_build_graph()
         elif usecase == "Chatbot with Tool":
             self.chatbot_with_tool_build_graph()
+        elif usecase == "Appointment Receptionist":
+            self.appointment_receptionist_bot_build_graph()
         else:
             raise ValueError("Invalid use case selected.")
         return self.graph_builder.compile()

src/langgraphagenticai/tools/customtool.py

@@ -0,0 +1,48 @@
+from langchain_core.tools import tool
+import datetime
+
+
+APPOINTMENTS = []
+
+@tool
+def get_next_available_appointment():
+    """Returns the next available appointment"""
+    current_time = datetime.datetime.now()
+    return f"One appointment available at {current_time + datetime.timedelta(minutes=(30 - current_time.minute % 30))}"
+
+@tool
+def book_appointment(appointment_year: int, appointment_month: int, appointment_day: int, appointment_hour: int, appointment_minute: int, appointment_name: str):
+    """Book an appointment at the given time, you must know the exact time to book
+
+    Args:
+        appointment_year: The year of the appointment
+        appointment_month: The month of the appointment
+        appointment_day: The day of the appointment
+        appointment_hour: The hour of the appointment
+        appointment_minute: The minute of the appointment
+        appointment_name: The name of the person booking the appointment
+    """
+    time = datetime.datetime(appointment_year, appointment_month, appointment_day, appointment_hour, appointment_minute)
+    for appointment in APPOINTMENTS:
+        if appointment.time >= time and appointment.time < time + datetime.timedelta(minutes=30):
+            return f"Appointment at {time} is already booked"
+    APPOINTMENTS.append({"time": time, "name": appointment_name})
+    return f"Appointment booked for {time}"
+
+@tool
+def cancel_appointment(appointment_year: int, appointment_month: int, appointment_day: int, appointment_hour: int, appointment_minute: int):
+    """Cancel the appointment at the given time
+
+    Args:
+        appointment_year: The year of the appointment
+        appointment_month: The month of the appointment
+        appointment_day: The day of the appointment
+        appointment_hour: The hour of the appointment
+        appointment_minute: The minute of the appointment
+    """
+    time = datetime.datetime(appointment_year, appointment_month, appointment_day, appointment_hour, appointment_minute)
+    for appointment in APPOINTMENTS:
+        if appointment["time"] == time:
+            APPOINTMENTS.remove(appointment)
+            return f"Appointment at {time} cancelled"
+    return f"No appointment found at {time}"

src/langgraphagenticai/ui/streamlitui/display_result.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from langchain_core.messages import HumanMessage,AIMessage,ToolMessage
 
+from src.langgraphagenticai.tools.customtool import APPOINTMENTS
+
 class DisplayResultStreamlit:
     def __init__(self,usecase,graph,user_message):
         self.usecase= usecase
@@ -37,8 +39,30 @@ class DisplayResultStreamlit:
                 elif type(message)==AIMessage and message.content:
                     with st.chat_message("assistant"):
                         st.write(message.content)
-            
-            # display graph
-            if graph:
-                st.write('state graph - workflow')
-                st.image(graph.get_graph(xray=True).draw_mermaid_png())
+        elif usecase == "Appointment Receptionist":
+            CONVERSATION=[]
+            CONVERSATION.append(HumanMessage(content=user_message, type="human"))
+            state = {
+                "messages": CONVERSATION,
+            }
+            print(state)
+            new_state = graph.invoke(state)
+            CONVERSATION.extend(new_state["messages"][len(CONVERSATION):])
+            col1, col2 = st.columns(2)
+            with col1:
+                for message in CONVERSATION:
+                    if message and message.content:
+                        if type(message) == HumanMessage:
+                            with st.chat_message("user"):
+                                st.write(message.content)
+                        else:
+                            with st.chat_message("assistant"):
+                                st.write(message.content)
+                            
+            with col2:
+                st.header("Appointments")
+                st.write(APPOINTMENTS)
+        # display graph
+        if graph:
+            st.write('state graph - workflow')
+            st.image(graph.get_graph(xray=True).draw_mermaid_png())

src/langgraphagenticai/ui/streamlitui/loadui.py

@@ -40,15 +40,10 @@ class LoadStreamlitUI:
                 
         if self.user_controls['selected_usecase'] == "Appointment Receptionist":
             col1, col2 = st.columns(2)
-
             with col1:
                 st.subheader("Appointment Manager")
-
             with col2:
                 st.subheader("Appointments")
-                
-            # TODO :: to be removed
-            st.warning("⌛Revamp is in progress...")
                     
         elif self.user_controls['selected_usecase']=="Customer Support":
             # TODO :: to be removed