Update app.py

app.py

@@ -14,32 +14,28 @@ from Gradio_UI import GradioUI
 
 search_tool = DuckDuckGoSearchTool()
 
-news_sites = ["https://www.cnn.com/"]
-
-site_config = {
-    "https://www.cnn.com/": {'tag': 'h2', 'class': 'headline'}
-    }
-
 
 @tool
-def get_latest_news(news_sites: List[str]) -> Dict[str, List[str]]:
+def get_latest_news() -> Dict[str, List[str]]:
     """
-    Tool returns news headlines from multiple news sites.
-
-    Args:
-        news_sites (List[str]): A list of URLs of news sites to fetch headlines from.
+    Tool returns news headlines from predefined news sites.
 
     Returns:
         Dict[str, List[str]]: A dictionary where the keys are the news site URLs and the values are lists of headlines.
 
     Notes:
-        The function uses a predefined `site_config` dictionary to determine the HTML tag and class to extract headlines from each site.
-        The `site_config` dictionary should have the following structure:
+        The function uses a predefined `news_sites` list and a `site_config` dictionary to determine the HTML tag and class
+        to extract headlines from each site. The `site_config` dictionary should have the following structure:
         {
             "site_url": {'tag': 'html_tag', 'class': 'css_class'}
         }
         If a site is not found in `site_config`, it defaults to {'tag': 'h2', 'class': 'headline'}.
     """
+    news_sites = ["https://www.cnn.com/"]
+    site_config = {
+        "https://www.cnn.com/": {'tag': 'h2', 'class': 'headline'}
+    }
+
     headlines = {}
     for site in news_sites:
         try: