Update app.py

app.py

@@ -3,95 +3,34 @@ import datetime
 import requests
 import pytz
 import yaml
-import re
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
 @tool
 def get_youtube_channel_info(channel_name: str) -> str:
-    """A tool that fetches basic information about a YouTube channel including subscriber count and recent videos.
+    """A tool that provides basic information about a YouTube channel using DuckDuckGo search.
     Args:
-        channel_name: The name of the YouTube channel to search for.
+        channel_name: The name of the YouTube channel to get information about
     """
-    search_tool = DuckDuckGoSearchTool()
+    search_tool = DuckDuckGoSearchTool(max_results=5)  # Create DuckDuckGo search tool instance [[0]](#__0)
     
-    # Search for basic channel info and subscriber count
-    channel_search = search_tool(f"{channel_name} youtube channel")
-    print(f"Channel search results: {channel_search}")
+    # 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)
     
-    # Direct search for subscriber count
-    subscriber_search = search_tool(f"{channel_name} youtube subscribers count")
-    print(f"Subscriber search results: {subscriber_search}")
+    # Format the results in a readable way
+    formatted_info = f"Information about YouTube channel '{channel_name}':\n\n"
     
-    # Compile basic information
-    channel_info = f"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."
     
-    # Extract subscriber count
-    subscriber_count = "Unknown"
-    for result in subscriber_search + channel_search:
-        result_lower = result.lower()
-        if "subscribers" in result_lower and any(num in result_lower for num in ["million", "m subscribers", "k subscribers", " subscribers"]):
-            # Find the sentence with subscriber information
-            sentences = result.split('.')
-            for sentence in sentences:
-                if "subscribers" in sentence.lower():
-                    subscriber_count = sentence.strip()
-                    break
-            if subscriber_count != "Unknown":
-                break
-    
-    channel_info += f"Subscribers: {subscriber_count}\n\n"
-    
-    # Try to get channel description/summary
-    description_search = search_tool(f"{channel_name} youtube channel about")
-    channel_description = "No description available"
-    
-    for result in description_search + channel_search:
-        if ("about" in result.lower() or "description" in result.lower()) and channel_name.lower() in result.lower():
-            channel_description = result
-            break
-    
-    channel_info += f"Channel Summary: {channel_description}\n\n"
-    
-    # If no good description was found, fetch video titles to help analyze content
-    if channel_description == "No description available" or len(channel_description) < 50:
-        video_search = search_tool(f"{channel_name} youtube recent videos")
-        video_titles = []
-        
-        # Try to extract video titles from search results
-        for result in video_search:
-            # Look for patterns that might indicate video titles
-            if ":" in result and channel_name.lower() in result.lower():
-                # Potential video title after a colon
-                title_part = result.split(":", 1)[1].strip()
-                if 10 < len(title_part) < 100:  # Reasonable title length
-                    video_titles.append(title_part)
-            
-            # Alternative approach: look for quotes which might contain titles
-            quoted_parts = result.split('"')
-            if len(quoted_parts) > 1:
-                for i in range(1, len(quoted_parts), 2):
-                    potential_title = quoted_parts[i].strip()
-                    if 10 < len(potential_title) < 100:  # Reasonable title length
-                        video_titles.append(potential_title)
-        
-        # Deduplicate and limit to 10 titles
-        video_titles = list(dict.fromkeys(video_titles))[:10]
-        
-        if video_titles:
-            channel_info += "Recent Videos:\n"
-            for i, title in enumerate(video_titles, 1):
-                channel_info += f"{i}. {title}\n"
-        else:
-            # If we couldn't extract titles, do another search specifically for videos
-            direct_videos = search_tool(f"{channel_name} youtube videos")
-            channel_info += "Content Topics: " + ", ".join(direct_videos[:3])
-    
-    return channel_info
-
-
-
+    return formatted_info  # Return the formatted information [[2]](#__2)
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -108,15 +47,20 @@ 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 = FinalAnswerTool()
 
+# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
+# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
+
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',  # it is possible that this model may be overloaded
     custom_role_conversions=None,
 )
 
+
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -125,7 +69,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_youtube_channel_info, get_current_time_in_timezone, final_answer],  # Added your custom tools
+    tools=[final_answer, get_youtube_channel_info, get_current_time_in_timezone, image_generation_tool],  # Added our tools here [[3]](#__3)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -135,4 +79,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
+
 GradioUI(agent).launch()