travel planner - completed

README.md

@@ -168,6 +168,11 @@ To stop the Streamlit app, press `Ctrl+C` in the terminal where the app is runni
 ![cs flow](screenshots/cs_flow.png)  
 ![customer support](screenshots/customer_support.png)  
 
+### 5. Travel Planner 
+**Reference**:  
+![tp flow](screenshots/tp_flow.png)  
+![Travel planner](screenshots/travel_planner.png)  
+
 ---
 
 ## 📚 Templates & Community  

src/langgraphagenticai/graph/graph_builder.py

@@ -3,12 +3,14 @@ from langgraph.prebuilt import tools_condition,ToolNode
 from langchain_core.prompts import ChatPromptTemplate
 import datetime
 #module import
+from src.langgraphagenticai.node import travel_planner_node
 from src.langgraphagenticai.node.customer_support_chatbot import Customer_Support_Bot
 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
 from src.langgraphagenticai.state.state import State
+from src.langgraphagenticai.node.travel_planner_node import TravelPlannerNode
 
 class GraphBuilder:
     """
@@ -62,6 +64,23 @@ class GraphBuilder:
         # Set entry point and compile graph
         self.graph_builder.set_entry_point("chatbot")
     
+    def travel_planner_build_graph(self):
+        """
+            Builds a standalone travel planning graph with itinerary generation.
+        """
+        # Initialize the Travel Planner node
+        travel_planner_node = TravelPlannerNode(self.llm)
+
+        # Add the Travel Planner node to the graph
+        self.graph_builder.add_node("travel_planner", travel_planner_node.process)
+
+        # Set the entry point to the Travel Planner node
+        self.graph_builder.set_entry_point("travel_planner")
+
+        # Define the edge to end the graph after the Travel Planner completes
+        self.graph_builder.add_edge("travel_planner", END)
+    
+    # Helper methods          
     # Nodes
     def call_caller_model(self,state: MessagesState):
         state["current_time"] = datetime.datetime.now().strftime("%Y-%m-%d %H:%M")
@@ -122,6 +141,8 @@ class GraphBuilder:
             self.basic_chatbot_build_graph()
         elif usecase == "Chatbot with Tool":
             self.chatbot_with_tool_build_graph()
+        elif usecase == "Travel Planner":
+            self.travel_planner_build_graph()
         elif usecase == "Appointment Receptionist":
             self.appointment_receptionist_bot_build_graph()
         elif usecase =="Customer Support":

src/langgraphagenticai/main.py

@@ -4,6 +4,7 @@ from src.langgraphagenticai.graph.graph_builder import GraphBuilder
 from src.langgraphagenticai.ui.streamlitui.loadui import LoadStreamlitUI
 
 import streamlit as st
+import json
 
 
 # MAIN Function START
@@ -52,6 +53,17 @@ def load_langgraph_agenticai_app():
 
                 # Display output in UI
                 try:
+                     # Add travel-specific parameters to message
+                    if usecase == "Travel Planner":
+                        travel_params = {
+                            'source' : user_input.get('source'),
+                            'city': user_input.get('destination'),
+                            'start_date': user_input.get('start_date').isoformat(),
+                            'end_date': user_input.get('end_date').isoformat(),
+                            'interests': user_input.get('preferences'),
+                            'user_message' : user_message
+                        }
+                        user_message = travel_params
                     DisplayResultStreamlit(usecase,graph,user_message).display_result_on_ui()
                 except Exception as e:
                     st.error(f"Error: Failed to display results on UI - {e}")
@@ -62,19 +74,4 @@ def load_langgraph_agenticai_app():
     except Exception as e:
         st.error(f"Unexpected error occurred: {e}")
 
-        
-        
-                        
-       
-                
-                        
-                    
-                
-
-
-            
-
-        
-                
-                
-                
+        

src/langgraphagenticai/node/travel_planner_node.py

@@ -0,0 +1,71 @@
+from typing import Dict
+from langchain_core.prompts import ChatPromptTemplate
+from src.langgraphagenticai.state.state import State
+from langchain_core.messages import HumanMessage, AIMessage
+import json
+
+class TravelPlannerNode:
+    """
+    Travel Planner logic implementation.
+    """
+    def __init__(self, model):
+        self.llm = model
+        self.itinerary_prompt = ChatPromptTemplate.from_messages([
+    ("system", "You are a creative and detail-oriented travel assistant. Create a visually appealing and highly detailed travel itinerary from {source} to {city} for the user. "
+               "The itinerary should be based on the user's interests: {interests} and their additional request: {user_message}. "
+               "The trip will take place from {start_date} to {end_date}. "
+               "**Follow these guidelines to create the itinerary:**\n"
+               "1. **Structure:** Organize the itinerary into clear sections with subheadings for each day (e.g., 'Day 1: Arrival and Exploration').\n"
+               "2. **Activities:** Include a mix of activities (e.g., sightseeing, dining, adventure, relaxation) tailored to the user's interests.\n"
+               "3. **Flights and Accommodations:** Provide flight options (if applicable) and recommend accommodations with actual links to their location on Google Maps.\n"
+               "4. **Cost Estimation:** Add approximate costs for activities, flights, and accommodations.\n"
+               "5. **Emojis:** Use relevant emojis (e.g., ✈️ for flights, 🏨 for hotels, 🍽️ for dining) to make the itinerary visually engaging.\n"
+               "6. **Markdown Formatting:** Use bullet points, bold text, and subheadings for clarity and readability.\n"
+               "7. **Links:** Include clickable links for booking flights, accommodations, and activity reservations where applicable.\n"
+               "8. **No Introductory Explanation:** Jump straight into the itinerary without unnecessary introductions.\n"
+               "**Example Format:**\n"
+               "### Day 1: Arrival in {city} ✈️\n"
+               "- **Morning:** Arrive at {city} Airport. Transfer to your hotel 🏨. [Hotel Map Link](https://maps.google.com)\n"
+               "- **Afternoon:** Explore landmark and enjoy lunch at restaurant 🍽️. (Cost: ~ 50)\n"
+               "- **Evening:** Relax at location and enjoy dinner at restaurant 🍷. (Cost: ~ 70)\n"
+               "### Day 2: Adventure and Sightseeing 🌄\n"
+               "- **Morning:** Visit attraction and enjoy activity. (Cost: ~ 30)\n"
+               "- **Afternoon:** Lunch at restaurant 🍽️. (Cost: ~ 40)\n"
+               "- **Evening:** Explore location and enjoy a guided tour. (Cost: ~ 60)\n"
+               "Ensure the itinerary is engaging, practical, and tailored to the user's preferences and cost should be in local currency"),
+    ("human", "Create an itinerary for my trip"),
+])
+    def process(self, state: State) -> Dict:
+        """
+        Processes the input state and generates a travel itinerary.
+        """
+        # Validate required fields
+        content_str = state.get("messages")[0].content.replace("'", '"')
+        content = json.loads(content_str)
+
+        # Validate required fields
+        if not content.get("city") or not content.get("interests") or not content.get("start_date") or not content.get("end_date"):
+            return {
+                "messages": [AIMessage(content="Please provide a city, interests, start date, and end date to generate an itinerary.")],
+                "itinerary": ""
+            }
+
+        # Generate the itinerary using the LLM
+        response = self.llm.invoke(self.itinerary_prompt.format_messages(
+            source=content["source"],
+            user_message = content["user_message"],
+            city=content["city"],
+            interests=content["interests"],  # No need to join, as it's already a string
+            start_date=content["start_date"],
+            end_date=content["end_date"]
+        ))
+
+        # Update the state with the generated itinerary
+        return {
+            "messages": state.get("messages", []) + [AIMessage(content=response.content)],  # Safely access "messages"
+            "itinerary": AIMessage(content=response.content),
+            "city": content["city"],
+            "interests": content["interests"],
+            "start_date": content["start_date"],
+            "end_date": content["end_date"]
+        }

src/langgraphagenticai/state/state.py

@@ -1,9 +1,20 @@
 from typing import Annotated
 from typing_extensions import TypedDict
 from langgraph.graph.message import add_messages
+from typing import TypedDict, Annotated, List
+from langchain_core.messages import HumanMessage, AIMessage
 
 class State(TypedDict):
     """
     Represents the structure of the state used in the graph.
     """
     messages: Annotated[list, add_messages]
+    
+    
+class PlannerState(TypedDict):
+    messages: Annotated[List[HumanMessage | AIMessage], "Conversation history"]
+    city: str
+    interests: List[str]
+    itinerary: str
+    start_date: str 
+    end_date: str    

src/langgraphagenticai/ui/streamlitui/display_result.py

@@ -1,5 +1,6 @@
 import streamlit as st
 from langchain_core.messages import HumanMessage,AIMessage,ToolMessage
+import json
 
 from src.langgraphagenticai.tools.customtool import APPOINTMENTS
 from src.langgraphagenticai.tools.customer_support_tools import customers_database, data_protection_checks
@@ -40,6 +41,8 @@ class DisplayResultStreamlit:
                 elif type(message)==AIMessage and message.content:
                     with st.chat_message("assistant"):
                         st.write(message.content)
+        elif usecase == "Travel Planner":
+            self._display_travel_planner_results()
         elif usecase == "Appointment Receptionist":
             CONVERSATION=[]
             CONVERSATION.append(HumanMessage(content=user_message, type="human"))
@@ -88,4 +91,61 @@ class DisplayResultStreamlit:
         # display graph
         if graph:
             st.write('state graph - workflow')
-            st.image(graph.get_graph(xray=True).draw_mermaid_png())
+            st.image(graph.get_graph(xray=True).draw_mermaid_png())
+            
+    def _display_travel_planner_results(self):
+        # Extract travel parameters from message
+        CONVERSATION=[]
+        CONVERSATION.append(HumanMessage(content=str(self.user_message), type="human"))
+        state = {
+                "messages": CONVERSATION,
+            }
+        print(state)
+        # Invoke the graph
+        response = self.graph.invoke(state)
+
+        # Display results
+        main_col, side_col = st.columns([3, 1])
+        
+        with main_col:
+            st.subheader("✈️ Travel Itinerary")
+            if type(response['messages'][1]) == AIMessage:
+                self._display_itinerary_details(response['messages'][1].content)
+            else:
+                st.warning("No itinerary generated yet.")
+
+        with side_col:
+            st.subheader("📌 Travel Details")
+            st.json({
+                "📍 Source": self.user_message.get('source', ''),
+                "📍 Destination": self.user_message.get('city', ''),
+                "📅 Dates": f"{self.user_message.get('start_date', '')} to {self.user_message.get('end_date', '')}",
+                "🎯 Interests": self.user_message.get('interests', ''),
+                "🧑‍💻 User Addition Request": self.user_message.get('user_message', '')
+            })
+
+    def _display_itinerary_details(self, itinerary_content):
+        """
+        Displays the itinerary details in a structured format.
+        """
+        with st.expander("📅 Full Itinerary"):
+            # Parse the itinerary content
+            sections = {
+                "Destination": itinerary_content
+            }
+            
+            # Display destination and dates
+            st.markdown(f"{sections['Destination']}")
+           
+
+    def _display_tool_calls(self, message):
+        """
+        Displays details of tool calls made during the itinerary generation.
+        """
+        with st.expander("⚙️ System Operations"):
+            st.markdown("**🔧 Tool Execution Details**")
+            st.json({
+                "Tool Used": message.name,
+                "Parameters": message.additional_kwargs,
+                "Result": message.content
+            })

src/langgraphagenticai/ui/streamlitui/loadui.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import os
+from datetime import date
 
 from src.langgraphagenticai.ui.uiconfigfile import Config
 from langchain_core.messages import AIMessage, HumanMessage
@@ -29,7 +30,10 @@ class LoadStreamlitUI:
                 # API key input
                 self.user_controls["GROQ_API_KEY"] = st.session_state["GROQ_API_KEY"] = st.text_input("API Key",
                                                                                                       type="password")
-                
+                # Validate API key
+                if not self.user_controls["GROQ_API_KEY"]:
+                    st.warning("⚠️ Please enter your GROQ API key to proceed.")
+                   
             
             # Use case selection
             self.user_controls["selected_usecase"] = st.selectbox("Select Usecases", usecase_options)
@@ -64,7 +68,31 @@ class LoadStreamlitUI:
                 if st.button('Clear Chat'):
                     st.session_state.message_history = []
 
-       
+        elif self.user_controls['selected_usecase']=="Travel Planner":
+            st.subheader("✈️ AI Travel Planner")
+            col1, col2 = st.columns(2)
+
+            with col1:
+                source = st.text_input("📍 Source", value="London", help="Enter your travel source")
+                destination = st.text_input("📍 Destination", value="Goa", help="Enter your travel destination")
+                preferences = st.text_area(
+                    "🎯 Travel Preferences",
+                    placeholder="E.g., I prefer beach destinations, luxury stays, and adventure activities.",
+                    help="Describe your travel preferences"
+                )
+
+            with col2:
+                start_date = st.date_input("📅 Start Date", value=date.today(), help="Select your travel start date")
+                end_date = st.date_input("📅 End Date", value=date.today(), help="Select your travel end date")
+
+            if destination and preferences and start_date and end_date:
+                self.user_controls.update({
+                "source": source,
+                "destination": destination,
+                "preferences": preferences,
+                "start_date": start_date,
+                "end_date": end_date,
+            })
             
 
         return self.user_controls

src/langgraphagenticai/ui/uiconfigfile.ini

@@ -1,6 +1,6 @@
 [DEFAULT]
 PAGE_TITLE = LangGraph: Build Stateful Agentic AI graph
 LLM_OPTIONS = Groq
-USECASE_OPTIONS = Basic Chatbot, Chatbot with Tool, Appointment Receptionist, Customer Support
+USECASE_OPTIONS = Basic Chatbot, Chatbot with Tool, Travel Planner, Appointment Receptionist, Customer Support
 GROQ_MODEL_OPTIONS = mixtral-8x7b-32768, llama3-8b-8192, llama3-70b-8192, gemma-7b-i