Update app.py

app.py

@@ -7,7 +7,6 @@ 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:
     """Fetch basic information about a YouTube channel using DuckDuckGo search.
@@ -17,19 +16,31 @@ def get_youtube_channel_info(channel_name: str) -> str:
     try:
         # Initialize DuckDuckGoSearchTool with a maximum of 5 results
         search_tool = DuckDuckGoSearchTool(max_results=5)
-        query = f"{channel_name} YouTube channel Description"
+        query = f"{channel_name} YouTube channel information"
         print(f"Searching for: {query}")  # Debug print statement
 
         # Perform the search
         search_results = search_tool.forward(query)
         print(f"Raw search results:\n{search_results}")  # Debug print statement
 
-        # Return the formatted search results
-        return f"Search results for '{channel_name}':\n\n{search_results[0:3]}"
+        # Split the results into a list of lines
+        results_lines = search_results.split("\n\n")[1:]  # Skip the "## Search Results" header
+        print(f"Parsed results lines:\n{results_lines}")  # Debug print statement
+
+        # Extract the top 3 results
+        top_3_results = results_lines[:3]
+        print(f"Top 3 results:\n{top_3_results}")  # Debug print statement
+
+        # Concatenate the top 3 results into a single string
+        concatenated_results = "\n\n".join(top_3_results)
+        print(f"Concatenated top 3 results:\n{concatenated_results}")  # Debug print statement
+
+        return f"Top 3 search results for '{channel_name}':\n\n{concatenated_results}"
     except Exception as e:
         print(f"Error occurred while searching for '{channel_name}': {str(e)}")  # Debug print statement
         return f"Error fetching YouTube channel information: {str(e)}"
 
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.