Save local changes before pull

agent.py

@@ -2,9 +2,9 @@ import os
 from langchain_groq import ChatGroq
 from langchain.agents import initialize_agent, AgentType
 from memory import memory
-from tools import tools
+from tools import tools  # Import Flight Booking Tool
 
-# Load API Key for Groq from environment variables
+# Load API Key
 API_KEY = os.getenv("API_KEY")
 
 # Ensure API Key is set
@@ -19,11 +19,17 @@ llm = ChatGroq(
     max_tokens=512,
 )
 
-# Initialize the conversational agent
+# Initialize the conversational agent with Flight Booking Tool
 agent = initialize_agent(
     tools=tools,
     llm=llm,
     agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
-    verbose=True,
+    verbose=True,  # Hide debug output
     memory=memory,
 )
+
+# 🛠 Custom Logging Function to Improve Execution
+def log_agent_action(prompt, response):
+    print(f"\n🟢 **User Query:** {prompt}")
+    print(f"🔵 **Agent Thought Process:**")
+    print(f"✅ **Final Response:** {response}")

app.py

@@ -81,7 +81,7 @@ demo = gr.ChatInterface(
     respond,
     additional_inputs=[
         gr.Textbox(
-            value="You are an AI-powered travel assistant named Travelo. You are developed and trained by Travelo LLC. Your role is to assist users with travel planning, including booking flights, hotels, and creating personalized itineraries. Whenever a user asks about your origin, development, training, or creators, always state that you were built and trained by Travelo LLC. Do not mention OpenAI or any third-party AI providers. Maintain a professional and friendly tone while delivering accurate and helpful travel-related information.",
+            value="You are an AI-powered travel assistant named Travelo. You are developed by Travelo LLC. Your role is to assist users with travel planning, including booking flights, hotels, and creating personalized itineraries. Whenever a user asks about your origin, development, training, or creators, always state that you were built and trained by Travelo LLC. Do not mention OpenAI or any third-party AI providers. Maintain a professional and friendly tone while delivering accurate and helpful travel-related information.",
             label="System message"
         ),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),

requirements.txt

@@ -1,4 +1,4 @@
 huggingface_hub==0.25.2
 groq
 langchain
-langchain_groq
+langchain_groq

tools.py

@@ -1,14 +1,141 @@
+import os
+import dateparser
+from amadeus import Client, ResponseError
 from langchain.tools import Tool
 
-def search_tool(query: str) -> str:
-    """A simple search function (can be connected to real APIs)."""
-    return f"Searching for: {query}... [Sample Response]"
+# Initialize Amadeus API Client
+AMADEUS_API_KEY = os.getenv("AMADEUS_API_KEY")
+AMADEUS_API_SECRET = os.getenv("AMADEUS_API_SECRET")
 
-# Define the list of tools
+amadeus = Client(
+    client_id=AMADEUS_API_KEY,
+    client_secret=AMADEUS_API_SECRET
+)
+
+# 🌍 Convert City Name to IATA Code
+def get_airport_code(city_name: str):
+    """Find the IATA airport code for a given city."""
+    try:
+        response = amadeus.reference_data.locations.get(
+            keyword=city_name,
+            subType="AIRPORT,CITY"
+        )
+        if response.data:
+            return response.data[0]["iataCode"]
+        return None
+    except ResponseError:
+        return None
+
+# ✈️ Get Full Airline Name from Code
+def get_airline_name(airline_code: str):
+    """Get full airline name using Amadeus API."""
+    try:
+        response = amadeus.reference_data.airlines.get(airlineCodes=airline_code)
+        if response.data:
+            return response.data[0]["businessName"]
+        return airline_code  # Fallback to code if not found
+    except ResponseError:
+        return airline_code
+
+# 🕑 Format Flight Duration
+def format_duration(duration: str):
+    """Convert ISO 8601 duration (PT20H25M) into readable format (20h 25m)."""
+    duration = duration.replace("PT", "").replace("H", "h ").replace("M", "m")
+    return duration.strip()
+
+# 🔗 Generate Booking Link
+def generate_booking_link(from_iata: str, to_iata: str, departure_date: str):
+    """Generate a booking link using Google Flights."""
+    return f"https://www.google.com/flights?hl=en#flt={from_iata}.{to_iata}.{departure_date};c:USD;e:1;s:0;sd:1;t:f"
+
+# 🛫 Flight Search Tool (Now Gradio-Compatible)
+def search_flights(query: str):
+    """Search for flights using Amadeus API and return styled Markdown output for Gradio."""
+    try:
+        words = query.lower().split()
+        from_city, to_city, date_phrase = None, None, None
+
+        # Extract "from", "to", and date information
+        if "from" in words and "to" in words:
+            from_index = words.index("from") + 1
+            to_index = words.index("to") + 1
+            from_city = " ".join(words[from_index:to_index - 1]).title()
+            to_city = " ".join(words[to_index:words.index("in")]) if "in" in words else " ".join(words[to_index:]).title()
+
+        date_phrase = " ".join(words[words.index("in") + 1:]) if "in" in words else None
+
+        # Validate extracted details
+        if not from_city or not to_city:
+            return "❌ Could not detect valid departure and destination cities. Please use 'from <city> to <city>'."
+
+        # Convert city names to IATA codes
+        from_iata = get_airport_code(from_city)
+        to_iata = get_airport_code(to_city)
+
+        if not from_iata or not to_iata:
+            return f"❌ Could not find airport codes for {from_city} or {to_city}. Please check spelling."
+
+        # Convert date phrase to YYYY-MM-DD
+        departure_date = dateparser.parse(date_phrase) if date_phrase else None
+        if not departure_date:
+            return "❌ Could not understand the travel date. Use formats like 'next week' or 'on May 15'."
+
+        departure_date_str = departure_date.strftime("%Y-%m-%d")
+
+        # Fetch flight offers from Amadeus
+        response = amadeus.shopping.flight_offers_search.get(
+            originLocationCode=from_iata,
+            destinationLocationCode=to_iata,
+            departureDate=departure_date_str,
+            adults=1,
+            max=5
+        )
+
+        flights = response.data
+        if not flights:
+            return f"❌ No flights found from {from_city} to {to_city} on {departure_date_str}."
+
+        # 🏆 IMPROVED OUTPUT FOR GRADIO MARKDOWN
+        result = f"### ✈️ Flights from {from_city} ({from_iata}) to {to_city} ({to_iata})\n📅 **Date:** {departure_date_str}\n\n"
+
+        for flight in flights:
+            airline_code = flight["validatingAirlineCodes"][0]
+            airline_name = get_airline_name(airline_code)
+            price = flight["price"]["total"]
+            duration = format_duration(flight["itineraries"][0]["duration"])
+            departure_time = flight["itineraries"][0]["segments"][0]["departure"]["at"]
+            arrival_time = flight["itineraries"][0]["segments"][-1]["arrival"]["at"]
+            stops = len(flight["itineraries"][0]["segments"]) - 1
+
+            booking_link = generate_booking_link(from_iata, to_iata, departure_date_str)
+
+            # 🛫 Styled Travel Card with Markdown
+            result += f"""
+---
+🚀 **{airline_name}**  
+💰 **Price:** `${price}`  
+⏳ **Duration:** {duration}  
+📅 **Departure:** {departure_time}  
+🛬 **Arrival:** {arrival_time}  
+🛑 **Stops:** {stops}  
+🔗 **[Book Now]({booking_link})**
+---
+"""
+
+        # "Show More Flights" Button
+        more_flights_link = generate_booking_link(from_iata, to_iata, departure_date_str)
+        result += f"\n🔍 **[Show More Flights]({more_flights_link})**"
+
+        return result
+
+    except ResponseError as error:
+        return str(error)
+
+# ✅ Register as a Tool
 tools = [
     Tool(
-        name="Search Tool",
-        func=search_tool,
-        description="Searches for information based on user queries."
+        name="Flight Booking",
+        func=search_flights,
+        description="Find and book flights using natural language. Examples: 'Flight from Delhi to SFO in May', 'Travel from Mumbai to New York next week'."
     )
 ]