Update app.py

app.py

@@ -1,37 +1,19 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
-import datetime
-import requests
-import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-
 from Gradio_UI import GradioUI
 
 def get_ai_agent_info(tool: DuckDuckGoSearchTool) -> str:
-    """A tool that searches the net for information about becoming an AI agent.
-    Args:
-        tool: The DuckDuckGoSearchTool
-    """
+    """Searches the web for information about becoming an AI agent."""
     search_query = "becoming an AI agent"
-    result = tool.search(search_query)
-    return result
+    return tool.search(search_query)
 
-def get_ai_agent_roadmap() -> str:
-    """A tool that provides a roadmap for becoming an AI agent.
-    Args:
-        None
-    """
-    roadmap = """
-    Step 1: Learn the basics of programming
-    Step 2: Study machine learning and AI
-    Step 3: Build a portfolio of projects
-    Step 4: Join online communities and forums
-    Step 5: Read books and articles on AI
-    Step 6: Take online courses and certifications
-    """
-    return roadmap
+def get_ai_agent_roadmap(model: HfApiModel) -> str:
+    """Generates a roadmap for becoming an AI agent using AI model."""
+    prompt = "Provide a detailed roadmap for becoming an AI agent."
+    return model.generate(prompt)
 
-final_answer = FinalAnswerTool()
+# Initialize the AI model
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -39,26 +21,25 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-
-# Create instances of the DuckDuckGoSearchTool class
+# Load tools
+final_answer = FinalAnswerTool()
 search_tool = DuckDuckGoSearchTool()
-roadmap_tool = FinalAnswerTool()
+image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
+# Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
+# Create AI Agent
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, search_tool, roadmap_tool],  # add your tools here
+    tools=[final_answer, search_tool, image_generation_tool],
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
+    name="AI Career Guide",
+    description="An AI-powered assistant for learning how to become an AI agent.",
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+# Launch UI
+GradioUI(agent).launch()