nextchat / ai /trip.py
shoaibamin-dev
audit logs and mongo util
c206733
import requests
import os
from dotenv import load_dotenv
load_dotenv()
def get_trip_planner(days, destination):
try:
# Fetch the API token from environment variables
api_token = os.environ.get("ZYLA_API_TOKEN")
if not api_token:
raise ValueError("API token is required. Set ZYLA_API_TOKEN in environment variables.")
# Validate inputs
if not isinstance(days, int) or days <= 0:
raise ValueError("Days must be a positive integer.")
if not destination.strip():
raise ValueError("Destination cannot be empty.")
# Construct API URL
api_url = f"https://zylalabs.com/api/2242/trip+planner+api/2103/get+planning?days={days}&destination={destination}"
# Make the API request
response = requests.get(api_url, headers={"Authorization": f"Bearer {api_token}"})
response.raise_for_status()
# Parse the response
data = response.json()
if 'error' in data:
raise ValueError(f"API Error: {data['error']}")
trip_plan = data.get("plan", [])
if not trip_plan:
print("No trip plan found.")
return []
formatted_plan = {
"_id": data.get("_id", "Unknown ID"),
"plan": [
{
"day": day_plan.get("day"),
"activities": [
{
"time": activity.get("time"),
"description": activity.get("description")
}
for activity in day_plan.get("activities", [])
]
}
for day_plan in trip_plan
],
"key": data.get("key", "Unknown Key")
}
# Print the formatted output
print("\nFormatted Trip Plan:")
print(formatted_plan)
return formatted_plan
except Exception as e:
print(f"Error occurred: {e}")
return None
if __name__ == "__main__":
try:
# Gather user inputs
days = int(input("Enter the number of days for the trip: ").strip())
destination = input("Enter the destination: ").strip()
# Fetch trip plans
print("\nFetching trip plans...")
trip_plan = get_trip_planner(days, destination)
if not trip_plan:
print("No trip plans found or an error occurred.")
except Exception as e:
print(f"Error occurred: {e}")