Update app.py

app.py

@@ -49,7 +49,10 @@ def get_latest_news(source: str) -> str:
             return f"Error: {source} is not a supported news source"
         
         # Fetch and parse the page
-        response = requests.get(url, timeout=10)
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
+        }
+        response = requests.get(url, headers=headers, timeout=10)
         response.raise_for_status()
         soup = BeautifulSoup(response.text, 'html.parser')
         
@@ -61,6 +64,16 @@ def get_latest_news(source: str) -> str:
             headlines = [div.text.strip() for div in soup.select('div[class^="container__headline"]')][:5]
         elif source == "Reuters":
             headlines = [a.text.strip() for a in soup.select('a[data-testid="Heading"]')][:5]
+        elif source == "AP":
+            headlines = [a.text.strip() for a in soup.select('a[class*="Component-headline"]')][:5]
+        elif source == "Fox News":
+            headlines = [h2.text.strip() for h2 in soup.select('h2[class*="title"]')][:5]
+        elif source == "CBS News":
+            headlines = [h3.text.strip() for h3 in soup.select('h3[class*="item__hed"]')][:5]
+        elif source == "Wall Street Journal":
+            headlines = [h3.text.strip() for h3 in soup.select('h3[class*="WSJTheme--headline"]')][:5]
+        elif source == "Daily Mail":
+            headlines = [h2.text.strip() for h2 in soup.select('h2[class*="linkro-darkred"]')][:5]
         # Add other sources' parsing logic here...
         
         return f"Latest headlines from {source}:\n- " + "\n- ".join(headlines[:5])
@@ -68,6 +81,7 @@ def get_latest_news(source: str) -> str:
     except Exception as e:
         return f"Error fetching news from {source}: {str(e)}"
 
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """Fetches the current local time in a specified timezone and provides a human-friendly description.