modify prompts

gemini_api.py

@@ -11,20 +11,19 @@ genai.configure(api_key=GEMINI_API_KEY)
 model = genai.GenerativeModel('gemini-2.0-flash')
 
 def get_travel_info(user_input):
-    """Uses Gemini AI to extract travel intent and location from user input."""
+    """Uses Gemini AI to extract travel intent, origin, destination, and other travel details from user input."""
 
     prompt = f"""
-      You are a travel assistant. Your job is to extract the travel intent, location, and departure date from a user's query.
-      The intent can be anything from planning a trip to finding a hotel to suggesting a restaurant.
-      The location is the destination of the travel.
-      If the location is not specified, return None.
+      You are a travel assistant. Your job is to extract the travel intent, origin (from where), destination (to where), and departure date from a user's query.
+      If the origin is not mentioned, return null.
       If the intent is not related to travel, return the intent and location in JSON format ONLY, like this:
-      If either the intent, the location, the departure_date, duration, budget, or num_people is not found, return null for that field.
+      If either the intent, the origin, the destination, the departure_date, duration, budget, or num_people is not found, return null for that field.
 
       ```json
       {{
         "intent": "the extracted intent",
-        "location": "the extracted location",
+        "from": "the extracted origin",
+        "to": "the extracted destination",
         "departure_date": "the extracted departure date",
         "duration": "the extracted duration",
         "budget": "the extracted budget",
@@ -32,11 +31,12 @@ def get_travel_info(user_input):
       }}
       ```
 
-      User Query: I want to plan a trip to Paris for 5 days with a budget of $2000 for 2 people, leaving on March 15th.
+      User Query: I want to plan a trip from Los Angeles to Paris for 5 days with a budget of $2000 for 2 people, leaving on March 15th.
       ```json
       {{
         "intent": "plan a trip",
-        "location": "Paris",
+        "from": "Los Angeles",
+        "to": "Paris",
         "departure_date": "March 15th",
         "duration": "5 days",
         "budget": "$2000",
@@ -48,7 +48,8 @@ def get_travel_info(user_input):
       ```json
       {{
         "intent": "Find a hotel",
-        "location": "New York City",
+        "from": null,
+        "to": "New York City",
         "departure_date": "April 10th",
         "duration": "3 nights",
         "budget": "$1500",
@@ -60,7 +61,8 @@ def get_travel_info(user_input):
       ```json
       {{
         "intent": null,
-        "location": null,
+        "from": null,
+        "to": null,
         "departure_date": null,
         "duration": null,
         "budget": null,
@@ -82,14 +84,16 @@ def get_travel_info(user_input):
             json_string = match.group(0).strip()
             data = json.loads(json_string)
             intent = data.get("intent")
-            location = data.get("location")
+            from_location = data.get("from")
+            to_location = data.get("to")
             departure_date = data.get("departure_date")
             duration = data.get("duration")
             budget = data.get("budget")
             num_people = data.get("num_people")
             return {
                 "intent": intent,
-                "location": location,
+                "from": from_location,
+                "to": to_location,
                 "departure_date": departure_date,
                 "duration": duration,
                 "budget": budget,
@@ -106,62 +110,66 @@ def get_travel_info(user_input):
         print(f"An unexpected error occurred: {e}")
         return None
 
-def get_travel_recommendations(intent, location, departure_date, duration, budget, num_people):
-    """Uses Gemini AI to provide travel recommendations based on intent and location."""
+def get_travel_recommendations(intent, from_location, to_location, departure_date, duration, budget, num_people):
+    """Uses Gemini AI to provide travel recommendations based on intent, origin, and destination."""
 
     prompt = f"""
-      You are a travel assistant. Based on the given intent, location, duration, budget, and number of people, provide helpful travel recommendations in Markdown format.
-      Attractions should provide at least 10 recommendations.
-      Accommodations should provide at least 5 recommendations.
-      Activities should provide at least 10 recommendations.
-      Food should provide at least 10 recommendations.
-      Keep in mind the budget and the number of people when suggesting accommodations and activities.
+      You are a travel assistant. Based on the given intent, origin (from where), destination (to where), duration, budget, and number of people, provide helpful travel recommendations in Markdown format.
+      - Provide some common websites for booking flights as well as the like to the web.
+      - List at least 2 of the cheapest flight options.
+      - Provide at least 10 recommended attractions.
+      - Provide at least 5 accommodations.
+      - Provide at least 10 activities.
+      - Provide at least 10 food recommendations.
+      - Keep in mind the budget and number of people when suggesting accommodations and activities.
+      - Use NTD as the currency
 
       Intent: {intent}
-      Location: {location}
+      From: {from_location}
+      To: {to_location}
       Departure Date: {departure_date}
       Duration: {duration}
       Budget: {budget}
-      Number of people: {num_people}
+      Number of People: {num_people}
 
       Your response should look like this:
 
-      # 🌍 Travel Guide: {location}
+      # 🌍 Travel Guide: {to_location}
 
       ## 🌤️ Weather:
-      - Weather 1
+      - Weather details for {to_location}
 
-      ## ✈️ Flight:
-      - Flight 1
-      - Flight 2
-
-      ## 🏨 Accommodations:
-      - Hotel 1
-      - Hotel 2
+      ## ✈️ Flights from {from_location} to {to_location}:
+      - 
+      - 
 
       ## 💳 Mobile Payment:
-      - Mobile Payment 1
-      - Mobile Payment 2
+      - 
+      - 
 
       ## 🚗 Local Transportations:
-      - Local Transportations 1
-      - Local Transportations 2
+      - 
+      - 
+
+      ## 🏨 Accommodations:
+      - 
+      - 
 
       ## 🏰 Attractions:
-      - Attraction 1
-      - Attraction 2
+      - 
+      - 
 
       ## 🚀 Activities:
-      - Activity 1
-      - Activity 2
+      - 
+      - 
 
-      ## 🍽️ Foods:
-      - Foods 1
-      - Foods 2
+      ## 🍽️ Foods to Try:
+      - 
+      - 
 
       ## 💡 Tips for Planning:
-      - Tips for Planning 1
-      - Tips for Planning 2
+      - 
+      - 
     """
 
     response = model.generate_content(prompt)