Update app.py

app.py

@@ -1,14 +1,12 @@
 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
 
-# Step 1: Define the Web Search Tool
-search_tool = DuckDuckGoSearchTool(max_results=5)  # Searches the web using DuckDuckGo
+# Import the DuckDuckGo search tool
+search_tool = DuckDuckGoSearchTool(max_results=5)
 
-# Step 2: Define a Custom Summarization Tool
+# Custom summarization tool
 @tool
 def extract_summary_from_text(text: str) -> str:
     """Extracts a short summary from the given text.
@@ -17,7 +15,7 @@ def extract_summary_from_text(text: str) -> str:
     """
     return text[:300] + "..." if len(text) > 300 else text
 
-# Step 3: Define the AI Model
+# Hugging Face API model
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -25,12 +23,12 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Step 4: Load Prompt Templates
+# Load prompt templates
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Step 5: Create the WebAgent
-agent = WebAgent(
+# ✅ Use CodeAgent instead of WebAgent
+agent = CodeAgent(
     model=model,
     tools=[search_tool, extract_summary_from_text, FinalAnswerTool()],  # Add tools
     max_steps=6,
@@ -42,5 +40,5 @@ agent = WebAgent(
     prompt_templates=prompt_templates
 )
 
-# Step 6: Launch the Gradio UI
-GradioUI(agent).launch()
+# Launch Gradio UI
+GradioUI(agent).launch()