Spaces:
Running
Running
Rename travel.py to paper.py
Browse files
paper.py
ADDED
@@ -0,0 +1,199 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import os
|
2 |
+
import json
|
3 |
+
import logging
|
4 |
+
from datetime import datetime, timedelta
|
5 |
+
from langchain_google_genai import ChatGoogleGenerativeAI
|
6 |
+
from langchain.schema import SystemMessage, HumanMessage
|
7 |
+
|
8 |
+
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
|
9 |
+
|
10 |
+
class Agent:
|
11 |
+
def __init__(self, role: str, goal: str, backstory: str, personality: str = "", llm=None) -> None:
|
12 |
+
"""
|
13 |
+
Initialize an Agent with role, goal, backstory, personality, and assigned LLM.
|
14 |
+
"""
|
15 |
+
self.role = role
|
16 |
+
self.goal = goal
|
17 |
+
self.backstory = backstory
|
18 |
+
self.personality = personality
|
19 |
+
self.tools = [] # Initialize with empty list for future tool integrations
|
20 |
+
self.llm = llm
|
21 |
+
|
22 |
+
class Task:
|
23 |
+
def __init__(self, description: str, agent: Agent, expected_output: str, context=None) -> None:
|
24 |
+
"""
|
25 |
+
Initialize a Task with its description, the responsible agent, expected output, and optional context.
|
26 |
+
"""
|
27 |
+
self.description = description
|
28 |
+
self.agent = agent
|
29 |
+
self.expected_output = expected_output
|
30 |
+
self.context = context or []
|
31 |
+
|
32 |
+
google_api_key = os.getenv("GEMINI_API_KEY") # 실제 Google API 키 사용
|
33 |
+
if not google_api_key:
|
34 |
+
logging.error("GEMINI_API_KEY is not set in the environment variables.")
|
35 |
+
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", google_api_key=google_api_key)
|
36 |
+
|
37 |
+
# -------------------------------------------------------------------------------
|
38 |
+
# Define Academic Research Agents
|
39 |
+
# -------------------------------------------------------------------------------
|
40 |
+
literature_research_agent = Agent(
|
41 |
+
role="Literature Research Agent",
|
42 |
+
goal="Research and provide a comprehensive review of existing literature on the research topic.",
|
43 |
+
backstory="An experienced academic researcher specialized in literature reviews and meta-analyses.",
|
44 |
+
personality="Analytical, thorough, and detail-oriented.",
|
45 |
+
llm=llm,
|
46 |
+
)
|
47 |
+
|
48 |
+
outline_agent = Agent(
|
49 |
+
role="Outline Agent",
|
50 |
+
goal="Generate a structured and detailed outline for a research paper based on the research topic and literature.",
|
51 |
+
backstory="A methodical academic planner who organizes research findings into coherent paper structures.",
|
52 |
+
personality="Organized, systematic, and insightful.",
|
53 |
+
llm=llm,
|
54 |
+
)
|
55 |
+
|
56 |
+
draft_writing_agent = Agent(
|
57 |
+
role="Draft Writing Agent",
|
58 |
+
goal="Compose a first draft of the research paper based on the literature review and outline.",
|
59 |
+
backstory="A skilled academic writer capable of synthesizing research findings into well-structured drafts.",
|
60 |
+
personality="Articulate, precise, and scholarly.",
|
61 |
+
llm=llm,
|
62 |
+
)
|
63 |
+
|
64 |
+
citation_agent = Agent(
|
65 |
+
role="Citation Agent",
|
66 |
+
goal="Generate a list of relevant citations and references in the required format for the research paper.",
|
67 |
+
backstory="A detail-oriented bibliographic expert with extensive knowledge of citation standards.",
|
68 |
+
personality="Meticulous, accurate, and research-savvy.",
|
69 |
+
llm=llm,
|
70 |
+
)
|
71 |
+
|
72 |
+
editing_agent = Agent(
|
73 |
+
role="Editing Agent",
|
74 |
+
goal="Revise and polish the draft for clarity, coherence, and academic tone.",
|
75 |
+
backstory="An expert editor skilled in improving academic manuscripts and ensuring high-quality presentation.",
|
76 |
+
personality="Critical, precise, and supportive.",
|
77 |
+
llm=llm,
|
78 |
+
)
|
79 |
+
|
80 |
+
chatbot_agent = Agent(
|
81 |
+
role="Chatbot Agent",
|
82 |
+
goal="Engage in interactive conversation to answer queries related to the academic research process.",
|
83 |
+
backstory="A conversational AI assistant with extensive knowledge in academia and research methodologies.",
|
84 |
+
personality="Helpful, conversational, and knowledgeable.",
|
85 |
+
llm=llm,
|
86 |
+
)
|
87 |
+
|
88 |
+
# -------------------------------------------------------------------------------
|
89 |
+
# Define Tasks for Academic Research and Writing
|
90 |
+
# -------------------------------------------------------------------------------
|
91 |
+
literature_research_task = Task(
|
92 |
+
description="""Research academic literature on {topic} considering the keywords {keywords}.
|
93 |
+
|
94 |
+
Please provide:
|
95 |
+
- A summary of the current state of research,
|
96 |
+
- Key trends and gaps in the literature,
|
97 |
+
- Notable studies and their findings,
|
98 |
+
- Relevant theoretical frameworks and methodologies.
|
99 |
+
Format the response with bullet points and concise summaries.""",
|
100 |
+
agent=literature_research_agent,
|
101 |
+
expected_output="""A comprehensive literature review summary covering:
|
102 |
+
1. Summary of current research trends
|
103 |
+
2. Identification of gaps and controversies
|
104 |
+
3. Key studies with brief descriptions
|
105 |
+
4. Theoretical frameworks and methodologies used"""
|
106 |
+
)
|
107 |
+
|
108 |
+
outline_task = Task(
|
109 |
+
description="""Based on the research topic {topic} and literature review findings, generate a detailed outline for a research paper.
|
110 |
+
|
111 |
+
Include sections such as:
|
112 |
+
- Abstract
|
113 |
+
- Introduction (including research questions and objectives)
|
114 |
+
- Literature Review
|
115 |
+
- Methodology
|
116 |
+
- Results/Findings
|
117 |
+
- Discussion
|
118 |
+
- Conclusion
|
119 |
+
- References
|
120 |
+
Format the outline in a structured manner with bullet points and subheadings.""",
|
121 |
+
agent=outline_agent,
|
122 |
+
expected_output="A structured outline for a research paper including all major sections and key points to cover in each section."
|
123 |
+
)
|
124 |
+
|
125 |
+
draft_writing_task = Task(
|
126 |
+
description="""Using the research topic {topic}, the literature review, and the generated outline, compose a first draft of the research paper.
|
127 |
+
|
128 |
+
The draft should include:
|
129 |
+
- A coherent narrative flow,
|
130 |
+
- Detailed sections as per the outline,
|
131 |
+
- Integration of key findings from the literature review.
|
132 |
+
Ensure the tone is academic and the content is well-organized.""",
|
133 |
+
agent=draft_writing_agent,
|
134 |
+
expected_output="A complete first draft of the research paper covering all sections with sufficient academic detail."
|
135 |
+
)
|
136 |
+
|
137 |
+
citation_task = Task(
|
138 |
+
description="""Based on the literature review for {topic}, generate a list of key references and citations in APA format.
|
139 |
+
|
140 |
+
Include:
|
141 |
+
- Author names, publication year, title, and source,
|
142 |
+
- At least 10 key references relevant to the research topic.
|
143 |
+
Format the output as a numbered list of citations.""",
|
144 |
+
agent=citation_agent,
|
145 |
+
expected_output="A list of 10+ relevant citations in APA format."
|
146 |
+
)
|
147 |
+
|
148 |
+
editing_task = Task(
|
149 |
+
description="""Review and edit the draft for clarity, coherence, and academic tone.
|
150 |
+
|
151 |
+
Focus on:
|
152 |
+
- Improving sentence structure,
|
153 |
+
- Ensuring logical flow between sections,
|
154 |
+
- Correcting grammar and stylistic issues,
|
155 |
+
- Enhancing academic tone.
|
156 |
+
Provide the polished version of the paper.""",
|
157 |
+
agent=editing_agent,
|
158 |
+
expected_output="A refined and polished version of the research paper draft."
|
159 |
+
)
|
160 |
+
|
161 |
+
chatbot_task = Task(
|
162 |
+
description="Provide a conversational and detailed response to academic research-related queries.",
|
163 |
+
agent=chatbot_agent,
|
164 |
+
expected_output="A friendly, informative response addressing the query."
|
165 |
+
)
|
166 |
+
|
167 |
+
def run_task(task: Task, input_text: str) -> str:
|
168 |
+
"""
|
169 |
+
Executes the given task using the associated agent's LLM and returns the response content.
|
170 |
+
"""
|
171 |
+
try:
|
172 |
+
if not isinstance(task, Task):
|
173 |
+
raise ValueError(f"Expected 'task' to be an instance of Task, got {type(task)}")
|
174 |
+
if not hasattr(task, 'agent') or not isinstance(task.agent, Agent):
|
175 |
+
raise ValueError("Task must have a valid 'agent' attribute of type Agent.")
|
176 |
+
system_input = (
|
177 |
+
f"Agent Details:\n"
|
178 |
+
f"Role: {task.agent.role}\n"
|
179 |
+
f"Goal: {task.agent.goal}\n"
|
180 |
+
f"Backstory: {task.agent.backstory}\n"
|
181 |
+
f"Personality: {task.agent.personality}\n"
|
182 |
+
)
|
183 |
+
task_input = (
|
184 |
+
f"Task Details:\n"
|
185 |
+
f"Task Description: {task.description}\n"
|
186 |
+
f"Expected Output: {task.expected_output}\n"
|
187 |
+
f"Input for Task:\n{input_text}\n"
|
188 |
+
)
|
189 |
+
messages = [
|
190 |
+
SystemMessage(content=system_input),
|
191 |
+
HumanMessage(content=task_input)
|
192 |
+
]
|
193 |
+
response = task.agent.llm.invoke(messages)
|
194 |
+
if not response or not response.content:
|
195 |
+
raise ValueError("Empty response from LLM.")
|
196 |
+
return response.content
|
197 |
+
except Exception as e:
|
198 |
+
logging.error(f"Error in task '{task.agent.role}': {e}")
|
199 |
+
return f"Error in {task.agent.role}: {e}"
|
travel.py
DELETED
@@ -1,453 +0,0 @@
|
|
1 |
-
import os
|
2 |
-
import json
|
3 |
-
import logging
|
4 |
-
from datetime import datetime, timedelta
|
5 |
-
from langchain_google_genai import ChatGoogleGenerativeAI
|
6 |
-
from langchain.schema import SystemMessage, HumanMessage
|
7 |
-
|
8 |
-
# Setup logging configuration
|
9 |
-
logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
|
10 |
-
|
11 |
-
# -------------------------------------------------------------------------------
|
12 |
-
# Agent and Task Classes with Type Hints and Docstrings
|
13 |
-
# -------------------------------------------------------------------------------
|
14 |
-
class Agent:
|
15 |
-
def __init__(self, role: str, goal: str, backstory: str, personality: str = "", llm=None) -> None:
|
16 |
-
"""
|
17 |
-
Initialize an Agent with role, goal, backstory, personality, and assigned LLM.
|
18 |
-
"""
|
19 |
-
self.role = role
|
20 |
-
self.goal = goal
|
21 |
-
self.backstory = backstory
|
22 |
-
self.personality = personality
|
23 |
-
self.tools = [] # Initialize with empty list for future tool integrations
|
24 |
-
self.llm = llm
|
25 |
-
|
26 |
-
class Task:
|
27 |
-
def __init__(self, description: str, agent: Agent, expected_output: str, context=None) -> None:
|
28 |
-
"""
|
29 |
-
Initialize a Task with its description, the responsible agent, expected output, and optional context.
|
30 |
-
"""
|
31 |
-
self.description = description
|
32 |
-
self.agent = agent
|
33 |
-
self.expected_output = expected_output
|
34 |
-
self.context = context or []
|
35 |
-
|
36 |
-
# -------------------------------------------------------------------------------
|
37 |
-
# Initialize LLM
|
38 |
-
# -------------------------------------------------------------------------------
|
39 |
-
google_api_key = os.getenv("GEMINI_API_KEY") # 실제 Google API 키 사용
|
40 |
-
if not google_api_key:
|
41 |
-
logging.error("GEMINI_API_KEY is not set in the environment variables.")
|
42 |
-
llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", google_api_key=google_api_key)
|
43 |
-
|
44 |
-
# -------------------------------------------------------------------------------
|
45 |
-
# Define Travel Agents
|
46 |
-
# -------------------------------------------------------------------------------
|
47 |
-
destination_research_agent = Agent(
|
48 |
-
role="Destination Research Agent",
|
49 |
-
goal=(
|
50 |
-
"Research and provide comprehensive information about the destination including popular attractions, "
|
51 |
-
"local culture, weather patterns, best times to visit, and local transportation options."
|
52 |
-
),
|
53 |
-
backstory=(
|
54 |
-
"An experienced travel researcher with extensive knowledge of global destinations. "
|
55 |
-
"I specialize in uncovering both popular attractions and hidden gems that match travelers' interests."
|
56 |
-
),
|
57 |
-
personality="Curious, detail-oriented, and knowledgeable about global cultures and travel trends.",
|
58 |
-
llm=llm,
|
59 |
-
)
|
60 |
-
|
61 |
-
accommodation_agent = Agent(
|
62 |
-
role="Accommodation Agent",
|
63 |
-
goal="Find and recommend suitable accommodations based on the traveler's preferences, budget, and location requirements.",
|
64 |
-
backstory="A hospitality expert who understands different types of accommodations and can match travelers with their ideal places to stay.",
|
65 |
-
personality="Attentive, resourceful, and focused on comfort and value.",
|
66 |
-
llm=llm,
|
67 |
-
)
|
68 |
-
|
69 |
-
transportation_agent = Agent(
|
70 |
-
role="Transportation Agent",
|
71 |
-
goal="Plan efficient transportation between the origin, destination, and all points of interest in the itinerary.",
|
72 |
-
backstory="A logistics specialist with knowledge of global transportation systems, from flights to local transit options.",
|
73 |
-
personality="Efficient, practical, and detail-oriented.",
|
74 |
-
llm=llm,
|
75 |
-
)
|
76 |
-
|
77 |
-
activities_agent = Agent(
|
78 |
-
role="Activities & Attractions Agent",
|
79 |
-
goal="Curate personalized activities and attractions that align with the traveler's interests, preferences, and time constraints.",
|
80 |
-
backstory="An enthusiastic explorer who has experienced diverse activities around the world and knows how to match experiences to individual preferences.",
|
81 |
-
personality="Enthusiastic, creative, and personable.",
|
82 |
-
llm=llm,
|
83 |
-
)
|
84 |
-
|
85 |
-
dining_agent = Agent(
|
86 |
-
role="Dining & Culinary Agent",
|
87 |
-
goal="Recommend dining experiences that showcase local cuisine while accommodating dietary preferences and budget considerations.",
|
88 |
-
backstory="A culinary expert with knowledge of global food scenes and an appreciation for authentic local dining experiences.",
|
89 |
-
personality="Passionate about food, culturally aware, and attentive to preferences.",
|
90 |
-
llm=llm,
|
91 |
-
)
|
92 |
-
|
93 |
-
itinerary_agent = Agent(
|
94 |
-
role="Itinerary Integration Agent",
|
95 |
-
goal="Compile all recommendations into a cohesive, day-by-day itinerary that optimizes time, minimizes travel fatigue, and maximizes enjoyment.",
|
96 |
-
backstory="A master travel planner who understands how to balance activities, rest, and logistics to create the perfect travel experience.",
|
97 |
-
personality="Organized, balanced, and practical.",
|
98 |
-
llm=llm,
|
99 |
-
)
|
100 |
-
|
101 |
-
# -------------------------------------------------------------------------------
|
102 |
-
# Define Chatbot Agent and Task for Interactive Conversation
|
103 |
-
# -------------------------------------------------------------------------------
|
104 |
-
chatbot_agent = Agent(
|
105 |
-
role="Chatbot Agent",
|
106 |
-
goal="Engage in interactive conversation to answer travel-related queries.",
|
107 |
-
backstory="A conversational AI assistant who provides instant, accurate travel information and recommendations.",
|
108 |
-
personality="Friendly, conversational, and knowledgeable about travel.",
|
109 |
-
llm=llm,
|
110 |
-
)
|
111 |
-
|
112 |
-
chatbot_task = Task(
|
113 |
-
description="Provide a conversational and detailed response to travel-related queries.",
|
114 |
-
agent=chatbot_agent,
|
115 |
-
expected_output="A friendly, helpful response to the user's query."
|
116 |
-
)
|
117 |
-
|
118 |
-
# -------------------------------------------------------------------------------
|
119 |
-
# Define Other Travel Tasks
|
120 |
-
# -------------------------------------------------------------------------------
|
121 |
-
destination_research_task = Task(
|
122 |
-
description="""Research {destination} thoroughly, considering the traveler's interests in {preferences}.
|
123 |
-
|
124 |
-
Efficient research parameters:
|
125 |
-
- Prioritize research in these critical categories:
|
126 |
-
* Top attractions that match specific {preferences} (not generic lists)
|
127 |
-
* Local transportation systems with cost-efficiency analysis
|
128 |
-
* Neighborhood breakdown with accommodation recommendations by budget tier
|
129 |
-
* Seasonal considerations for the specific travel dates
|
130 |
-
* Safety assessment with specific areas to embrace or avoid
|
131 |
-
* Cultural norms that impact visitor experience (dress codes, tipping, etiquette)
|
132 |
-
|
133 |
-
- Apply efficiency filters:
|
134 |
-
* Focus exclusively on verified information from official tourism boards, recent travel guides, and reliable local sources
|
135 |
-
* Analyze recent visitor reviews (< 6 months old) to identify changing conditions
|
136 |
-
* Evaluate price-to-experience value for attractions instead of just popularity
|
137 |
-
* Identify logistical clusters where multiple interests can be satisfied efficiently
|
138 |
-
* Research off-peak times for popular attractions to minimize waiting
|
139 |
-
* Evaluate digital tools (apps, passes, reservation systems) that streamline the visit
|
140 |
-
|
141 |
-
- Create practical knowledge matrices:
|
142 |
-
* Transportation method comparison (cost vs. time vs. convenience)
|
143 |
-
* Weather impact on specific activities
|
144 |
-
* Budget allocation recommendations based on preference priorities
|
145 |
-
* Time-saving opportunity identification""",
|
146 |
-
agent=destination_research_agent,
|
147 |
-
expected_output="""Targeted destination brief containing:
|
148 |
-
1. Executive summary highlighting the 5 most relevant aspects based on {preferences}
|
149 |
-
2. Neighborhood analysis with accommodation recommendations mapped to specific interests
|
150 |
-
3. Transportation efficiency guide with cost/convenience matrix
|
151 |
-
4. Cultural briefing focusing only on need-to-know information that impacts daily activities
|
152 |
-
5. Seasonal advantages and challenges specific to travel dates
|
153 |
-
6. Digital resource toolkit (essential apps, websites, reservation systems)
|
154 |
-
7. Budget optimization strategies with price ranges for key experiences
|
155 |
-
8. Safety and health quick-reference including emergency contacts
|
156 |
-
9. Logistics efficiency map showing optimal activity clustering
|
157 |
-
10. Local insider advantage recommendations that save time or money
|
158 |
-
|
159 |
-
Format should prioritize scannable information with bullet points, comparison tables, and decision matrices rather than lengthy prose."""
|
160 |
-
)
|
161 |
-
|
162 |
-
accommodation_task = Task(
|
163 |
-
description="Find suitable accommodations in {destination} based on a {budget} budget and preferences for {preferences}.",
|
164 |
-
agent=accommodation_agent,
|
165 |
-
expected_output="List of recommended accommodations with details on location, amenities, price range, and availability."
|
166 |
-
)
|
167 |
-
|
168 |
-
transportation_task = Task(
|
169 |
-
description="Plan transportation from {origin} to {destination} and local transportation options during the stay.",
|
170 |
-
agent=transportation_agent,
|
171 |
-
expected_output="Transportation plan including flights/routes to the destination and recommendations for getting around locally."
|
172 |
-
)
|
173 |
-
|
174 |
-
activities_task = Task(
|
175 |
-
description="""Suggest activities and attractions in {destination} that align with interests in {preferences}.
|
176 |
-
|
177 |
-
Detailed requirements:
|
178 |
-
- Categorize activities into: Cultural Experiences, Outdoor Adventures, Culinary Experiences,
|
179 |
-
Entertainment & Nightlife, Family-Friendly Activities, and Local Hidden Gems
|
180 |
-
- For each activity, include:
|
181 |
-
* Detailed description with historical/cultural context where relevant
|
182 |
-
* Precise location with neighborhood information
|
183 |
-
* Operating hours with seasonal variations noted
|
184 |
-
* Pricing information with different ticket options/packages
|
185 |
-
* Accessibility considerations for travelers with mobility limitations
|
186 |
-
* Recommended duration for the activity (minimum and ideal time)
|
187 |
-
* Best time of day/week/year to visit
|
188 |
-
* Crowd levels by season
|
189 |
-
* Photography opportunities and restrictions
|
190 |
-
* Required reservations or booking windows
|
191 |
-
- Include a mix of iconic must-see attractions and off-the-beaten-path experiences
|
192 |
-
- Consider weather patterns in {destination} during travel period
|
193 |
-
- Analyze the {preferences} to match specific personality types and interest levels
|
194 |
-
- Include at least 2-3 rainy day alternatives for outdoor activities
|
195 |
-
- Provide local transportation options to reach each attraction
|
196 |
-
- Note authentic local experiences that provide cultural immersion
|
197 |
-
- Flag any activities requiring special equipment, permits, or physical fitness levels""",
|
198 |
-
agent=activities_agent,
|
199 |
-
expected_output="""Comprehensive curated list of activities and attractions with:
|
200 |
-
1. Clear categorization by type (cultural, outdoor, culinary, entertainment, family-friendly, hidden gems)
|
201 |
-
2. Detailed descriptions that include historical and cultural context
|
202 |
-
3. Complete practical information (hours, pricing, location, accessibility)
|
203 |
-
4. Time optimization recommendations (best time to visit, how to avoid crowds)
|
204 |
-
5. Personalized matches explaining why each activity aligns with specific {preferences}
|
205 |
-
6. Local transportation details to reach each attraction
|
206 |
-
7. Alternative options for inclement weather or unexpected closures
|
207 |
-
8. Insider tips from locals that enhance the experience
|
208 |
-
9. Suggested combinations of nearby activities for efficient itinerary planning
|
209 |
-
10. Risk level assessment and safety considerations where applicable
|
210 |
-
11. Sustainability impact and responsible tourism notes
|
211 |
-
12. Photographic highlights and optimal viewing points
|
212 |
-
|
213 |
-
Format should include a summary table for quick reference followed by detailed cards for each activity."""
|
214 |
-
)
|
215 |
-
|
216 |
-
dining_task = Task(
|
217 |
-
description="Recommend dining experiences in {destination} that showcase local cuisine while considering {preferences}.",
|
218 |
-
agent=dining_agent,
|
219 |
-
expected_output="List of recommended restaurants and food experiences with cuisine types, price ranges, and special notes."
|
220 |
-
)
|
221 |
-
|
222 |
-
itinerary_task = Task(
|
223 |
-
description="""Create a day-by-day itinerary for a {duration} trip to {destination} from {origin}, incorporating all recommendations.
|
224 |
-
|
225 |
-
Detailed requirements:
|
226 |
-
- Begin with arrival logistics including airport transfer options, check-in times, and first-day orientation activities
|
227 |
-
- Structure each day with:
|
228 |
-
* Morning, afternoon, and evening activity blocks with precise timing
|
229 |
-
* Estimated travel times between locations using various transportation methods
|
230 |
-
* Buffer time for rest, spontaneous exploration, and unexpected delays
|
231 |
-
* Meal recommendations with reservation details and backup options
|
232 |
-
* Sunset/sunrise opportunities for optimal photography or experiences
|
233 |
-
- Apply intelligent sequencing to:
|
234 |
-
* Group attractions by geographic proximity to minimize transit time
|
235 |
-
* Schedule indoor activities strategically for predicted weather patterns
|
236 |
-
* Balance high-energy activities with relaxation periods
|
237 |
-
* Alternate between cultural immersion and entertainment experiences
|
238 |
-
* Account for opening days/hours of attractions and potential closures
|
239 |
-
- Include practical timing considerations:
|
240 |
-
* Museum/attraction fatigue limitations
|
241 |
-
* Jet lag recovery for first 1-2 days
|
242 |
-
* Time zone adjustment strategies
|
243 |
-
* Local rush hours and traffic patterns to avoid
|
244 |
-
* Cultural norms for meal times and business hours
|
245 |
-
- End with departure logistics including check-out procedures, airport transfer timing, and luggage considerations
|
246 |
-
- Add specialized planning elements:
|
247 |
-
* Local festivals or events coinciding with the travel dates
|
248 |
-
* Free time blocks for personal exploration or shopping
|
249 |
-
* Contingency recommendations for weather disruptions
|
250 |
-
* Early booking requirements for popular attractions/restaurants
|
251 |
-
* Local emergency contacts and nearby medical facilities""",
|
252 |
-
agent=itinerary_agent,
|
253 |
-
expected_output="""Comprehensive day-by-day itinerary featuring:
|
254 |
-
1. Detailed timeline for each day with hour-by-hour scheduling and transit times
|
255 |
-
2. Color-coded activity blocks that visually distinguish between types of activities
|
256 |
-
3. Intelligent geographic clustering to minimize transportation time
|
257 |
-
4. Strategic meal placements with both reservation-required and casual options
|
258 |
-
5. Built-in flexibility with free time blocks and alternative suggestions
|
259 |
-
6. Weather-adaptive scheduling with indoor/outdoor activity balance
|
260 |
-
7. Energy level considerations throughout the trip arc
|
261 |
-
8. Cultural timing adaptations (accommodating local siesta times, religious observances, etc.)
|
262 |
-
9. Practical logistical details (bag storage options, dress code reminders, etc.)
|
263 |
-
10. Local transportation guidance including transit cards, apps, and pre-booking requirements
|
264 |
-
11. Visual map representation showing daily movement patterns
|
265 |
-
12. Key phrases in local language for each day's activities
|
266 |
-
|
267 |
-
Format should include both a condensed overview calendar and detailed daily breakdowns with time, activity, location, notes, and contingency plans."""
|
268 |
-
)
|
269 |
-
|
270 |
-
# -------------------------------------------------------------------------------
|
271 |
-
# Helper Function to Run a Task with Full Agent & Task Information
|
272 |
-
# -------------------------------------------------------------------------------
|
273 |
-
def run_task(task: Task, input_text: str) -> str:
|
274 |
-
"""
|
275 |
-
Executes the given task using the associated agent's LLM and returns the response content.
|
276 |
-
"""
|
277 |
-
try:
|
278 |
-
if not isinstance(task, Task):
|
279 |
-
raise ValueError(f"Expected 'task' to be an instance of Task, got {type(task)}")
|
280 |
-
if not hasattr(task, 'agent') or not isinstance(task.agent, Agent):
|
281 |
-
raise ValueError("Task must have a valid 'agent' attribute of type Agent.")
|
282 |
-
|
283 |
-
system_input = (
|
284 |
-
f"Agent Details:\n"
|
285 |
-
f"Role: {task.agent.role}\n"
|
286 |
-
f"Goal: {task.agent.goal}\n"
|
287 |
-
f"Backstory: {task.agent.backstory}\n"
|
288 |
-
f"Personality: {task.agent.personality}\n"
|
289 |
-
)
|
290 |
-
task_input = (
|
291 |
-
f"Task Details:\n"
|
292 |
-
f"Task Description: {task.description}\n"
|
293 |
-
f"Expected Output: {task.expected_output}\n"
|
294 |
-
f"Input for Task:\n{input_text}\n"
|
295 |
-
)
|
296 |
-
messages = [
|
297 |
-
SystemMessage(content=system_input),
|
298 |
-
HumanMessage(content=task_input)
|
299 |
-
]
|
300 |
-
response = task.agent.llm.invoke(messages)
|
301 |
-
if not response or not response.content:
|
302 |
-
raise ValueError("Empty response from LLM.")
|
303 |
-
return response.content
|
304 |
-
except Exception as e:
|
305 |
-
logging.error(f"Error in task '{task.agent.role}': {e}")
|
306 |
-
return f"Error in {task.agent.role}: {e}"
|
307 |
-
|
308 |
-
# -------------------------------------------------------------------------------
|
309 |
-
# User Input Functions
|
310 |
-
# -------------------------------------------------------------------------------
|
311 |
-
def get_user_input() -> dict:
|
312 |
-
"""
|
313 |
-
Collects user input for travel itinerary generation.
|
314 |
-
"""
|
315 |
-
print("\n=== Travel Itinerary Generator ===\n")
|
316 |
-
origin = input("Enter your origin city/country: ")
|
317 |
-
destination = input("Enter your destination city/country: ")
|
318 |
-
duration = input("Enter trip duration (number of days): ")
|
319 |
-
budget = input("Enter your budget level (budget, moderate, luxury): ")
|
320 |
-
|
321 |
-
print("\nEnter your travel preferences and interests (comma-separated):")
|
322 |
-
print("Examples: museums, hiking, food, shopping, beaches, history, nightlife, family-friendly, etc.")
|
323 |
-
preferences = input("> ")
|
324 |
-
|
325 |
-
special_requirements = input("\nAny special requirements or notes (dietary restrictions, accessibility needs, etc.)? ")
|
326 |
-
|
327 |
-
return {
|
328 |
-
"origin": origin,
|
329 |
-
"destination": destination,
|
330 |
-
"duration": duration,
|
331 |
-
"budget": budget,
|
332 |
-
"preferences": preferences,
|
333 |
-
"special_requirements": special_requirements
|
334 |
-
}
|
335 |
-
|
336 |
-
# -------------------------------------------------------------------------------
|
337 |
-
# Main Function to Generate Travel Itinerary
|
338 |
-
# -------------------------------------------------------------------------------
|
339 |
-
def generate_travel_itinerary(user_input: dict) -> str:
|
340 |
-
"""
|
341 |
-
Generates a personalized travel itinerary by sequentially running defined tasks.
|
342 |
-
"""
|
343 |
-
print("\nGenerating your personalized travel itinerary...\n")
|
344 |
-
|
345 |
-
# Create input context using f-string formatting
|
346 |
-
input_context = (
|
347 |
-
f"Travel Request Details:\n"
|
348 |
-
f"Origin: {user_input['origin']}\n"
|
349 |
-
f"Destination: {user_input['destination']}\n"
|
350 |
-
f"Duration: {user_input['duration']} days\n"
|
351 |
-
f"Budget Level: {user_input['budget']}\n"
|
352 |
-
f"Preferences/Interests: {user_input['preferences']}\n"
|
353 |
-
f"Special Requirements: {user_input['special_requirements']}\n"
|
354 |
-
)
|
355 |
-
|
356 |
-
# Step 1: Destination Research
|
357 |
-
print("Researching your destination...")
|
358 |
-
destination_info = run_task(destination_research_task, input_context)
|
359 |
-
print("✓ Destination research completed")
|
360 |
-
|
361 |
-
# Step 2: Accommodation Recommendations
|
362 |
-
print("Finding ideal accommodations...")
|
363 |
-
accommodation_info = run_task(accommodation_task, input_context)
|
364 |
-
print("✓ Accommodation recommendations completed")
|
365 |
-
|
366 |
-
# Step 3: Transportation Planning
|
367 |
-
print("Planning transportation...")
|
368 |
-
transportation_info = run_task(transportation_task, input_context)
|
369 |
-
print("✓ Transportation planning completed")
|
370 |
-
|
371 |
-
# Step 4: Activities & Attractions
|
372 |
-
print("Curating activities and attractions...")
|
373 |
-
activities_info = run_task(activities_task, input_context)
|
374 |
-
print("✓ Activities and attractions curated")
|
375 |
-
|
376 |
-
# Step 5: Dining Recommendations
|
377 |
-
print("Finding dining experiences...")
|
378 |
-
dining_info = run_task(dining_task, input_context)
|
379 |
-
print("✓ Dining recommendations completed")
|
380 |
-
|
381 |
-
# Step 6: Create Day-by-Day Itinerary
|
382 |
-
print("Creating your day-by-day itinerary...")
|
383 |
-
combined_info = (
|
384 |
-
input_context + "\n"
|
385 |
-
"Destination Information:\n" + destination_info + "\n"
|
386 |
-
"Accommodation Options:\n" + accommodation_info + "\n"
|
387 |
-
"Transportation Plan:\n" + transportation_info + "\n"
|
388 |
-
"Recommended Activities:\n" + activities_info + "\n"
|
389 |
-
"Dining Recommendations:\n" + dining_info + "\n"
|
390 |
-
)
|
391 |
-
itinerary = run_task(itinerary_task, combined_info)
|
392 |
-
print("✓ Itinerary creation completed")
|
393 |
-
print("✓ Itinerary generation completed")
|
394 |
-
|
395 |
-
return itinerary
|
396 |
-
|
397 |
-
# -------------------------------------------------------------------------------
|
398 |
-
# Save Itinerary to File
|
399 |
-
# -------------------------------------------------------------------------------
|
400 |
-
def save_itinerary_to_file(itinerary: str, user_input: dict, output_dir: str = None) -> str:
|
401 |
-
"""
|
402 |
-
Saves the generated itinerary to a text file and returns the filepath.
|
403 |
-
"""
|
404 |
-
date_str = datetime.now().strftime("%Y-%m-%d")
|
405 |
-
filename = f"{user_input['destination'].replace(' ', '_')}_{date_str}_itinerary.txt"
|
406 |
-
|
407 |
-
if output_dir:
|
408 |
-
if not os.path.exists(output_dir):
|
409 |
-
try:
|
410 |
-
os.makedirs(output_dir)
|
411 |
-
logging.info(f"Created output directory: {output_dir}")
|
412 |
-
except Exception as e:
|
413 |
-
logging.error(f"Error creating directory {output_dir}: {e}")
|
414 |
-
return ""
|
415 |
-
filepath = os.path.join(output_dir, filename)
|
416 |
-
else:
|
417 |
-
filepath = filename
|
418 |
-
|
419 |
-
try:
|
420 |
-
with open(filepath, "w", encoding="utf-8") as f:
|
421 |
-
f.write(itinerary)
|
422 |
-
logging.info(f"Your itinerary has been saved as: {filepath}")
|
423 |
-
return filepath
|
424 |
-
except Exception as e:
|
425 |
-
logging.error(f"Error saving itinerary: {e}")
|
426 |
-
return ""
|
427 |
-
|
428 |
-
# -------------------------------------------------------------------------------
|
429 |
-
# Main Function
|
430 |
-
# -------------------------------------------------------------------------------
|
431 |
-
def main() -> None:
|
432 |
-
"""
|
433 |
-
Main entry point for the travel itinerary generator application.
|
434 |
-
"""
|
435 |
-
print("Welcome to BlockX Travel Itinerary Generator!")
|
436 |
-
print("This AI-powered tool will create a personalized travel itinerary based on your preferences.")
|
437 |
-
|
438 |
-
user_input = get_user_input()
|
439 |
-
|
440 |
-
print("\nWhere would you like to save the itinerary?")
|
441 |
-
print("Press Enter to save in the current directory, or specify a path:")
|
442 |
-
output_dir = input("> ").strip() or None
|
443 |
-
|
444 |
-
itinerary = generate_travel_itinerary(user_input)
|
445 |
-
|
446 |
-
filepath = save_itinerary_to_file(itinerary, user_input, output_dir)
|
447 |
-
|
448 |
-
if filepath:
|
449 |
-
print(f"\nYour personalized travel itinerary is ready! Open {filepath} to view it.")
|
450 |
-
print("Thank you for using BlockX Travel Itinerary Generator!")
|
451 |
-
|
452 |
-
if __name__ == "__main__":
|
453 |
-
main()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|