Update app.py

app.py

@@ -10,72 +10,89 @@ from Gradio_UI import GradioUI
 
 @tool
 def get_youtube_channel_info(channel_name: str) -> str:
-    """A tool that fetches information about a YouTube channel including subscriber count.
+    """A tool that fetches basic information about a YouTube channel including subscriber count and recent videos.
     Args:
         channel_name: The name of the YouTube channel to search for.
     """
     search_tool = DuckDuckGoSearchTool()
     
-    # Search for channel statistics specifically
-    stats_search_results = search_tool(f"{channel_name} youtube channel statistics subscribers")
-    print(f"stats_search_results: {stats_search_results}")
+    # 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 channel information
-    info_search_results = search_tool(f"{channel_name} youtube channel about")
-    print(f"info_search_results: {info_search_results}")
+    # Direct search for subscriber count
+    subscriber_search = search_tool(f"{channel_name} youtube subscribers count")
+    print(f"Subscriber search results: {subscriber_search}")
     
-    # Search specifically for subscriber count
-    sub_search_results = search_tool(f"how many subscribers does {channel_name} have on youtube")
-    print(f"sub_search_results: {sub_search_results}")
+    # Compile basic information
+    channel_info = f"YouTube Channel: {channel_name}\n\n"
     
-    # Compile information
-    channel_info = f"Information about YouTube channel '{channel_name}':\n\n"
-    channel_info += f"Channel Name: {channel_name}\n"
-    
-    # Extract subscriber count from search results
-    subscriber_info = "Could not determine subscriber count"
-    
-    # Check all search results for subscriber information
-    all_results = stats_search_results + sub_search_results + info_search_results
-    
-    for result in all_results:
+    # Extract subscriber count
+    subscriber_count = "Unknown"
+    for result in subscriber_search + channel_search:
         result_lower = result.lower()
-        # Look for common subscriber count patterns without regex
-        if "subscribers" in result_lower:
-            # Extract the sentence containing subscriber information
+        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_info = sentence.strip()
+                    subscriber_count = sentence.strip()
                     break
-            if subscriber_info != "Could not determine subscriber count":
+            if subscriber_count != "Unknown":
                 break
     
-    channel_info += f"Subscriber Count: {subscriber_info}\n"
+    channel_info += f"Subscribers: {subscriber_count}\n\n"
     
-    # Add channel description
+    # Try to get channel description/summary
+    description_search = search_tool(f"{channel_name} youtube channel about")
     channel_description = "No description available"
-    for result in info_search_results:
-        if "about" in result.lower() and channel_name.lower() in result.lower():
+    
+    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"Description: {channel_description}\n"
+    channel_info += f"Channel Summary: {channel_description}\n\n"
     
-    # Add content type/category if available
-    content_info = ""
-    for result in info_search_results:
-        if "content" in result.lower() or "videos" in result.lower():
-            content_info = result
-            break
-    
-    if content_info:
-        channel_info += f"Content: {content_info}\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
 
 
 
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.