Update app.py

app.py

@@ -3,17 +3,14 @@ import datetime
 import pytz
 import yaml
 from transformers import pipeline
-from smolagents.tools.search import DuckDuckGoSearchTool
+from duckduckgo_search import DDGS
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# Real summarization model (Facebook BART)
+# Real summarization model
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-# Real DuckDuckGo search tool
-search_tool = DuckDuckGoSearchTool()
-
-# 🌍 Real functional web search + summarization tool
+# Real DuckDuckGo search + summarization tool
 @tool
 def search_and_summarize(query: str, max_sentences: int) -> str:
     """A tool that performs a web search and returns a summary.
@@ -22,8 +19,14 @@ def search_and_summarize(query: str, max_sentences: int) -> str:
         max_sentences: Maximum number of sentences in the summary.
     """
     try:
-        pages = search_tool(query=query)
-        combined_text = "\n".join(pages[:3])
+        with DDGS() as ddgs:
+            results = ddgs.text(query, max_results=5)
+            search_texts = [r["body"] for r in results if "body" in r]
+            combined_text = "\n".join(search_texts)
+
+        if not combined_text.strip():
+            return "No content found to summarize."
+
         summary = summarizer(
             combined_text,
             max_length=60 * max_sentences,
@@ -34,7 +37,7 @@ def search_and_summarize(query: str, max_sentences: int) -> str:
     except Exception as e:
         return f"Error during search or summarization: {str(e)}"
 
-# ⏰ Working timezone tool
+# Timezone tool
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -48,10 +51,10 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Final answer tool (required)
+# Final answer tool
 final_answer = FinalAnswerTool()
 
-# Model setup (can change model here)
+# Model configuration
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -59,14 +62,14 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-# Load any extra tools if needed
+# Load image tool (optional)
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Load prompt templates
+# Load system prompts
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create agent
+# Agent setup
 agent = CodeAgent(
     model=model,
     tools=[search_and_summarize, get_current_time_in_timezone, final_answer],
@@ -79,9 +82,9 @@ agent = CodeAgent(
     prompt_templates=prompt_templates,
 )
 
-# Expose tools to REPL scope in case model tries to call them via python_interpreter
+# Register tools for use in python_interpreter
 globals()["search_and_summarize"] = search_and_summarize
 globals()["get_current_time_in_timezone"] = get_current_time_in_timezone
 
-# Launch the agent UI
+# Launch the UI
 GradioUI(agent).launch()