Update app.py

app.py

@@ -7,30 +7,27 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+# Tool to fetch basic information about a YouTube channel using DuckDuckGo search
 @tool
 def get_youtube_channel_info(channel_name: str) -> str:
-    """A tool that provides basic information about a YouTube channel using DuckDuckGo search.
+    """Fetch basic information about a YouTube channel using DuckDuckGo search.
     Args:
-        channel_name: The name of the YouTube channel to get information about
+        channel_name: The name of the YouTube channel to search for.
     """
-    search_tool = DuckDuckGoSearchTool(max_results=5)  # Create DuckDuckGo search tool instance [[0]](#__0)
-    
-    # Search for the YouTube channel information
-    query = f"{channel_name} youtube channel information subscribers videos"
-    search_results = search_tool(query)  # Execute the search [[1]](#__1)
-    
-    # Format the results in a readable way
-    formatted_info = f"Information about YouTube channel '{channel_name}':\n\n"
-    
-    if search_results and len(search_results) > 0:
-        for i, result in enumerate(search_results[:3]):  # Limit to top 3 results
-            formatted_info += f"Source {i+1}: {result.get('title', 'No title')}\n"
-            formatted_info += f"Description: {result.get('snippet', 'No description')}\n"
-            formatted_info += f"URL: {result.get('link', 'No link')}\n\n"
-    else:
-        formatted_info += "No information found for this channel."
-    
-    return formatted_info  # Return the formatted information [[2]](#__2)
+    try:
+        # Use DuckDuckGoSearchTool to search for the YouTube channel
+        search_tool = DuckDuckGoSearchTool()
+        query = f"{channel_name} YouTube channel information"
+        search_results = search_tool.search(query)
+        
+        # Extract and return the top result
+        if search_results and len(search_results) > 0:
+            top_result = search_results[0]  # Get the first result
+            return f"Top search result for '{channel_name}': {top_result}"
+        else:
+            return f"No information found for YouTube channel '{channel_name}'."
+    except Exception as e:
+        return f"Error fetching YouTube channel information: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -60,7 +57,6 @@ model = HfApiModel(
     custom_role_conversions=None,
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -69,7 +65,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_youtube_channel_info, get_current_time_in_timezone, image_generation_tool],  # Added our tools here [[3]](#__3)
+    tools=[final_answer, get_youtube_channel_info, get_current_time_in_timezone],  # Add tools here
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -79,5 +75,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
 GradioUI(agent).launch()