Update app.py with new search features.

app.py

@@ -3,6 +3,10 @@ import datetime
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from tools.book_tool import BookSearchTool, BookDetailsTool
+from tools.arxiv_tool import ArxivSearchTool, LatestPapersTool
+from tools.search_tool import DuckDuckGoSearchTool, VisitWebpageTool
+from scheduler import initialize_scheduler
 from Gradio_UI import GradioUI
 
 @tool
@@ -48,41 +52,55 @@ def initialize_agent():
     
     # Initialize tools
     final_answer = FinalAnswerTool()
+    book_search = BookSearchTool()
+    book_details = BookDetailsTool()
+    arxiv_search = ArxivSearchTool()
+    latest_papers = LatestPapersTool()
     web_search = DuckDuckGoSearchTool(max_results=5)  # Initialize DDG search with 5 results
     visit_webpage = VisitWebpageTool()
     
-# Initialize components
-final_answer = FinalAnswerTool()
+    # Initialize the model
+    model = HfApiModel(
+        max_tokens=2096,
+        temperature=0.5,
+        model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+        custom_role_conversions=None,
+    )
 
-# Initialize the model
-model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
-)
+    # Load prompt templates
+    with open("prompts.yaml", 'r') as stream:
+        prompt_templates = yaml.safe_load(stream)
 
-# Load prompt templates
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
+    # Initialize agent with all tools
+    agent = CodeAgent(
+        model=model,
+        tools=[
+            final_answer,
+            calculator,
+            get_current_time_in_timezone,
+            book_search,
+            book_details,
+            arxiv_search,
+            latest_papers,
+            web_search,
+            visit_webpage
+        ],
+        max_steps=8,
+        verbosity_level=1,
+        grammar=None,
+        planning_interval=None,
+        name="ResearchAssistant",
+        description="An AI assistant specialized in helping users discover and research books, papers, and web content on AI and machine learning.",
+        prompt_templates=prompt_templates
+    )
 
-# Initialize agent with all tools
-agent = CodeAgent(
-    model=model,
-    tools=[
-        final_answer,
-        calculator,
-        get_current_time_in_timezone
-    ],
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
-)
+    return agent
 
 # Launch Gradio UI
 if __name__ == "__main__":
+    # Initialize the scheduler for daily ArXiv updates
+    scheduler = initialize_scheduler()
+    
+    # Initialize and launch the agent
+    agent = initialize_agent()
     GradioUI(agent).launch()